Update app.py

app.py

@@ -12,6 +12,10 @@ from google.generativeai import types # Import types for configuration and tools
 from huggingface_hub import create_repo, list_models, upload_file, constants
 from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status
 
+# Add debugging print to check environment variable immediately
+print(f"Attempting to read GOOGLE_API_KEY from environment: {os.environ.get('GOOGLE_API_KEY')}")
+
+
 # --- Helper functions for Hugging Face integration ---
 
 def show_profile(profile: gr.OAuthProfile | None) -> str:
@@ -304,7 +308,7 @@ def ai_workflow_chat(
                 history = add_bot_message(history, "Workflow reset.")
                 # Yield updated history and reset state variables to their initial values
                 yield (history, None, STATE_IDLE, "<p>No Space created yet.</p>", "", "", 0,
-                       None, None, None, use_grounding) # Include use_grounding
+                       None, None, None, False) # Reset use_grounding to default False
                 # No return needed after yield in this generator pattern; execution for this click ends here.
 
             elif generate_match:
@@ -417,6 +421,8 @@ Return **only** the python code block for `app.py`. Do not include any extra tex
 """
              try:
                  history = add_bot_message(history, f"🧠 Generating `{prompt_desc}` `{space_sdk}` app (`app.py`) code with Gemini...")
+                 if use_grounding:
+                     history = add_bot_message(history, "(Using Grounding with Google Search)")
                  # Yield to show message before the potentially time-consuming API call
                  yield (history, repo_id, state, updated_preview, updated_run, updated_build,
                         attempts, app_desc, repo_name, generated_code, use_grounding) # Include use_grounding
@@ -513,7 +519,7 @@ Return **only** the python code block for `app.py`. Do not include any extra tex
              # Yield updated state variables and history
              yield (history, repo_id, state, updated_preview, updated_run, updated_build,
                     attempts, app_desc, repo_name, generated_code, use_grounding) # Include use_grounding
-             # No return needed
+                     # No return needed
 
 
         elif state == STATE_UPLOADING_REQUIREMENTS:
@@ -687,6 +693,8 @@ This Space was automatically generated by an AI workflow using Google Gemini and
 
         elif state == STATE_DEBUGGING_CODE:
              history = add_bot_message(history, f"🧠 Calling Gemini to generate fix based on logs...")
+             if use_grounding:
+                  history = add_bot_message(history, "(Using Grounding with Google Search)")
              # Yield message before Gemini API call
              yield (history, repo_id, state, updated_preview, updated_run, updated_build,
                     attempts, app_desc, repo_name, generated_code, use_grounding) # Include use_grounding
@@ -696,6 +704,8 @@ This Space was automatically generated by an AI workflow using Google Gemini and
 You are debugging a {space_sdk} Space. The goal is to fix the code in `app.py` based on the container logs provided.
 
 Here are the container logs:
+Use code with caution.
+Python
 {updated_run}
 Generate the *complete, fixed* content for `app.py` based on these logs.
 Return **only** the python code block for app.py. Do not include any extra text, explanations, or markdown outside the code block.
@@ -790,7 +800,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
     # Gradio State variables - these persist their values across user interactions (clicks)
     hf_profile   = gr.State(None)
     hf_token     = gr.State(None)
-    # FIX: Initialize gemini_key state from env var on load
+    # FIX: Initialize gemini_key state from env var on load for robustness
     gemini_key   = gr.State(os.environ.get("GOOGLE_API_KEY"))
     gemini_model = gr.State("gemini-1.5-flash") # Default selected model
     repo_id      = gr.State(None) # Stores the ID of the created Space
@@ -811,11 +821,36 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             # Hugging Face Login Button
             login_btn    = gr.LoginButton(variant="huggingface")
 
-            # Initial load event to check login status (if cached)
-            # This also ensures the gemini_key state is potentially read from env var
-            ai_builder_tab.load(show_profile, outputs=login_status)
+            # Initial load event to check login status (if cached) and explicitly set gemini_key state
+            # Chain events: Show profile -> Set Gemini Key state -> Configure Gemini -> Update Send Button
+            def initial_setup_on_load(profile, token):
+                 # This function explicitly sets the gemini_key state from the env var
+                 # This runs *after* the profile is potentially loaded, ensuring gemini_key is set
+                 # before configure_gemini and update_send_button_state read it from state.
+                 # It needs to return the values for the next step in the chain.
+                 return profile, token, os.environ.get("GOOGLE_API_KEY")
+
+            ai_builder_tab.load(
+                 initial_setup_on_load,
+                 inputs=[hf_profile, hf_token], # Pass current state values
+                 outputs=[hf_profile, hf_token, gemini_key] # Update state values
+            ).then( # Chain after initial state setup
+                configure_gemini,
+                inputs=[gemini_key, gemini_model],
+                outputs=[gemini_status] # Update Gemini status text
+            ).then( # Chain after config status
+                update_send_button_state,
+                inputs=[hf_profile, hf_token, gemini_key, gemini_model],
+                outputs=[send_btn, prereq_status] # Update button interactivity and status text
+            )
+
+
             # Update status display when login button reports success
-            login_btn.click(show_profile, outputs=login_status)
+            login_btn.click(show_profile, outputs=login_status).then(
+                 update_send_button_state, # Recalculate status and button state
+                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
+                 outputs=[send_btn, prereq_status]
+            )
 
             gr.Markdown("## Google AI Studio / Gemini")
             # Textbox for Gemini API key. Read from environment variable if available.
@@ -840,14 +875,25 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 interactive=True
             )
 
-            # Configure Gemini status on initial load (if API key env var is set)
-            # This uses the *initial* value of gemini_key state
-            ai_builder_tab.load(
-                configure_gemini,
-                inputs=[gemini_key, gemini_model],
-                outputs=[gemini_status]
+            # Gemini Key Input change: update key state -> configure Gemini -> update send button
+            gemini_input.change(
+                lambda k: k, inputs=[gemini_input], outputs=[gemini_key] # Update gemini_key state
+            ).then(
+                configure_gemini, inputs=[gemini_key, gemini_model], outputs=[gemini_status] # Update Gemini status
+            ).then(
+                update_send_button_state, inputs=[hf_profile, hf_token, gemini_key, gemini_model], outputs=[send_btn, prereq_status] # Update button/prereq status
+            )
+
+            # Gemini Model Selector change: update model state -> configure Gemini -> update send button
+            model_selector.change(
+                lambda m: m, inputs=[model_selector], outputs=[gemini_model] # Update gemini_model state
+            ).then(
+                configure_gemini, inputs=[gemini_key, gemini_model], outputs=[gemini_status] # Update Gemini status
+            ).then(
+                update_send_button_state, inputs=[hf_profile, hf_token, gemini_key, gemini_model], outputs=[send_btn, prereq_status] # Update button/prereq status
             )
 
+
             # New checkbox for optional grounding
             use_grounding_checkbox = gr.Checkbox(
                 label="Enable Grounding with Google Search",
@@ -898,6 +944,8 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 **kwargs
             ):
                 """Determines if the send button should be active and updates status text."""
+                # Add debug prints to see what values this function is getting
+                print(f"update_send_button_state - profile: {profile is not None}, token: {token is not None}, key: {key is not None}, model: {model is not None}")
                 is_logged_in    = profile is not None and token is not None
                 is_gemini_ready = key     is not None and model is not None # Check if key and model are set
 
@@ -922,70 +970,6 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 # gr.update is used to dynamically change a component's properties
                 return gr.update(interactive=is_ready), status_str
 
-            # --- Implement Chained Events for Prerequisites ---
-            # Gradio's `.then()` allows chaining events: Action A happens, then Action B happens.
-
-            # 1. Login Button: When clicked and successful, update profile/token state,
-            # THEN update send button state based on all prereqs.
-            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 these State variables
-            ).then( # Chain the next action after state is updated
-                 update_send_button_state,
-                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                 outputs=[send_btn, prereq_status] # Update button interactivity and status text
-            )
-
-            # 2. Gemini Key Input: When text changes, update key state,
-            # THEN configure Gemini status, THEN update send button state.
-            # The Textbox 'change' event passes the new value as its input
-            gemini_input.change(
-                lambda k: k, # Simple function to pass the new value to the state variable
-                inputs=[gemini_input],
-                outputs=[gemini_key] # Update gemini_key state variable
-            ).then( # Chain configure_gemini after key state is updated
-                configure_gemini,
-                inputs=[gemini_key, gemini_model],
-                outputs=[gemini_status] # Update Gemini status text
-            ).then( # Chain update_send_button_state after config status is updated
-                update_send_button_state,
-                inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                outputs=[send_btn, prereq_status] # Update button interactivity and status text
-            )
-
-            # 3. Gemini Model Selector: When selection changes, update model state,
-            # THEN configure Gemini status, THEN update send button state.
-            # The Radio 'change' event passes the new value as its input
-            model_selector.change(
-                lambda m: m, # Simple function to pass the new value to the state variable
-                inputs=[model_selector],
-                outputs=[gemini_model] # Update gemini_model state variable
-            ).then( # Chain configure_gemini after model state is updated
-                configure_gemini,
-                inputs=[gemini_key, gemini_model],
-                outputs=[gemini_status] # Update Gemini status text
-            ).then( # Chain update_send_button_state after config status is updated
-                update_send_button_state,
-                inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                outputs=[send_btn, prereq_status] # Update button interactivity and status text
-            )
-
-            # 4. Initial Load: On page load, check prereqs and update send button/status.
-            # This accounts for cached logins or environment variables set before launch.
-            ai_builder_tab.load(
-                update_send_button_state,
-                inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                outputs=[send_btn, prereq_status] # Update button interactivity and status text
-            )
-
-            # UI elements to display the Space preview iframe and build/run logs
-            iframe    = gr.HTML("<p>No Space created yet.</p>") # HTML element for the Space iframe
-            # Textboxes for logs, interactive=False means user can't type here
-            build_txt = gr.Textbox(label="Build Logs", lines=10, interactive=False, value="", max_lines=20) # Set max_lines for scrollability
-            run_txt   = gr.Textbox(label="Container Logs", lines=10, interactive=False, value="", max_lines=20) # Set max_lines for scrollability
-
             # The main event handler for the Send button
             # This .click() event triggers the ai_workflow_chat generator function
             send_btn.click(
@@ -1047,4 +1031,4 @@ if __name__ == "__main__":
     os.makedirs(os.environ["GRADIO_TEMP_DIR"], exist_ok=True) # Ensure the directory exists
 
     # Launch the Gradio UI
-    ai_builder_tab.launch()
+    # The Gradio launch call blo