Update app.py

app.py

@@ -69,7 +69,7 @@ def fetch_logs(repo_id: str, level: str) -> str:
                     continue
     return "\n".join(lines)
 
-# — CORE LOOP — 
+# — CORE LOOP (now a generator) — 
 
 def handle_user_message(
     history,
@@ -80,18 +80,28 @@ def handle_user_message(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
+    # 1) Require login
     if profile is None or oauth_token is None:
-        return (
+        yield (
             history + [{"role":"assistant","content":"⚠️ Please log in first."}],
-            "",
-            "",
-            "<p>No Space yet.</p>"
+            "", "", "<p>No Space yet.</p>"
         )
+        return
 
-    username = profile.username
-    repo_name = f"{username}-{space_suffix}"
-    repo_id   = f"{username}/{repo_name}"
+    username   = profile.username
+    repo_name  = f"{username}-{space_suffix}"
+    repo_id    = f"{username}/{repo_name}"
+    preview_url = f"https://{username}-{space_suffix}.hf.space"
+    iframe_html = (
+        f'<iframe src="{preview_url}" '
+        f'width="100%" height="500px" frameborder="0"></iframe>'
+    )
+
+    # 2) Initial “in progress” state
+    msg_init = history + [{"role":"assistant","content":"🚀 Generating and deploying your Space..."}]
+    yield msg_init, "Fetching logs…", "Fetching logs…", iframe_html
 
+    # 3) LLM + push loop
     client = genai.Client(api_key=gemini_api_key)
     system_msg = {
         "role":"system",
@@ -106,11 +116,9 @@ def handle_user_message(
     readme_fn = "README.md"
     reqs_fn   = "requirements.txt"
 
-    # Try up to 5 times to generate & push working code
     for _ in range(5):
         tools = [Tool(google_search=GoogleSearch())] if grounding_enabled else []
         cfg   = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
-
         resp = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
@@ -121,11 +129,8 @@ def handle_user_message(
         code = extract_code(raw_code)
         chat.append({"role":"assistant","content":code})
 
-        # Write code file
-        with open(code_fn, "w") as f:
-            f.write(code)
-
-        # Write README with dynamic SDK version
+        # Write files
+        with open(code_fn, "w") as f: f.write(code)
         sdk_version = get_sdk_version(sdk_choice)
         readme = f"""---
 title: Wuhp Auto Space
@@ -140,16 +145,12 @@ pinned: false
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 """
-        with open(readme_fn, "w") as f:
-            f.write(readme)
-
-        # Write requirements
+        with open(readme_fn, "w") as f: f.write(readme)
         base_reqs = "pandas\n"
         extra     = "streamlit\n" if sdk_choice=="streamlit" else "gradio\n"
-        with open(reqs_fn, "w") as f:
-            f.write(base_reqs + extra)
+        with open(reqs_fn, "w") as f: f.write(base_reqs + extra)
 
-        # Push to Hugging Face
+        # Push to HF
         create_repo(
             repo_id=repo_id,
             token=oauth_token.token,
@@ -166,34 +167,30 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
                 repo_type="space"
             )
 
-        # If build succeeds without errors, break; otherwise feed logs back to LLM
-        build = fetch_logs(repo_id, "build")
-        run   = fetch_logs(repo_id, "run")
-        if "ERROR" not in build.upper() and "ERROR" not in run.upper():
+        # Fetch logs
+        build_logs = fetch_logs(repo_id, "build")
+        run_logs   = fetch_logs(repo_id, "run")
+
+        # Yield incremental update
+        msgs = [{"role":m["role"], "content":m["content"]} for m in chat if m["role"]!="system"]
+        yield msgs, build_logs, run_logs, iframe_html
+
+        # If successful, stop retrying
+        if "ERROR" not in build_logs.upper() and "ERROR" not in run_logs.upper():
             break
 
+        # Otherwise feed logs back to LLM
         chat.append({
             "role":"user",
             "content":(
-                f"Build logs:\n{build}\n\n"
-                f"Run logs:\n{run}\n\n"
+                f"Build logs:\n{build_logs}\n\n"
+                f"Run logs:\n{run_logs}\n\n"
                 "Please fix the code."
             )
         })
         time.sleep(2)
 
-    # Prepare messages + immediate iframe preview
-    messages = [{"role":m["role"],"content":m["content"]} for m in chat if m["role"]!="system"]
-    preview_url = f"https://huggingface.co/spaces/{repo_id}"
-    iframe = (
-        f'<iframe src="{preview_url}" '
-        f'width="100%" height="500px" frameborder="0"></iframe>'
-    )
-
-    placeholder = "✅ Space created! Click “Refresh Logs” to pull build/run logs."
-    return messages, placeholder, placeholder, iframe
-
-# — REFRESH LOGS — 
+# — REFRESH LOGS (unchanged) — 
 
 def refresh_logs(
     space_suffix: str,
@@ -221,27 +218,29 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     login_btn.click(list_private_models, inputs=None, outputs=models_md)
 
     # SETTINGS
-    sdk_choice  = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK template")
-    api_key     = gr.Textbox(label="Gemini API Key", type="password")
-    grounding   = gr.Checkbox(label="Enable grounding", value=False)
-    space_suffix= gr.Textbox(label="Space suffix", value="auto-space",
-                             info="E.g. 'auto-space', 'auto-space2', etc.")
+    sdk_choice   = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK template")
+    api_key      = gr.Textbox(label="Gemini API Key", type="password")
+    grounding    = gr.Checkbox(label="Enable grounding", value=False)
+    space_suffix = gr.Textbox(label="Space suffix", value="auto-space",
+                              info="E.g. 'auto-space', 'auto-space2', etc.")
 
     # CHAT + OUTPUTS
-    chatbot    = gr.Chatbot(type="messages")
-    user_in    = gr.Textbox(placeholder="Your prompt…", label="Prompt")
-    send_btn   = gr.Button("Send")
-    build_box  = gr.Textbox(label="Build logs", lines=5, interactive=False)
-    run_box    = gr.Textbox(label="Run logs",   lines=5, interactive=False)
-    preview    = gr.HTML("<p>No Space yet.</p>")
+    chatbot   = gr.Chatbot(type="messages")
+    user_in   = gr.Textbox(placeholder="Your prompt…", label="Prompt")
+    send_btn  = gr.Button("Send")
+    build_box = gr.Textbox(label="Build logs", lines=5, interactive=False)
+    run_box   = gr.Textbox(label="Run logs",   lines=5, interactive=False)
+    preview   = gr.HTML("<p>No Space yet.</p>")
 
     send_btn.click(
         fn=handle_user_message,
         inputs=[chatbot, sdk_choice, api_key, grounding, space_suffix],
-        outputs=[chatbot, build_box, run_box, preview]
+        outputs=[chatbot, build_box, run_box, preview],
+        queue=True,
+        stream=True
     )
 
-    # Manual log refresh
+    # Manual log refresh (optional)
     refresh_btn = gr.Button("Refresh Logs")
     refresh_btn.click(
         fn=refresh_logs,