Update app.py

app.py

@@ -2,15 +2,29 @@ import gradio as gr
 from models import generate_image, MODEL_ID
 from config import APPLE_TENCENT_THEME
 
+def generate_image_with_auth(prompt: str, profile: gr.OAuthProfile | None):
+    """
+    Wrapper function that checks if user is logged in before generating image.
+    """
+    if profile is None:
+        raise gr.Error("Please log in with Hugging Face to use this app.")
+    
+    # User is logged in, proceed with image generation
+    return generate_image(prompt)
+
 def create_ui():
     with gr.Blocks(title=f"Tencent HunyuanImage-3.0 Demo", theme=APPLE_TENCENT_THEME) as demo:
         gr.HTML(
             f"<div style='text-align: center; max-width: 700px; margin: 0 auto;'>"
             f"<h1>Tencent {MODEL_ID.split('/')[-1]}</h1>"
             f"<p>Generate images using Tencent's state-of-the-art model hosted by FAL AI.</p>"
+            f"<p style='color: orange;'>⚠️ You must log in with Hugging Face to use this app.</p>"
             f"Built with <a href='https://huggingface.co/spaces/akhaliq/anycoder' target='_blank'>anycoder</a>"
             f"</div>"
         )
+        
+        # Add login button - required for OAuth
+        gr.LoginButton()
 
         with gr.Row():
             with gr.Column(scale=1):
@@ -30,14 +44,14 @@ def create_ui():
                     show_download_button=True
                 )
 
-        # Set up the event listener with all queue-related features disabled
+        # Set up the event listener - note the function now takes OAuthProfile
         generate_btn.click(
-            fn=generate_image,
+            fn=generate_image_with_auth,
             inputs=[prompt_input],
             outputs=[output_image],
-            queue=False,  # Disable queue for this event
-            api_name=False,  # Disable API endpoint creation
-            show_api=False,  # Don't show this in API docs
+            queue=False,
+            api_name=False,
+            show_api=False,
         )
         
         # Example usage guidance with queue features disabled
@@ -47,19 +61,20 @@ def create_ui():
             ],
             inputs=prompt_input,
             outputs=output_image,
-            fn=generate_image,
-            cache_examples=False,  # Don't cache examples
-            api_name=False,  # No API endpoint for examples
-            show_api=False,  # Don't show in API docs
+            fn=generate_image,  # Examples use the original function
+            cache_examples=False,
+            api_name=False,
+            show_api=False,
         )
 
     return demo
 
 if __name__ == "__main__":
     app = create_ui()
-    # Launch without queue, API, and monitoring features
+    # Launch without special auth parameters
+    # OAuth is enabled via Space metadata (hf_oauth: true in README.md)
     app.launch(
-        show_api=False,  # Hide API documentation
-        enable_monitoring=False,  # Disable monitoring to prevent any background queuing
-        quiet=True,  # Reduce console output
+        show_api=False,
+        enable_monitoring=False,
+        quiet=True,
     )