Update Gradio app with multiple files

app.py

@@ -23,7 +23,7 @@ def verify_pro_status(token: Optional[Union[gr.OAuthToken, str]]) -> bool:
     elif isinstance(token, str):
         token_str = token
     else:
-        return False
+        return False # Should not happen with correct type hints, but for safety
     
     try:
         user_info = whoami(token=token_str)
@@ -153,7 +153,7 @@ def generate_image_to_video(
 
 def generate_with_pro_auth(
     prompt: str, 
-    oauth_token: Optional[gr.OAuthToken] = None
+    oauth_token: Optional[gr.OAuthToken] = None # Gradio will auto-inject this based on type hint
 ) -> Tuple[Optional[str], str]:
     """
     Wrapper function that checks if user is PRO before generating text-to-video.
@@ -179,7 +179,7 @@ def generate_with_pro_auth(
 def generate_image_to_video_with_pro_auth(
     image_path: str,
     prompt: str,
-    oauth_token: Optional[gr.OAuthToken] = None
+    oauth_token: Optional[gr.OAuthToken] = None # Gradio will auto-inject this based on type hint
 ) -> Tuple[Optional[str], str]:
     """
     Wrapper function that checks if user is PRO before generating image-to-video.
@@ -271,7 +271,6 @@ def create_ui():
                 </div>
             """)
             
-            # Changed gr.TabbedInterface to gr.Tabs as per Gradio API for inline tab content
             with gr.Tabs() as tab_selector: 
                 with gr.TabItem("Text-to-Video", id=0):
                     with gr.Row():
@@ -321,14 +320,14 @@ def create_ui():
                     # Event handler for generation with queue disabled
                     generate_btn_text.click(
                         fn=generate_with_pro_auth,
-                        inputs=[prompt_input_text, gr.OAuthToken()], # Pass OAuth token
+                        inputs=[prompt_input_text], # OAuth token is auto-injected by type hint
                         outputs=[video_output_text, status_output_text],
                         queue=False,
                         api_name=False,
                         show_api=False,
                     )
 
-                with gr.TabItem("Image-to-Video", id=1): # Changed gr.Tab to gr.TabItem
+                with gr.TabItem("Image-to-Video", id=1):
                     with gr.Row():
                         with gr.Column(scale=1):
                             image_input = gr.Image(
@@ -374,7 +373,7 @@ def create_ui():
 
                     generate_btn_image.click(
                         fn=generate_image_to_video_with_pro_auth,
-                        inputs=[image_input, prompt_input_image, gr.OAuthToken()], # Pass OAuth token
+                        inputs=[image_input, prompt_input_image], # OAuth token is auto-injected by type hint
                         outputs=[video_output_image, status_output_image],
                         queue=False,
                         api_name=False,
@@ -389,7 +388,9 @@ def create_ui():
             """)
         
         def control_access(profile: Optional[gr.OAuthProfile] = None, oauth_token: Optional[gr.OAuthToken] = None):
-            """Control interface visibility based on PRO status."""
+            """Control interface visibility based on PRO status.
+            Gradio automatically injects gr.OAuthProfile and gr.OAuthToken based on type hints
+            when OAuth is enabled for the Space."""
             if not profile:
                 # User not logged in
                 return gr.update(visible=False), gr.update(visible=False)
@@ -439,10 +440,11 @@ def create_ui():
                 return gr.update(visible=False), gr.update(visible=True, value=message)
         
         # Check access on load
-        # gr.OAuthProfile and gr.OAuthToken are automatically provided by Gradio when OAuth is enabled
+        # No explicit inputs are needed here as gr.OAuthProfile and gr.OAuthToken are
+        # provided automatically by Gradio to the function based on type hints.
         demo.load(
             control_access,
-            inputs=[gr.OAuthProfile(), gr.OAuthToken()],
+            inputs=None, # Removed explicit instantiation of OAuthProfile and OAuthToken
             outputs=[main_interface, pro_message]
         )