Update app.py

app.py

@@ -1,11 +1,10 @@
 import re
 import json
 import time
-import os
 import importlib.metadata
 
 import gradio as gr
-from huggingface_hub import create_repo, upload_file, list_models, constants, Repository
+from huggingface_hub import create_repo, upload_file, list_models, constants
 from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status
 from google import genai
 from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
@@ -40,9 +39,13 @@ def get_sdk_version(sdk_choice: str) -> str:
 
 def extract_code(text: str) -> str:
     """
-    Extract the last ```…``` block. If none, return whole text.
+    Extract the last 
+…
+ block. If none, return whole text.
     """
-    blocks = re.findall(r"```(?:\w*\n)?([\s\S]*?)```", text)
+    blocks = re.findall(r"
+(?:\w*\n)?([\s\S]*?)
+", text)
     return blocks[-1].strip() if blocks else text.strip()
 
 # — HF SPACE LOGGING — 
@@ -63,36 +66,13 @@ def fetch_logs(repo_id: str, level: str) -> str:
             if raw.startswith(b"data: "):
                 try:
                     ev = json.loads(raw[len(b"data: "):].decode())
-                    ts  = ev.get("timestamp", "")
-                    txt = ev.get("data", "")
+                    ts  = ev.get("timestamp","")
+                    txt = ev.get("data","")
                     lines.append(f"[{ts}] {txt}")
                 except:
                     continue
     return "\n".join(lines)
 
-def fetch_all_logs(repo_id: str) -> str:
-    """Return combined build + run logs."""
-    build = fetch_logs(repo_id, "build")
-    run = fetch_logs(repo_id, "run")
-    return f"--- BUILD ---\n{build}\n\n--- RUN ---\n{run}"
-
-# — SYNC MANUAL EDITS —
-
-def sync_and_read_code(username: str, repo_name: str, token: str, code_filename: str) -> str:
-    """Clone or pull the HF Space repo locally and return the latest code."""
-    local_dir = f"./{repo_name}"
-    repo_url  = f"https://huggingface.co/spaces/{username}/{repo_name}"
-    repo = Repository(
-        local_dir,
-        clone_from=repo_url,
-        use_auth_token=token,
-        repo_type="space"
-    )
-    repo.git_pull()
-    path = os.path.join(local_dir, code_filename)
-    with open(path, "r") as f:
-        return f.read()
-
 # — CORE LOOP — 
 
 def handle_user_message(
@@ -107,41 +87,26 @@ def handle_user_message(
     if profile is None or oauth_token is None:
         return (
             history + [{"role":"assistant","content":"⚠️ Please log in first."}],
-            "", "", "<p>No Space yet.</p>"
+            "",
+            "",
+            "<p>No Space yet.</p>"
         )
 
-    username   = profile.username
-    repo_name  = f"{username}-{space_suffix}"
-    repo_id    = f"{username}/{repo_name}"
-    code_fn    = "app.py" if sdk_choice == "gradio" else "streamlit_app.py"
-
-    # Sync manual edits into history before LLM generation
-    try:
-        latest_code = sync_and_read_code(username, repo_name, oauth_token.token, code_fn)
-        history.append({
-            "role": "assistant",
-            "content": (
-                f"🔄 Synced latest `{code_fn}` from the Space:\n"
-                "```python\n"
-                f"{latest_code}\n"
-                "```"
-            )
-        })
-    except Exception:
-        # If repo doesn't exist yet, skip sync
-        pass
+    username = profile.username
+    repo_name = f"{username}-{space_suffix}"
+    repo_id   = f"{username}/{repo_name}"
 
-    # Build chat context
     client = genai.Client(api_key=gemini_api_key)
     system_msg = {
-        "role": "system",
-        "content": (
-            f"You are an AI assistant writing a HuggingFace Space using the {sdk_choice} SDK. "
-            "After producing code, wait for logs; if errors appear, fix them."
+        "role":"system",
+        "content":(
+            f"You are an AI assistant writing a HuggingFace Space using the "
+            f"{sdk_choice} SDK. After producing code, wait for logs; if errors appear, fix them."
         )
     }
     chat = [system_msg] + history
 
+    code_fn   = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
     readme_fn = "README.md"
     reqs_fn   = "requirements.txt"
 
@@ -157,7 +122,7 @@ def handle_user_message(
         )
 
         raw_code = resp.text
-        code     = extract_code(raw_code)
+        code = extract_code(raw_code)
         chat.append({"role":"assistant","content":code})
 
         # Write code file
@@ -184,7 +149,7 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
 
         # Write requirements
         base_reqs = "pandas\n"
-        extra     = "streamlit\n" if sdk_choice == "streamlit" else "gradio\n"
+        extra     = "streamlit\n" if sdk_choice=="streamlit" else "gradio\n"
         with open(reqs_fn, "w") as f:
             f.write(base_reqs + extra)
 
@@ -205,30 +170,31 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
                 repo_type="space"
             )
 
-        # Fetch logs and break if no errors
+        # 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():
             break
 
-        # Feed errors back to LLM
         chat.append({
             "role":"user",
             "content":(
-                f"Build logs:\n{build}\n\nRun logs:\n{run}\n\nPlease fix the code."
+                f"Build logs:\n{build}\n\n"
+                f"Run logs:\n{run}\n\n"
+                "Please fix the code."
             )
         })
         time.sleep(2)
 
-    # Prepare messages + iframe preview
-    messages = [{"role":m["role"],"content":m["content"]} for m in chat if m["role"] != "system"]
-    preview_url = f"https://huggingface.co/spaces/{username}/{repo_name}"
+    # 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 logs."
+    placeholder = "✅ Space created! Click “Refresh Logs” to pull build/run logs."
     return messages, placeholder, placeholder, iframe
 
 # — REFRESH LOGS — 
@@ -239,12 +205,11 @@ def refresh_logs(
     oauth_token: gr.OAuthToken | None
 ):
     if profile is None or oauth_token is None:
-        return "⚠️ Please log in.", ""
+        return "⚠️ Please log in.", "⚠️ Please log in."
     repo_id = f"{profile.username}/{profile.username}-{space_suffix}"
-    combined = fetch_all_logs(repo_id)
-    return combined, ""
+    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"
@@ -260,19 +225,19 @@ 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=10, interactive=False)
-    run_box   = gr.Textbox(label="Run logs", lines=1, 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,
@@ -284,8 +249,8 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     refresh_btn = gr.Button("Refresh Logs")
     refresh_btn.click(
         fn=refresh_logs,
-        inputs=[space_suffix, status_md, api_key],
+        inputs=[space_suffix],
         outputs=[build_box, run_box]
     )
 
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)