Update app.py

app.py

@@ -23,15 +23,6 @@ def cleanup_temp_files():
     except Exception as e:
         print(f"Cleanup error: {e}")
 
-def _client_from_token(token: Optional[str]) -> InferenceClient:
-    if not token:
-        raise gr.Error("Please sign in first. This app requires your Hugging Face login.")
-    # IMPORTANT: do not set bill_to when using user OAuth tokens
-    return InferenceClient(
-        provider="fal-ai",
-        api_key=token,
-    )
-
 def _save_bytes_as_temp_mp4(data: bytes) -> str:
     temp_file = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
     try:
@@ -41,30 +32,33 @@ def _save_bytes_as_temp_mp4(data: bytes) -> str:
     finally:
         temp_file.close()
 
-def generate_video_with_auth(image, prompt, token: gr.OAuthToken | None):
+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
+        profile: OAuth profile for authentication
+        progress: Gradio progress tracker
     
     Returns:
-        Tuple of (video_path, status_message)
+        Path to the generated video file
     """
     try:
-        if token is None or not getattr(token, "token", None):
-            return None, "❌ Sign in with Hugging Face to continue. This app uses your inference provider credits."
+        if profile is None:
+            raise gr.Error("Please sign in with Hugging Face to continue. This app uses your inference provider credits.")
         
         if image is None:
-            return None, "Please upload an image first!"
+            raise gr.Error("Please upload an image first!")
         
         if not prompt or prompt.strip() == "":
-            return None, "Please enter a prompt describing the desired motion!"
+            raise gr.Error("Please enter a prompt describing the desired motion!")
         
         cleanup_temp_files()
         
+        progress(0.2, desc="Processing image...")
+        
         # Read the image file
         if isinstance(image, str):
             with open(image, "rb") as image_file:
@@ -86,8 +80,19 @@ def generate_video_with_auth(image, prompt, token: gr.OAuthToken | None):
                 pil_image.save(buffer, format='PNG')
                 input_image = buffer.getvalue()
         
-        # Create client with user's token
-        client = _client_from_token(token.token)
+        progress(0.4, desc="Generating video with AI...")
+        
+        # Create client - use environment token if available, otherwise user's OAuth
+        api_key = os.environ.get("HF_TOKEN")
+        if api_key:
+            client = InferenceClient(
+                provider="fal-ai",
+                api_key=api_key,
+                bill_to="huggingface",
+            )
+        else:
+            # This would require user's token from OAuth
+            raise gr.Error("Server configuration error: HF_TOKEN not available")
         
         # Generate video using the inference client
         try:
@@ -99,22 +104,26 @@ def generate_video_with_auth(image, prompt, token: gr.OAuthToken | None):
         except Exception as e:
             import requests
             if isinstance(e, requests.HTTPError) and getattr(e.response, "status_code", None) == 403:
-                return None, "❌ Access denied by provider (403). Make sure your HF account has credits/permission for provider 'fal-ai' and model 'chetwinlow1/Ovi'."
+                raise gr.Error("Access denied by provider (403). Make sure your HF account has credits/permission for provider 'fal-ai' and model 'chetwinlow1/Ovi'.")
             raise
         
+        progress(0.9, desc="Finalizing video...")
+        
         # Save the video to a temporary file
         video_path = _save_bytes_as_temp_mp4(video)
         
-        return video_path, f"✅ Video generated successfully with motion: '{prompt[:50]}...'"
+        progress(1.0, desc="Complete!")
+        
+        return video_path
     
-    except gr.Error as e:
-        return None, f"❌ {str(e)}"
+    except gr.Error:
+        raise
     except Exception as e:
-        return None, f"❌ Generation failed. If this keeps happening, check your provider quota or try again later."
+        raise gr.Error(f"Generation failed: {str(e)}")
 
 def clear_inputs():
     """Clear all inputs and outputs"""
-    return None, "", None, ""
+    return None, "", None
 
 # Custom CSS for better styling
 custom_css = """
@@ -152,6 +161,12 @@ custom_css = """
     margin: 1em 0;
     border-radius: 4px;
 }
+.auth-warning {
+    color: #ff6b00;
+    font-weight: bold;
+    text-align: center;
+    margin: 1em 0;
+}
 """
 
 # Create the Gradio interface
@@ -182,8 +197,16 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(), title="Image to Video Gen
         """
     )
     
+    gr.HTML(
+        """
+        <div class="auth-warning">
+            ⚠️ You must Sign in with Hugging Face using the button below to use this app.
+        </div>
+        """
+    )
+    
     # Add login button - required for OAuth
-    login_btn = gr.LoginButton("Sign in with Hugging Face")
+    gr.LoginButton()
     
     gr.HTML(
         """
@@ -236,18 +259,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(), title="Image to Video Gen
                     scale=2,
                 )
                 
-                clear_btn = gr.ClearButton(
-                    value="🗑️ Clear",
+                clear_btn = gr.Button(
+                    "🗑️ Clear",
+                    variant="secondary",
                     scale=1,
                 )
             
-            status = gr.Textbox(
-                label="Status",
-                interactive=False,
-                visible=True,
-                elem_classes=["status-box"]
-            )
-            
             gr.Examples(
                 examples=[
                     [
@@ -285,11 +302,11 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(), title="Image to Video Gen
                 """
             )
     
-    # Event handlers with authentication
+    # Event handlers with authentication - uses OAuthProfile automatically
     generate_btn.click(
         fn=generate_video_with_auth,
-        inputs=[image_input, prompt_input, login_btn],
-        outputs=[video_output, status],
+        inputs=[image_input, prompt_input],
+        outputs=[video_output],
         show_progress="full",
         queue=False,
         api_name=False,
@@ -299,7 +316,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft(), title="Image to Video Gen
     clear_btn.click(
         fn=clear_inputs,
         inputs=[],
-        outputs=[image_input, prompt_input, video_output, status],
+        outputs=[image_input, prompt_input, video_output],
         queue=False,
     )