Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 import tempfile
 import shutil
 from typing import Optional, Tuple, Union
-from huggingface_hub import InferenceClient, whoami
+from huggingface_hub import InferenceClient
 from pathlib import Path
 
 # Initialize Hugging Face Inference Client with fal-ai provider
@@ -13,28 +13,6 @@ client = InferenceClient(
     bill_to="huggingface",
 )
 
-def verify_pro_status(token: Optional[Union[gr.OAuthToken, str]]) -> bool:
-    """Verifies if the user is a Hugging Face PRO user or part of an enterprise org."""
-    if not token:
-        return False
-    
-    if isinstance(token, gr.OAuthToken):
-        token_str = token.token
-    elif isinstance(token, str):
-        token_str = token
-    else:
-        return False
-    
-    try:
-        user_info = whoami(token=token_str)
-        return (
-            user_info.get("isPro", False) or
-            any(org.get("isEnterprise", False) for org in user_info.get("orgs", []))
-        )
-    except Exception as e:
-        print(f"Could not verify user's PRO/Enterprise status: {e}")
-        return False
-
 def cleanup_temp_files():
     """Clean up old temporary video files to prevent storage overflow."""
     try:
@@ -88,7 +66,6 @@ def generate_video(
     except Exception as e:
         return None, f"❌ Error generating video: {str(e)}"
 
-# --- NEW: image -> video support ---
 def generate_video_from_image(
     image: Union[str, bytes],
     prompt: str,
@@ -136,13 +113,13 @@ def generate_video_from_image(
     except Exception as e:
         return None, f"❌ Error generating video from image: {str(e)}"
 
-def generate_with_pro_auth(
+def generate_with_auth(
     prompt: str, 
-    oauth_token: Optional[gr.OAuthToken] = None
+    profile: gr.OAuthProfile | None
 ) -> Tuple[Optional[str], str]:
-    """Wrapper function that checks if user is PRO before generating video."""
-    if not verify_pro_status(oauth_token):
-        raise gr.Error("Access Denied. This app is exclusively for Hugging Face PRO users.")
+    """Wrapper function that checks if user is logged in before generating video."""
+    if profile is None:
+        raise gr.Error("Click Sign in with Hugging Face button to use this app for free")
     
     if not prompt or prompt.strip() == "":
         return None, "❌ Please enter a prompt"
@@ -154,42 +131,24 @@ def generate_with_pro_auth(
         api_key=None
     )
 
-# --- NEW: PRO-gated wrapper for image -> video ---
-def generate_with_pro_auth_image(
+def generate_with_auth_image(
     prompt: str,
-    image_path: Optional[str] = None,
-    oauth_token: Optional[gr.OAuthToken] = None
+    image_path: Optional[str],
+    profile: gr.OAuthProfile | None
 ) -> Tuple[Optional[str], str]:
-    """Checks PRO status then calls image->video generator."""
-    if not verify_pro_status(oauth_token):
-        raise gr.Error("Access Denied. This app is exclusively for Hugging Face PRO users.")
+    """Checks login status then calls image->video generator."""
+    if profile is None:
+        raise gr.Error("Click Sign in with Hugging Face button to use this app for free")
     if not image_path:
         return None, "❌ Please upload an image"
     return generate_video_from_image(image=image_path, prompt=prompt, api_key=None)
 
-def simple_generate(prompt: str) -> Optional[str]:
-    """Simplified wrapper for examples that only returns video."""
-    if not prompt or prompt.strip() == "":
-        return None
-    video_path, _ = generate_video(prompt, duration=8, size="1280x720", api_key=None)
-    return video_path
-
 def create_ui():
     css = '''
     .logo-dark{display: none}
     .dark .logo-dark{display: block !important}
     .dark .logo-light{display: none}
     #sub_title{margin-top: -20px !important}
-    .pro-badge{
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: white;
-        padding: 4px 12px;
-        border-radius: 20px;
-        font-size: 0.9em;
-        font-weight: bold;
-        display: inline-block;
-        margin-left: 8px;
-    }
     '''
     
     with gr.Blocks(title="Sora-2 Text-to-Video Generator", theme=gr.themes.Soft(), css=css) as demo:
@@ -197,89 +156,77 @@ def create_ui():
             <div style="text-align: center; max-width: 800px; margin: 0 auto;">
                 <h1 style="font-size: 2.5em; margin-bottom: 0.5em;">
                     🎬 Sora-2 Text-to-Video Generator
-                    <span class="pro-badge">PRO</span>
                 </h1>
                 <p style="font-size: 1.1em; color: #666; margin-bottom: 20px;">Generate stunning videos using OpenAI's Sora-2 model</p>
-                <p id="sub_title" style="font-size: 1em; margin-top: 20px; margin-bottom: 15px;">
-                    <strong>Exclusive access for Hugging Face PRO users.</strong> 
-                    <a href="http://huggingface.co/subscribe/pro?source=sora2_video" target="_blank" style="color: #667eea;">Subscribe to PRO →</a>
-                </p>
+                <p style='color: orange;'>⚠️ You must Sign in with Hugging Face using the button to use this app.</p>
                 <p style="font-size: 0.9em; color: #999; margin-top: 15px;">
                     Built with <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank" style="color: #667eea;">anycoder</a>
                 </p>
             </div>
         """)
         
+        # Add login button - required for OAuth
         gr.LoginButton()
-        pro_message = gr.Markdown(visible=False)
-        main_interface = gr.Column(visible=False)
         
-        with main_interface:
-            gr.HTML("""<div style="text-align: center; margin: 20px 0;">
-                <p style="color: #28a745; font-weight: bold;">✨ Welcome PRO User! You have full access to Sora-2.</p>
-            </div>""")
-            
-            # Text -> Video
-            with gr.Row():
-                with gr.Column(scale=1):
-                    prompt_input = gr.Textbox(
-                        label="Enter your prompt",
-                        placeholder="Describe the video you want to create...",
-                        lines=4
-                    )
-                    generate_btn = gr.Button("🎥 Generate Video", variant="primary", size="lg")
-                with gr.Column(scale=1):
-                    video_output = gr.Video(label="Generated Video", height=400, interactive=False, show_download_button=True)
-                    status_output = gr.Textbox(label="Status", interactive=False, visible=True)
-            
-            generate_btn.click(
-                fn=generate_with_pro_auth,
-                inputs=[prompt_input],
-                outputs=[video_output, status_output],
-                queue=False
-            )
-
-            # --- NEW: Image -> Video UI ---
-            gr.HTML("""
-                <div style="text-align: center; margin: 40px 0 10px;">
-                    <h3 style="margin-bottom: 8px;">🖼️ ➜ 🎬 Image → Video (beta)</h3>
-                    <p style="color:#666; margin:0;">Turn a single image into a short video with a guiding prompt.</p>
-                </div>
-            """)
-            with gr.Row():
-                with gr.Column(scale=1):
-                    img_prompt_input = gr.Textbox(
-                        label="Describe how the scene should evolve",
-                        placeholder="e.g., The cat starts to dance and spins playfully",
-                        lines=3,
-                    )
-                    image_input = gr.Image(label="Upload an image", type="filepath")
-                    generate_img_btn = gr.Button("🎥 Generate from Image", variant="primary")
-                with gr.Column(scale=1):
-                    video_output_img = gr.Video(label="Generated Video (from Image)", height=400, interactive=False, show_download_button=True)
-                    status_output_img = gr.Textbox(label="Status", interactive=False, visible=True)
-
-            generate_img_btn.click(
-                fn=generate_with_pro_auth_image,
-                inputs=[img_prompt_input, image_input],
-                outputs=[video_output_img, status_output_img],
-                queue=False
-            )
-            
-            gr.HTML("""<div style="text-align: center; margin-top: 40px; padding: 20px; border-top: 1px solid #e0e0e0;">
-                <h3 style="color: #667eea;">Thank you for being a PRO user! 🤗</h3>
-            </div>""")
+        # Text -> Video
+        with gr.Row():
+            with gr.Column(scale=1):
+                prompt_input = gr.Textbox(
+                    label="Enter your prompt",
+                    placeholder="Describe the video you want to create...",
+                    lines=4
+                )
+                generate_btn = gr.Button("🎥 Generate Video", variant="primary", size="lg")
+            with gr.Column(scale=1):
+                video_output = gr.Video(label="Generated Video", height=400, interactive=False, show_download_button=True)
+                status_output = gr.Textbox(label="Status", interactive=False, visible=True)
         
-        def control_access(profile: Optional[gr.OAuthProfile] = None, oauth_token: Optional[gr.OAuthToken] = None):
-            if not profile:
-                return gr.update(visible=False), gr.update(visible=False)
-            if verify_pro_status(oauth_token):
-                return gr.update(visible=True), gr.update(visible=False)
-            else:
-                message = "## ✨ Exclusive Access for PRO Users\n\nThis tool is available exclusively for Hugging Face **PRO** members."
-                return gr.update(visible=False), gr.update(visible=True, value=message)
+        generate_btn.click(
+            fn=generate_with_auth,
+            inputs=[prompt_input],
+            outputs=[video_output, status_output],
+            queue=False
+        )
+
+        # Image -> Video UI
+        gr.HTML("""
+            <div style="text-align: center; margin: 40px 0 10px;">
+                <h3 style="margin-bottom: 8px;">🖼️ ➜ 🎬 Image → Video (beta)</h3>
+                <p style="color:#666; margin:0;">Turn a single image into a short video with a guiding prompt.</p>
+            </div>
+        """)
+        with gr.Row():
+            with gr.Column(scale=1):
+                img_prompt_input = gr.Textbox(
+                    label="Describe how the scene should evolve",
+                    placeholder="e.g., The cat starts to dance and spins playfully",
+                    lines=3,
+                )
+                image_input = gr.Image(label="Upload an image", type="filepath")
+                generate_img_btn = gr.Button("🎥 Generate from Image", variant="primary")
+            with gr.Column(scale=1):
+                video_output_img = gr.Video(label="Generated Video (from Image)", height=400, interactive=False, show_download_button=True)
+                status_output_img = gr.Textbox(label="Status", interactive=False, visible=True)
+
+        generate_img_btn.click(
+            fn=generate_with_auth_image,
+            inputs=[img_prompt_input, image_input],
+            outputs=[video_output_img, status_output_img],
+            queue=False
+        )
         
-        demo.load(control_access, inputs=None, outputs=[main_interface, pro_message])
+        # Example usage guidance
+        gr.Examples(
+            examples=[
+                "A majestic golden eagle soaring through a vibrant sunset sky"
+            ],
+            inputs=prompt_input,
+            outputs=video_output,
+            fn=generate_video,  # Examples use the original function
+            cache_examples=False,
+            api_name=False,
+            show_api=False,
+        )
     
     return demo
 
@@ -292,4 +239,11 @@ if __name__ == "__main__":
         print(f"Initial cleanup error: {e}")
     
     app = create_ui()
-    app.launch(show_api=False, enable_monitoring=False, quiet=True, max_threads=10)
+    # Launch without special auth parameters
+    # OAuth is enabled via Space metadata (hf_oauth: true in README.md)
+    app.launch(
+        show_api=False,
+        enable_monitoring=False,
+        quiet=True,
+        max_threads=10
+    )