Update app.py

app.py

@@ -45,15 +45,14 @@ def _save_bytes_as_temp_mp4(data: bytes) -> str:
     finally:
         temp_file.close()
 
-def generate_video_with_auth(image, prompt, token: gr.OAuthToken | None, *args, progress=gr.Progress()):
+def generate_video_with_auth(image, prompt, profile: gr.OAuthProfile | None, progress=gr.Progress()):
     """
     Generate a video from an image using the Ovi model with authentication check.
     
     Args:
         image: Input image (PIL Image or file path)
         prompt: Text prompt describing the desired motion/animation
-        token: OAuth token for authentication
-        *args: Additional arguments from Gradio
+        profile: OAuth profile for authentication
         progress: Gradio progress tracker
     
     Returns:
@@ -61,7 +60,7 @@ def generate_video_with_auth(image, prompt, token: gr.OAuthToken | None, *args,
     """
     try:
         # Check authentication
-        if token is None or not getattr(token, "token", None):
+        if profile is None:
             return None, "❌ Sign in with Hugging Face to continue. This app uses your inference provider credits."
         
         if image is None:
@@ -97,8 +96,12 @@ def generate_video_with_auth(image, prompt, token: gr.OAuthToken | None, *args,
         
         progress(0.4, desc="Generating video with AI...")
         
-        # Create client with user's token
-        client = _client_from_token(token.token)
+        # Create client with user's OAuth token (not HF_TOKEN)
+        # IMPORTANT: Do not use bill_to parameter - this ensures user gets billed
+        client = InferenceClient(
+            provider="fal-ai",
+            api_key=profile.oauth_info.access_token,  # Use user's token
+        )
         
         # Generate video using the inference client
         try: