Update app.py

app.py

@@ -38,9 +38,6 @@ def get_sdk_version(sdk_choice: str) -> str:
         return "UNKNOWN"
 
 def extract_code(text: str) -> str:
-    """
-    Extract the last ```…``` block. If none, return whole text.
-    """
     blocks = re.findall(r"```(?:\w*\n)?([\s\S]*?)```", text)
     return blocks[-1].strip() if blocks else text.strip()
 
@@ -69,7 +66,7 @@ def fetch_logs(repo_id: str, level: str) -> str:
                     continue
     return "\n".join(lines)
 
-# — CORE LOOP (now a generator) — 
+# — CORE LOOP (generator) — 
 
 def handle_user_message(
     history,
@@ -80,12 +77,8 @@ 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:
-        yield (
-            history + [{"role":"assistant","content":"⚠️ Please log in first."}],
-            "", "", "<p>No Space yet.</p>"
-        )
+        yield history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
         return
 
     username   = profile.username
@@ -93,15 +86,12 @@ def handle_user_message(
     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>'
+        f'<iframe src="{preview_url}" 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
+    # Initial “in progress” update
+    yield history + [{"role":"assistant","content":"🚀 Generating and deploying your Space..."}], "Fetching logs…", "Fetching logs…", iframe_html
 
-    # 3) LLM + push loop
     client = genai.Client(api_key=gemini_api_key)
     system_msg = {
         "role":"system",
@@ -119,14 +109,13 @@ def handle_user_message(
     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(
+        resp  = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
             config=cfg
         )
 
-        raw_code = resp.text
-        code = extract_code(raw_code)
+        code = extract_code(resp.text)
         chat.append({"role":"assistant","content":code})
 
         # Write files
@@ -150,7 +139,7 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
         extra     = "streamlit\n" if sdk_choice=="streamlit" else "gradio\n"
         with open(reqs_fn, "w") as f: f.write(base_reqs + extra)
 
-        # Push to HF
+        # Push to Hugging Face
         create_repo(
             repo_id=repo_id,
             token=oauth_token.token,
@@ -167,19 +156,15 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
                 repo_type="space"
             )
 
-        # Fetch logs
+        # Fetch and stream logs & iframe
         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":(
@@ -190,7 +175,7 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
         })
         time.sleep(2)
 
-# — REFRESH LOGS (unchanged) — 
+# — REFRESH LOGS —
 
 def refresh_logs(
     space_suffix: str,
@@ -202,7 +187,7 @@ def refresh_logs(
     repo_id = f"{profile.username}/{profile.username}-{space_suffix}"
     return fetch_logs(repo_id, "build"), fetch_logs(repo_id, "run")
 
-# — BUILD THE UI — 
+# — BUILD THE UI —
 
 with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     gr.Markdown("## Sign in + Auto‑Build Spaces\n\n"
@@ -236,11 +221,11 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
         fn=handle_user_message,
         inputs=[chatbot, sdk_choice, api_key, grounding, space_suffix],
         outputs=[chatbot, build_box, run_box, preview],
-        queue=True,
-        stream=True
+        queue=True
+        # stream=True  ← removed, not supported :contentReference[oaicite:2]{index=2}
     )
 
-    # Manual log refresh (optional)
+    # (Optional) manual refresh
     refresh_btn = gr.Button("Refresh Logs")
     refresh_btn.click(
         fn=refresh_logs,