Update app.py

app.py

@@ -168,7 +168,7 @@ def add_bot_message(history: list[dict], bot_message: str) -> list[dict]:
 
 def ai_workflow_chat(
     message: str,
-    history: list[dict], # Chatbot history format changed to list[dict]
+    history: list[dict],
     hf_profile: gr.OAuthProfile | None,
     hf_token:   gr.OAuthToken   | None,
     gemini_api_key: str         | None,
@@ -184,16 +184,16 @@ def ai_workflow_chat(
     repo_name_state: str | None,
     generated_code_state: str | None,
 ) -> tuple[
-    list[dict],             # history
-    str | None,             # repo_id
-    str,                    # workflow_state
-    str,                    # preview_html
-    str,                    # container_logs
-    str,                    # build_logs
-    int,                    # debug_attempts_state
-    str | None,             # app_description_state
-    str | None,             # repo_name_state
-    str | None,             # generated_code_state
+    list[dict],
+    str | None,
+    str,
+    str,
+    str,
+    str,
+    int,
+    str | None,
+    str | None,
+    str | None,
 ]:
     # Unpack state variables
     repo_id = repo_id_state
@@ -214,7 +214,7 @@ def ai_workflow_chat(
          user_message_entry["name"] = hf_profile.username
     history.append(user_message_entry)
 
-    # Yield immediately to show user message - this is the first yield
+    # Yield immediately to show user message
     yield history, repo_id, state, updated_preview, updated_run, updated_build, attempts, app_desc, repo_name, generated_code
 
 
@@ -272,7 +272,7 @@ def ai_workflow_chat(
 
         if state == STATE_AWAITING_REPO_NAME:
             new_repo_name = message.strip()
-            if not new_repo_name or re.search(r'[^a-zA-Z0-9_-]', new_repo_name): # More robust validation
+            if not new_repo_name or re.search(r'[^a-zA-Z0-9_-]', new_repo_name):
                  history = add_bot_message(history, "Invalid name. Please provide a single word/slug for the Space name (letters, numbers, underscores, hyphens only).")
                  yield history, repo_id, state, updated_preview, updated_run, updated_build, attempts, app_desc, repo_name, generated_code
             else:
@@ -364,15 +364,15 @@ Return **only** the python code block for app.py. Do not include any extra text,
             if "requests" in str(app_desc).lower():
                  reqs_list.append("requests")
             reqs_list.append("huggingface_hub")
-            if "image" in str(app_desc).lower() or "upload" in str(app_desc).lower() or "blur" in str(app_desc).lower() or "vision" in str(app_desc).lower(): # Added vision
+            if "image" in str(app_desc).lower() or "upload" in str(app_desc).lower() or "blur" in str(app_desc).lower() or "vision" in str(app_desc).lower():
                 reqs_list.append("Pillow")
             if "numpy" in str(app_desc).lower(): reqs_list.append("numpy")
             if "pandas" in str(app_desc).lower(): reqs_list.append("pandas")
-            if "scikit-image" in str(app_desc).lower() or "skimage" in str(app_desc).lower() or "cv2" in str(app_desc).lower() or "opencv-python" in str(app_desc).lower(): # Common image processing deps
-                 reqs_list.append("scikit-image") # skimage
-                 reqs_list.append("opencv-python") # cv2
+            if "scikit-image" in str(app_desc).lower() or "skimage" in str(app_desc).lower() or "cv2" in str(app_desc).lower() or "opencv-python" in str(app_desc).lower():
+                 reqs_list.append("scikit-image")
+                 reqs_list.append("opencv-python")
 
-            reqs_list = list(dict.fromkeys(reqs_list)) # Remove duplicates
+            reqs_list = list(dict.fromkeys(reqs_list))
 
             reqs_content = "\n".join(reqs_list) + "\n"
 
@@ -544,25 +544,15 @@ Return **only** the python code block for app.py. Do not include any extra text,
 
         elif state == STATE_COMPLETE:
              # Workflow finished. Stay in this state until reset.
-             # The message is set in the state that transitioned to COMPLETE.
-             pass # No further action needed in this state
+             pass
 
 
     except Exception as e:
-        # Catch-all for unexpected exceptions in any state
         error_message = f"Workflow step failed unexpectedly ({state}): {e}. Click 'Send' to re-attempt this step or 'reset'."
         history = add_bot_message(history, error_message)
         print(f"Critical Error in state {state}: {e}")
-        # Transition to idle state on unexpected errors
         yield history, None, STATE_IDLE, updated_preview, updated_run, updated_build, 0, None, None, None
-        return # End generator
-
-
-    # Note: Each state block should end with a yield to update the UI/state.
-    # If a state block finishes without yielding, the generator will stop until the next call.
-    # By yielding the state variables *after* adding the user message, and then yielding
-    # again after state transitions/actions *within* the state logic, we ensure updates.
-    # The final yield after the try/except is removed as yields should happen within the state logic.
+        return
 
 
 # --- Build the Gradio UI ---
@@ -579,7 +569,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
     debug_attempts = gr.State(0)
     app_description = gr.State(None)
     repo_name_state = gr.State(None)
-    generated_code_state = gr.State(None) # Reused
+    generated_code_state = gr.State(None)
 
     with gr.Row():
         # Sidebar
@@ -592,14 +582,17 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             # Removed _preprocess=False
             ai_builder_tab.load(show_profile, outputs=login_status)
             # Update status on login click
+            # Removed _preprocess=False
             login_btn.click(show_profile, outputs=login_status)
             # Store profile and token in state on login click
+            # Removed _preprocess=False
             login_btn.click(lambda x: x, inputs=[login_btn], outputs=[hf_profile, hf_token])
 
             gr.Markdown("## Google AI Studio API Key")
             gemini_input  = gr.Textbox(label="API Key", type="password", interactive=True)
             gemini_status = gr.Markdown("")
 
+            # Removed _preprocess=False
             gemini_input.change(lambda k: k, inputs=gemini_input, outputs=gemini_key)
 
             gr.Markdown("## Gemini Model")
@@ -613,6 +606,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 label="Select model",
                 interactive=True
             )
+            # Removed _preprocess=False
             model_selector.change(lambda m: m, inputs=model_selector, outputs=gemini_model)
 
             # Configure Gemini status on load and when key/model changes
@@ -622,23 +616,23 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 inputs=[gemini_key, gemini_model],
                 outputs=[gemini_status]
             )
-            # Keep _preprocess=False on change events as they have implicit outputs (the new value)
+            # Removed _preprocess=False
             gemini_input.change(
                 configure_gemini,
                 inputs=[gemini_key, gemini_model],
-                outputs=[gemini_status],
-                 _preprocess=False
+                outputs=[gemini_status]
             )
+            # Removed _preprocess=False
             model_selector.change(
                 configure_gemini,
                 inputs=[gemini_key, gemini_model],
-                outputs=[gemini_status],
-                 _preprocess=False
+                outputs=[gemini_status]
             )
 
 
             gr.Markdown("## Space SDK")
             sdk_selector = gr.Radio(choices=["gradio","streamlit"], value="gradio", label="Template SDK", interactive=True)
+            # Removed _preprocess=False
             sdk_selector.change(lambda s: s, inputs=sdk_selector, outputs=sdk_state)
 
             gr.Markdown("## Workflow Status")
@@ -663,24 +657,23 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
                 outputs=[send_btn]
             )
-            # Keep _preprocess=False on click/change events with explicit inputs
+            # Removed _preprocess=False
             login_btn.click(
                 update_send_button_state,
                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                outputs=[send_btn],
-                _preprocess=False
+                outputs=[send_btn]
             )
+            # Removed _preprocess=False
             gemini_input.change(
                 update_send_button_state,
                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                outputs=[send_btn],
-                 _preprocess=False
+                outputs=[send_btn]
             )
+            # Removed _preprocess=False
             model_selector.change(
                 update_send_button_state,
                 inputs=[hf_profile, hf_token, gemini_key, gemini_model],
-                outputs=[send_btn],
-                 _preprocess=False
+                outputs=[send_btn]
             )
 
             iframe    = gr.HTML("<p>No Space created yet.</p>")
@@ -688,6 +681,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
             run_txt   = gr.Textbox(label="Container Logs", lines=10, interactive=False, value="")
 
             # The main event handler for the Send button
+            # Removed _preprocess=False (it wasn't on this one anyway)
             send_btn.click(
                 ai_workflow_chat,
                 inputs=[
@@ -704,14 +698,16 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                     iframe, run_txt, build_txt,
                     debug_attempts, app_description, repo_name_state, generated_code_state
                 ]
-            ).success( # Clear input after successful send
+            ).success(
                  lambda: gr.update(value=""),
                  inputs=None,
                  outputs=user_input
             )
 
             # Link state variables to UI status displays (reactive updates)
+            # Removed _preprocess=False
             workflow.change(lambda s: s, inputs=workflow, outputs=status_text)
+            # Removed _preprocess=False
             repo_id.change(lambda r: r if r else "None", inputs=repo_id, outputs=repo_id_text)
 
 
@@ -719,7 +715,6 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
     def greet():
         return [{"role": "assistant", "content": "Welcome! Please log in to Hugging Face and provide your Google AI Studio API key to start building Spaces. Once ready, type 'generate me a gradio app called myapp' or 'create' to begin."}]
 
-    # THIS CALL IS INSIDE the with gr.Blocks() block
     ai_builder_tab.load(greet, outputs=chatbot)