Update app.py

app.py

@@ -545,7 +545,7 @@ 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:
-             pass
+             pass # No further action needed in this state
 
 
     except Exception as e:
@@ -553,7 +553,7 @@ Return **only** the python code block for app.py. Do not include any extra text,
         history = add_bot_message(history, error_message)
         print(f"Critical Error in state {state}: {e}")
         yield history, None, STATE_IDLE, updated_preview, updated_run, updated_build, 0, None, None, None
-        return
+        # Removed the 'return' here
 
 
 # --- Build the Gradio UI ---
@@ -646,7 +646,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                 status_str = ""
                 if is_logged_in and is_gemini_ready:
                     status_str = "✅ Ready to send commands."
-                else: # Consolidated status checks
+                else:
                      status_parts = []
                      if not is_logged_in:
                          status_parts.append("⚠️ Not logged in to Hugging Face.")
@@ -655,7 +655,7 @@ with gr.Blocks(title="AI-Powered HF Space App Builder") as ai_builder_tab:
                      if not model:
                           status_parts.append("⚠️ Gemini model not selected.")
                      status_str = " ".join(status_parts)
-                     if not status_str: # Should be ready if status_parts is empty
+                     if not status_str:
                           status_str = "✅ Ready to send commands."