Update app.py

app.py

@@ -230,9 +230,9 @@ def greet():
 
 # Helper function to update send button interactivity based on prereqs
 # MODIFIED: Signature takes exactly 4 inputs, removed *args, **kwargs
-# MODIFIED: Return type hint changed to gr.Button.update
+# FIXED: Return type hint corrected to gr.update
 # MODIFIED: Internal logic uses bool() check for simplicity
-def check_send_button_ready(hf_profile: gr.OAuthProfile | None, hf_token: gr.OAuthToken | None, gemini_key: str | None, gemini_model: str | None) -> gr.Button.update:
+def check_send_button_ready(hf_profile: gr.OAuthProfile | None, hf_token: gr.OAuthToken | None, gemini_key: str | None, gemini_model: str | None) -> gr.update:
     """Checks if HF login and Gemini configuration are complete and returns update for button interactivity."""
     # --- START ENHANCED DEBUGGING LOGS ---
     print("\n--- check_send_button_ready START ---")
@@ -257,8 +257,8 @@ def check_send_button_ready(hf_profile: gr.OAuthProfile | None, hf_token: gr.OAu
     print(f"check_send_button_ready - HF Ready: {is_logged_in}, Gemini Ready: {is_gemini_ready}, Button Ready (boolean): {is_ready}")
     print("--- check_send_button_ready END ---\n")
 
-    # Use gr.Button.update for clearer intent, although gr.update works
-    return gr.Button.update(interactive=is_ready)
+    # FIXED: Call gr.update instead of gr.Button.update
+    return gr.update(interactive=is_ready)
 
 
 # This is the main generator function for the workflow, triggered by the 'Send' button
@@ -271,7 +271,7 @@ def ai_workflow_chat(
     history: list[dict],
     hf_profile: gr.OAuthProfile | None,
     hf_token:   gr.OAuthToken   | None,
-    # Input parameters corresponding to the State variables read by .click
+    # Pass gemini_api_key and gemini_model as inputs - these come from the State variables
     gemini_api_key_state: str | None,
     gemini_model_state:   str | None,
     repo_id_state:  str | None,
@@ -1044,7 +1044,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                  # Call the update function and bind its output to the button component
                  check_send_button_ready,
                  inputs=send_button_interactive_binding_inputs, # Use the explicit list with renamed states
-                 outputs=[send_btn] # Update button interactivity using gr.Button.update return value
+                 outputs=[send_btn] # Update button interactivity using gr.update return value
             )
 
             # Handle Gemini Key Input change: Update key state -> Configure Gemini status -> Check prereqs and update button interactivity
@@ -1062,7 +1062,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 # Call the update function and bind its output to the button component
                 check_send_button_ready,
                 inputs=send_button_interactive_binding_inputs, # Use the explicit list with renamed states
-                outputs=[send_btn] # Update button interactivity using gr.Button.update return value
+                outputs=[send_btn] # Update button interactivity using gr.update return value
             )
 
             # Handle Gemini Model Selector change: Update model state -> Configure Gemini status -> Check prereqs and update button interactivity
@@ -1080,7 +1080,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 # Call the update function and bind its output to the button component
                 check_send_button_ready,
                 inputs=send_button_interactive_binding_inputs, # Use the explicit list with renamed states
-                outputs=[send_btn] # Update button interactivity using gr.Button.update return value
+                outputs=[send_btn] # Update button interactivity using gr.update return value
             )
 
             # Handle Grounding checkbox change: update grounding state
@@ -1155,7 +1155,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
         # Since gemini_api_key_state state starts as "", button will initially be disabled.
         check_send_button_ready,
         inputs=send_button_interactive_binding_inputs, # Use the explicit list with renamed states
-        outputs=[send_btn] # Update button interactivity using gr.Button.update return value
+        outputs=[send_btn] # Update button interactivity using gr.update return value
     ).then(
         # Action 4: Add the initial welcome message to the chatbot
         greet,