Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 import os
 import tempfile
-from typing import Optional, Tuple
-from huggingface_hub import InferenceClient
+from typing import Optional, Tuple, Union
+from huggingface_hub import InferenceClient, whoami
 
 # Initialize Hugging Face Inference Client with fal-ai provider
 client = InferenceClient(
@@ -11,6 +11,28 @@ 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 generate_video(
     prompt: str,
     duration: int = 8,
@@ -52,85 +74,218 @@ def generate_video(
         error_msg = f"❌ Error generating video: {str(e)}"
         return None, error_msg
 
-def simple_generate(prompt: str) -> Optional[str]:
-    """Simplified wrapper for generate_video that only takes prompt and returns video."""
+def generate_with_pro_auth(
+    prompt: str, 
+    manual_token: str,
+    oauth_token: Optional[gr.OAuthToken] = None
+) -> Tuple[Optional[str], str]:
+    """
+    Wrapper function that checks if user is PRO before generating video.
+    """
+    # Check if user is PRO (either through OAuth or manual token)
+    if not (verify_pro_status(oauth_token) or verify_pro_status(manual_token)):
+        raise gr.Error("Access Denied. This app is exclusively for Hugging Face PRO users. Please subscribe to PRO to use this app.")
+    
     if not prompt or prompt.strip() == "":
-        return None
+        return None, "❌ Please enter a prompt"
     
-    video_path, status = generate_video(prompt, duration=8, size="1280x720", api_key=None)
+    # Use the oauth token if available, otherwise manual token
+    token_to_use = oauth_token.token if oauth_token else manual_token
     
-    # If there's an error, you might want to handle it
-    if video_path is None:
-        print(f"Generation failed: {status}")
+    # Generate video with the PRO user's token
+    video_path, status = generate_video(
+        prompt, 
+        duration=8, 
+        size="1280x720", 
+        api_key=token_to_use
+    )
     
-    return video_path
+    return video_path, status
 
-def generate_with_auth(prompt: str, profile: gr.OAuthProfile | None) -> Optional[str]:
-    """
-    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")
+def simple_generate(prompt: str) -> Optional[str]:
+    """Simplified wrapper for examples that only returns video."""
+    if not prompt or prompt.strip() == "":
+        return None
     
-    # User is logged in, proceed with video generation
-    return simple_generate(prompt)
+    video_path, _ = generate_video(prompt, duration=8, size="1280x720", api_key=None)
+    return video_path
 
 def create_ui():
-    """Create the Gradio interface."""
-    with gr.Blocks(title="Sora-2 Text-to-Video Generator", theme=gr.themes.Soft()) as demo:
+    """Create the Gradio interface with PRO verification."""
+    
+    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:
         gr.HTML("""
-            <div style="text-align: center; max-width: 700px; margin: 0 auto;">
-                <h1>🎬 Sora-2 Text-to-Video Generator</h1>
-                <p>Generate videos using Sora-2 model hosted by FAL AI.</p>
-                <p style='color: orange;'>⚠️ You must Sign in with Hugging Face using the button to use this app.</p>
-                <p>Built with <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank">anycoder</a></p>
+            <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;">Generate stunning videos using OpenAI's Sora-2 model</p>
+                <p id="sub_title" style="font-size: 1em; margin-top: 10px;">
+                    <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>
             </div>
         """)
         
-        # Add login button - required for OAuth
+        # Login button for OAuth
         gr.LoginButton()
         
-        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")
+        # PRO message for non-PRO users
+        pro_message = gr.Markdown(visible=False)
+        
+        # Main interface (hidden by default)
+        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>
+            """)
+            
+            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
+                    )
+                    
+                    with gr.Accordion("Advanced Settings", open=False):
+                        gr.Markdown("*Coming soon: Duration and resolution controls*")
+                    
+                    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
+                    )
             
-            with gr.Column(scale=1):
-                video_output = gr.Video(
-                    label="Generated Video",
-                    height=400,
-                    interactive=False,
-                    show_download_button=True
-                )
+            # Hidden manual token input (for API usage)
+            manual_token = gr.Textbox(
+                label="Manual Token (for API usage)",
+                placeholder="Paste your HF token here if using the API",
+                type="password",
+                visible=False
+            )
+            
+            # Examples section with queue disabled
+            gr.Examples(
+                examples=[
+                    "A serene beach at sunset with waves gently rolling onto the shore",
+                    "A butterfly emerging from its chrysalis in slow motion",
+                    "Northern lights dancing across a starry night sky",
+                    "A bustling city street transitioning from day to night in timelapse",
+                    "A close-up of coffee being poured into a cup with steam rising",
+                    "Cherry blossoms falling in slow motion in a Japanese garden"
+                ],
+                inputs=prompt_input,
+                outputs=video_output,
+                fn=simple_generate,  # Examples use simplified function
+                cache_examples=False,
+                api_name=False,
+                show_api=False,
+            )
+            
+            # Event handler for generation with queue disabled
+            generate_btn.click(
+                fn=generate_with_pro_auth,
+                inputs=[prompt_input, manual_token],
+                outputs=[video_output, status_output],
+                queue=False,
+                api_name=False,
+                show_api=False,
+            )
+            
+            # Footer
+            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>
+                    <p>Built with ❤️ using <a href="https://huggingface.co/spaces/akhaliq/anycoder" target="_blank">anycoder</a></p>
+                </div>
+            """)
         
-        # Event handler - note the function now takes OAuthProfile
-        generate_btn.click(
-            fn=generate_with_auth,
-            inputs=[prompt_input],
-            outputs=video_output,
-            queue=False,
-            api_name=False,
-            show_api=False,
-        )
+        def control_access(profile: Optional[gr.OAuthProfile] = None, oauth_token: Optional[gr.OAuthToken] = None):
+            """Control interface visibility based on PRO status."""
+            if not profile:
+                # User not logged in
+                return gr.update(visible=False), gr.update(visible=False)
+            
+            if verify_pro_status(oauth_token):
+                # User is PRO - show main interface
+                return gr.update(visible=True), gr.update(visible=False)
+            else:
+                # User is not PRO - show upgrade message
+                message = """
+                ## ✨ Exclusive Access for PRO Users
+                
+                Thank you for your interest in the Sora-2 Text-to-Video Generator!
+                
+                This advanced AI video generation tool is available exclusively for Hugging Face **PRO** members.
+                
+                ### What you get with PRO:
+                - ✅ Unlimited access to Sora-2 video generation
+                - ✅ High-quality video outputs up to 1280x720
+                - ✅ Fast generation times with priority queue
+                - ✅ Access to other exclusive PRO Spaces
+                - ✅ Support the development of cutting-edge AI tools
+                
+                ### Ready to create amazing videos?
+                
+                <div style="text-align: center; margin: 30px 0;">
+                    <a href="http://huggingface.co/subscribe/pro?source=sora2_video" target="_blank" style="
+                        display: inline-block;
+                        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                        color: white;
+                        padding: 12px 30px;
+                        border-radius: 25px;
+                        text-decoration: none;
+                        font-weight: bold;
+                        font-size: 1.1em;
+                        box-shadow: 0 4px 15px rgba(102, 126, 234, 0.3);
+                        transition: transform 0.2s;
+                    ">
+                        🚀 Become a PRO Today!
+                    </a>
+                </div>
+                
+                <p style="text-align: center; color: #666; margin-top: 20px;">
+                    Join thousands of creators who are already using PRO tools to bring their ideas to life.
+                </p>
+                """
+                return gr.update(visible=False), gr.update(visible=True, value=message)
         
-        # Example usage guidance with queue features disabled
-        gr.Examples(
-            examples=[
-                "A serene beach at sunset with waves gently rolling onto the shore",
-                "A butterfly emerging from its chrysalis in slow motion",
-                "Northern lights dancing across a starry night sky",
-                "A bustling city street transitioning from day to night in timelapse"
-            ],
-            inputs=prompt_input,
-            outputs=video_output,
-            fn=simple_generate,  # Examples use the simple function without auth
-            cache_examples=False,
-            api_name=False,
-            show_api=False,
+        # Check access on load
+        demo.load(
+            control_access,
+            inputs=None,
+            outputs=[main_interface, pro_message]
         )
     
     return demo
@@ -138,7 +293,7 @@ def create_ui():
 # Launch the application
 if __name__ == "__main__":
     app = create_ui()
-    # Launch without special auth parameters
+    # Launch without special auth parameters and no queue
     # OAuth is enabled via Space metadata (hf_oauth: true in README.md)
     app.launch(
         show_api=False,