Update app.py

app.py

@@ -166,7 +166,6 @@ def add_bot_message(history: list[dict], bot_message: str) -> list[dict]:
     history.append({"role": "assistant", "content": bot_message})
     return history
 
-# Modified function signature to accept *args and **kwargs
 def ai_workflow_chat(
     message: str,
     history: list[dict],
@@ -225,15 +224,15 @@ def ai_workflow_chat(
         # --- State Machine Logic ---
 
         if state == STATE_IDLE:
-            # Check prerequisites first
+            # Check prerequisites first within the state logic as well
             if not (hf_profile and hf_token):
-                history = add_bot_message(history, "Please log in to Hugging Face first.")
+                history = add_bot_message(history, "Workflow paused: Please log in to Hugging Face first.")
                 yield history, repo_id, state, updated_preview, updated_run, updated_build, attempts, app_desc, repo_name, generated_code
-                return
+                return # Stop workflow until login
             if not (gemini_api_key and gemini_model):
-                 history = add_bot_message(history, "Please enter your API key and select a Gemini model.")
+                 history = add_bot_message(history, "Workflow paused: Please enter your API key and select a Gemini model.")
                  yield history, repo_id, state, updated_preview, updated_run, updated_build, attempts, app_desc, repo_name, generated_code
-                 return
+                 return # Stop workflow until API key/model set
 
             # Look for commands
             reset_match = "reset" in message.lower()
@@ -379,7 +378,7 @@ Return **only** the python code block for app.py. Do not include any extra text,
 
             reqs_content = "\n".join(reqs_list) + "\n"
 
-            history = add_bot_message(history, "✅ `requirements.txt` generated. Click 'Send' to upload.")
+            history = add_bot_message(history, "✅ `requirements.txt` generated. Click 'Send' to generate README.")
             state = STATE_UPLOADING_REQUIREMENTS
             generated_code = reqs_content
             yield history, repo_id, state, updated_preview, updated_run, updated_build, attempts, app_desc, repo_name, generated_code
@@ -546,7 +545,6 @@ Return **only** the python code block for app.py. Do not include any extra text,
                   yield history, None, STATE_IDLE, updated_preview, updated_run, updated_build, 0, None, None, None
 
         elif state == STATE_COMPLETE:
-             # Workflow finished. Stay in this state until reset.
              pass
 
 
@@ -613,7 +611,6 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 inputs=[gemini_key, gemini_model],
                 outputs=[gemini_status]
             )
-            # These should trigger configure_gemini whenever the relevant state changes
             gemini_key.change(
                 configure_gemini,
                 inputs=[gemini_key, gemini_model],
@@ -625,7 +622,6 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 outputs=[gemini_status]
             )
 
-
             gr.Markdown("## Space SDK")
             sdk_selector = gr.Radio(choices=["gradio","streamlit"], value="gradio", label="Template SDK", interactive=True)
             sdk_selector.change(lambda s: s, inputs=sdk_selector, outputs=sdk_state)
@@ -634,6 +630,11 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             status_text = gr.Textbox(label="Current State", value=STATE_IDLE, interactive=False)
             repo_id_text = gr.Textbox(label="Current Space ID", value="None", interactive=False)
 
+            # --- Added Debugging Indicator ---
+            gr.Markdown("## Prerequisite Status")
+            prereq_status = gr.Markdown("Checking...")
+
+
         # Main content
         with gr.Column(scale=3):
             chatbot    = gr.Chatbot(type='messages', label="AI Workflow Chat")
@@ -641,44 +642,65 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             send_btn   = gr.Button("Send", interactive=False)
 
             # Logic to enable send button only when logged in and API key is set
-            # Modified function signature to accept *args and **kwargs
             def update_send_button_state(profile: gr.OAuthProfile | None, token: gr.OAuthToken | None, key: str | None, model: str | None, *args, **kwargs):
                 """
                 Gradio sometimes passes extra positional/keyword args (old_value, event, etc.),
                 so we absorb them with *args/**kwargs.
+                Returns Gradio.update for the button's interactive state and a status string.
                 """
                 is_logged_in    = profile is not None and token is not None
                 is_gemini_ready = key     is not None and model is not None
-                return gr.update(interactive=is_logged_in and is_gemini_ready)
+
+                # Print states for debugging
+                print(f"update_send_button_state called:")
+                print(f"  Profile: {profile.username if profile else None}")
+                print(f"  Token: {'Set' if token else 'None'}")
+                print(f"  API Key: {'Set' if key else 'None'}")
+                print(f"  Model: {model}")
+                print(f"  Logged in: {is_logged_in}, Gemini Ready: {is_gemini_ready}")
+
+                status_str = ""
+                if is_logged_in and is_gemini_ready:
+                    status_str = "✅ Ready to send commands."
+                elif not is_logged_in:
+                    status_str = "⚠️ Please log in to Hugging Face."
+                elif not is_gemini_ready:
+                     if not key: status_str += "⚠️ Gemini API key not set. "
+                     if not model: status_str += "⚠️ Gemini model not selected."
+                     status_str = status_str.strip()
+
+
+                return gr.update(interactive=is_logged_in and is_gemini_ready), status_str
+
 
             # Bind update_send_button_state to the *change* of the relevant state variables
             # This ensures it's called with the correct 4 state values plus any extras
             hf_profile.change(
                  update_send_button_state,
                  inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                 outputs=[send_btn]
+                 outputs=[send_btn, prereq_status] # Update button and status indicator
             )
             hf_token.change(
                  update_send_button_state,
                  inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                 outputs=[send_btn]
+                 outputs=[send_btn, prereq_status]
             )
             gemini_key.change(
                  update_send_button_state,
                  inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                 outputs=[send_btn]
+                 outputs=[send_btn, prereq_status]
             )
             gemini_model.change(
                  update_send_button_state,
                  inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                 outputs=[send_btn]
+                 outputs=[send_btn, prereq_status]
             )
 
             # Keep the initial load trigger as well
             ai_builder_tab.load(
                 update_send_button_state,
                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                outputs=[send_btn]
+                outputs=[send_btn, prereq_status] # Update button and status indicator
             )