Update app.py

app.py

@@ -261,6 +261,15 @@ def check_send_button_ready(hf_profile: gr.OAuthProfile | None, hf_token: gr.OAu
     # FIXED: Call gr.update instead of gr.Button.update
     return gr.update(interactive=is_ready)
 
+# --- New wrapper function to handle the state updates and call the checker ---
+# This function takes the required state variables as inputs and calls the logic.
+# This function will be called by the .change() events of the state variables.
+def update_send_button_state(profile: gr.OAuthProfile | None, token: gr.OAuthToken | None, gemini_key: str | None, gemini_model: str | None) -> gr.update:
+    """Wrapper function to read states and update button interactivity."""
+    # Pass the received values directly to the core check function
+    return check_send_button_ready(profile, token, gemini_key, gemini_model)
+# --- End of new wrapper function ---
+
 
 # This is the main generator function for the workflow, triggered by the 'Send' button
 # NOTE: This function MUST accept ALL state variables as inputs that it might need to modify or pass through.
@@ -1037,42 +1046,41 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
     send_button_update_output = [send_btn]
 
 
-    # Trigger check_send_button_ready whenever any prerequisite state changes
-    # These now directly call check_send_button_ready with the specific state variables
+    # Trigger update_send_button_state whenever any prerequisite state changes
+    # This uses the new wrapper function which expects exactly 4 inputs.
+    # The inputs list ensures those 4 state values are passed.
     hf_profile.change(
-        check_send_button_ready,
-        inputs=send_button_interactive_binding_inputs,
-        outputs=send_button_update_output,
-        # run_as_event=True # Removed this unless necessary, simplifies behavior
+        update_send_button_state, # Call the wrapper function
+        inputs=send_button_interactive_binding_inputs, # Pass all 4 prerequisite states
+        outputs=send_button_update_output, # Update only the send button
     )
     hf_token.change(
-        check_send_button_ready,
-        inputs=send_button_interactive_binding_inputs,
-        outputs=send_button_update_output,
-        # run_as_event=True # Removed
+        update_send_button_state, # Call the wrapper function
+        inputs=send_button_interactive_binding_inputs, # Pass all 4 prerequisite states
+        outputs=send_button_update_output, # Update only the send button
     )
+    # gemini_input.change updates gemini_api_key_state, which triggers gemini_api_key_state.change
     gemini_api_key_state.change(
-        check_send_button_ready,
-        inputs=send_button_interactive_binding_inputs,
-        outputs=send_button_update_output,
-        # run_as_event=True # Removed
+        update_send_button_state, # Call the wrapper function
+        inputs=send_button_interactive_binding_inputs, # Pass all 4 prerequisite states
+        outputs=send_button_update_output, # Update only the send button
     )
+    # model_selector.change updates gemini_model_state, which triggers gemini_model_state.change
     gemini_model_state.change(
-        check_send_button_ready,
-        inputs=send_button_interactive_binding_inputs,
-        outputs=send_button_update_output,
-        # run_as_event=True # Removed
+        update_send_button_state, # Call the wrapper function
+        inputs=send_button_interactive_binding_inputs, # Pass all 4 prerequisite states
+        outputs=send_button_update_output, # Update only the send button
     )
 
 
-    # Handle login button click: Update profile/token state -> Their .change handlers trigger check_send_button_ready
+    # Handle login button click: Update profile/token state -> Their .change handlers trigger update_send_button_state
     login_btn.click(
          lambda x: (x[0], x[1]),
          inputs=[login_btn],
          outputs=[hf_profile, hf_token] # Updating these states will trigger their .change handlers
-    ) # Removed .then(check_send_button_ready, ...)
+    )
 
-    # Handle Gemini Key Input change: Update key state -> Configure Gemini status -> State .change handler triggers check_send_button_ready
+    # Handle Gemini Key Input change: Update key state -> Configure Gemini status -> State .change handler triggers update_send_button_state
     gemini_input.change(
         lambda k: k,
         inputs=[gemini_input],
@@ -1082,9 +1090,9 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
         configure_gemini,
         inputs=[gemini_api_key_state, gemini_model_state],
         outputs=[gemini_status] # Update Gemini status based on new key
-    ) # Removed .then(check_send_button_ready, ...)
+    )
 
-    # Handle Gemini Model Selector change: Update model state -> Configure Gemini status -> State .change handler triggers check_send_button_ready
+    # Handle Gemini Model Selector change: Update model state -> Configure Gemini status -> State .change handler triggers update_send_button_state
     model_selector.change(
         lambda m: m,
         inputs=[model_selector],
@@ -1094,7 +1102,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
         configure_gemini,
         inputs=[gemini_api_key_state, gemini_model_state],
         outputs=[gemini_status] # Update Gemini status based on new model
-    ) # Removed .then(check_send_button_ready, ...)
+    )
 
 
     # Handle Grounding checkbox change: update grounding state
@@ -1166,8 +1174,18 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
         configure_gemini,
         inputs=[gemini_api_key_state, gemini_model_state], # Use RENAMED state variables as inputs
         outputs=[gemini_status] # Update Gemini status display
-    ) # Removed .then(check_send_button_ready, ...). The state changes from LoginButton/Gemini init *should* trigger the .change handlers above.
-    # The greet message will be added automatically by the chat UI initialization.
+    ).then(
+        # Action 3: After initial load checks, update the button state based on initial states
+        # Use the new wrapper function here as well.
+        update_send_button_state,
+        inputs=send_button_interactive_binding_inputs, # Pass all 4 prerequisite states
+        outputs=send_button_update_output, # Update the send button
+    ).then(
+        # Action 4: Add the initial welcome message to the chat history
+        greet,
+        inputs=None,
+        outputs=chatbot # Updates the chatbot display
+    )
 
 
 # The main workflow function and other helper functions are correctly defined OUTSIDE the gr.Blocks context