Update app.py

app.py

@@ -1,10 +1,11 @@
 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
+from huggingface_hub import create_repo, upload_file, list_models, constants, Repository
 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
@@ -62,13 +63,36 @@ 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(
@@ -83,26 +107,41 @@ 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}"
+    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
 
+    # 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 "
-            f"{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 {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"
 
@@ -118,7 +157,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
@@ -145,7 +184,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)
 
@@ -166,31 +205,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
+        # Fetch logs and break if no errors
         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\n"
-                f"Run logs:\n{run}\n\n"
-                "Please fix the code."
+                f"Build logs:\n{build}\n\nRun logs:\n{run}\n\nPlease 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}"
+    # 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}"
     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."
+    placeholder = "✅ Space created! Click “Refresh Logs” to pull logs."
     return messages, placeholder, placeholder, iframe
 
 # — REFRESH LOGS — 
@@ -201,11 +239,12 @@ def refresh_logs(
     oauth_token: gr.OAuthToken | None
 ):
     if profile is None or oauth_token is None:
-        return "⚠️ Please log in.", "⚠️ Please log in."
+        return "⚠️ Please log in.", ""
     repo_id = f"{profile.username}/{profile.username}-{space_suffix}"
-    return fetch_logs(repo_id, "build"), fetch_logs(repo_id, "run")
+    combined = fetch_all_logs(repo_id)
+    return combined, ""
 
-# — 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"
@@ -221,19 +260,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=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=10, interactive=False)
+    run_box   = gr.Textbox(label="Run logs", lines=1, interactive=False)
+    preview   = gr.HTML("<p>No Space yet.</p>")
 
     send_btn.click(
         fn=handle_user_message,
@@ -245,7 +284,7 @@ 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],
+        inputs=[space_suffix, status_md, api_key],
         outputs=[build_box, run_box]
     )