Update app.py

app.py

@@ -22,6 +22,8 @@ def show_profile(profile: gr.OAuthProfile | None) -> str:
     """Displays the logged-in Hugging Face profile username."""
     if profile is None:
         return "*Not logged in.*"
+    # More robust check: ensure token is also present for "logged in" status for actions
+    # However, the LoginButton itself updates profile/token state. Let's rely on the State check.
     return f"✅ Logged in as **{profile.username}**"
 
 def list_private_models(
@@ -303,8 +305,10 @@ def ai_workflow_chat(
         if state == STATE_IDLE:
             # Check prerequisites before starting any workflow actions
             # Use the passed-in state variable values for checks
+            # Note: Button interactivity should prevent reaching here if HF login or Gemini isn't ready
+            # But this check is a safeguard.
             if not (hf_profile and hf_token):
-                history = add_bot_message(history, "Workflow paused: Please log in to Hugging Face first.")
+                history = add_bot_message(history, "Workflow paused: Hugging Face login is required.")
                 # Yield updated history and current state, then exit for this click
                 yield (history, repo_id, state, updated_preview, updated_run, updated_build,
                        attempts, app_desc, repo_name, generated_code, use_grounding)
@@ -312,12 +316,13 @@ def ai_workflow_chat(
 
             if not (gemini_api_key and gemini_model):
                  # The button is *enabled* if HF is logged in, but we still need Gemini for the workflow
-                 history = add_bot_message(history, "Workflow cannot start: Please enter your API key and select a Gemini model.")
+                 history = add_bot_message(history, "Workflow cannot start: Google AI Studio / Gemini configuration is required.")
                  # Yield updated history and current state, then exit for this click
                  yield (history, repo_id, state, updated_preview, updated_run, updated_build,
                         attempts, app_desc, repo_name, generated_code, use_grounding)
                  return # Exit the generator for this click
 
+
             # Look for specific commands in the user's message
             reset_match = "reset" in message.lower()
             # Capture app description AND repo name using regex
@@ -525,7 +530,6 @@ Return **only** the python code block for `app.py`. Do not include any extra tex
              reqs_list = ["gradio"] if space_sdk == "gradio" else ["streamlit"]
              # Add essential libraries regardless of description keywords or grounding
              essential_libs = ["google-generativeai", "huggingface_hub"]
-             # Only add if Gemini is actually needed for the app (determined by description or if key is present)
              # If we are here, key and model are available based on STATE_IDLE checks
              reqs_list.extend(essential_libs)
 
@@ -628,7 +632,7 @@ This Space was automatically generated by an AI workflow using Google Gemini and
             history = add_bot_message(history, "✅ `README.md` generated. Click 'Send' to upload.")
             state = STATE_UPLOADING_README # Transition state
             generated_code = readme_content # Store README content
-            # Yield updated state variables and history (pass UI outputs and other states through)
+            # Yield updated state variables and history (pass UI outputs and states through)
             yield (history, repo_id, state, updated_preview, updated_run, updated_build,
                    attempts, app_desc, repo_name, generated_code, use_grounding)
             # No return needed
@@ -655,7 +659,7 @@ This Space was automatically generated by an AI workflow using Google Gemini and
                      history = add_bot_message(history, "✅ Uploaded `README.md`. All files uploaded. Space is now building. Click 'Send' to check build logs.")
                      state = STATE_CHECKING_LOGS_BUILD # Transition to checking build logs
                      generated_code = None # Clear content after use
-                     # Yield updated state variables and history (pass UI outputs and other states through)
+                     # Yield updated state variables and history (pass UI outputs and states through)
                      yield (history, repo_id, state, updated_preview, updated_run, updated_build,
                             attempts, app_desc, repo_name, generated_code, use_grounding)
                      # No return needed
@@ -906,6 +910,8 @@ 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)
 
+            # Removed prereq_status markdown component
+
 
         # Main content area column
         with gr.Column(scale=3):
@@ -925,6 +931,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             # --- Define Event Handlers and Chains AFTER all components are defined ---
 
             # Helper function to update send button interactivity based on prereqs
+            # Defined outside Block but used inside event handlers
             def update_send_button_interactivity(profile, token, api_key, model_name):
                  is_logged_in    = profile is not None and token is not None
                  is_gemini_ready = api_key is not None and model_name is not None
@@ -935,11 +942,13 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
 
             # Handle login button click: Update profile/token state -> Update send button interactivity
             login_btn.click(
+                 # The LoginButton outputs a tuple (OAuthProfile, OAuthToken) on success
                  lambda x: (x[0], x[1]),
                  inputs=[login_btn],
                  outputs=[hf_profile, hf_token] # Update HF State variables
             ).then(
                  update_send_button_interactivity,
+                 # Pass required state variables to determine interactivity
                  inputs=[hf_profile, hf_token, gemini_key, gemini_model],
                  outputs=[send_btn] # Update button interactivity
             )
@@ -951,6 +960,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 configure_gemini, inputs=[gemini_key, gemini_model], outputs=[gemini_status] # Update Gemini status
             ).then(
                 update_send_button_interactivity,
+                # Pass required state variables to determine interactivity
                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
                 outputs=[send_btn] # Update button interactivity
             )
@@ -962,6 +972,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 configure_gemini, inputs=[gemini_key, gemini_model], outputs=[gemini_status] # Update Gemini status
             ).then(
                 update_send_button_interactivity,
+                # Pass required state variables to determine interactivity
                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
                 outputs=[send_btn] # Update button interactivity
             )