Update app.py

app.py

@@ -1,30 +1,29 @@
-import gradio as gr
-import json, time
+import re
+import time
+import importlib.metadata
 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
+import gradio as gr
 
-# — USER INFO & MODEL LISTING —
+# — UTILITIES — 
 
-def show_profile(profile: gr.OAuthProfile | None) -> str:
-    if profile is None:
-        return "*Not logged in.*"
-    return f"✅ Logged in as **{profile.username}**"
+def get_sdk_version(sdk_choice: str) -> str:
+    pkg = "gradio" if sdk_choice == "gradio" else "streamlit"
+    try:
+        return importlib.metadata.version(pkg)
+    except importlib.metadata.PackageNotFoundError:
+        return "UNKNOWN"
 
-def list_private_models(
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-) -> str:
-    if profile is None or oauth_token is None:
-        return "Please log in to see your models."
-    models = [
-        f"{m.id} ({'private' if m.private else 'public'})"
-        for m in list_models(author=profile.username, token=oauth_token.token)
-    ]
-    return "No models found." if not models else "Models:\n\n" + "\n - ".join(models)
+def extract_code(text: str) -> str:
+    """
+    Pull out the last ```…``` block, or fall back to the whole text.
+    """
+    blocks = re.findall(r"```(?:\w*\n)?([\s\S]*?)```", text)
+    return blocks[-1].strip() if blocks else text.strip()
 
-# — HELPERS FOR HF SPACE LOGS —
+# — HF SPACE LOGGING — 
 
 def _get_space_jwt(repo_id: str):
     url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
@@ -32,7 +31,7 @@ def _get_space_jwt(repo_id: str):
     hf_raise_for_status(r)
     return r.json()["token"]
 
-def fetch_logs(repo_id: str, level: str):
+def fetch_logs(repo_id: str, level: str) -> str:
     jwt = _get_space_jwt(repo_id)
     logs_url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
     lines = []
@@ -49,27 +48,28 @@ def fetch_logs(repo_id: str, level: str):
                     continue
     return "\n".join(lines)
 
-# — CORE LOOP: send prompt & (iteratively) deploy —
+# — CORE LOOP — 
 
 def handle_user_message(
-    history,               # list of {"role","content"} dicts
+    history,
     sdk_choice: str,
     gemini_api_key: str,
     grounding_enabled: bool,
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
-    if profile is None or oauth_token is None:
+    if not profile or not oauth_token:
         return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
 
     client = genai.Client(api_key=gemini_api_key)
-    chat   = [{
+    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."
         )
-    }] + history
+    }
+    chat = [system_msg] + history
 
     code_fn   = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
     readme_fn = "README.md"
@@ -78,65 +78,57 @@ def handle_user_message(
 
     build_logs = run_logs = ""
     for _ in range(5):
-        tools = []
-        if grounding_enabled:
-            tools.append(Tool(google_search=GoogleSearch()))
-        cfg = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
+        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],
             config=cfg
         )
-        code = resp.text
+
+        raw = resp.text
+        code = extract_code(raw)
         chat.append({"role":"assistant","content":code})
 
-        # write code file
+        # write code
         with open(code_fn, "w") as f:
             f.write(code)
 
-        # write README.md
-        with open(readme_fn, "w") as f:
-            f.write(f"""---
+        # write dynamic README
+        sdk_version = get_sdk_version(sdk_choice)
+        readme = f"""---
 title: Wuhp Auto Space
 emoji: 🐢
 colorFrom: red
 colorTo: pink
 sdk: {sdk_choice}
-sdk_version: 1.44.1
+sdk_version: {sdk_version}
 app_file: {code_fn}
 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.txt
+        # write requirements
         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)
 
-        # deploy all three files
-        create_repo(
-            repo_id=repo_id,
-            token=oauth_token.token,
-            exist_ok=True,
-            repo_type="space",
-            space_sdk=sdk_choice
-        )
+        # push to HF
+        create_repo(repo_id=repo_id, token=oauth_token.token,
+                    exist_ok=True, repo_type="space", space_sdk=sdk_choice)
         for fn in (code_fn, readme_fn, reqs_fn):
-            upload_file(
-                path_or_fileobj=fn,
-                path_in_repo=fn,
-                repo_id=repo_id,
-                token=oauth_token.token,
-                repo_type="space"
-            )
+            upload_file(path_or_fileobj=fn, path_in_repo=fn,
+                        repo_id=repo_id, token=oauth_token.token,
+                        repo_type="space")
 
         build_logs = fetch_logs(repo_id, "build")
         run_logs   = fetch_logs(repo_id, "run")
-
         if "ERROR" not in build_logs.upper() and "ERROR" not in run_logs.upper():
             break
 
@@ -154,29 +146,23 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
     iframe   = f'<iframe src="https://huggingface.co/spaces/{repo_id}" width="100%" height="500px"></iframe>'
     return messages, build_logs, run_logs, iframe
 
-# — 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"
-                "1. Sign in\n2. Enter your prompt\n3. Watch code, README, requirements, logs, and preview\n\n---")
-
-    # LOGIN & MODEL LISTING
+    gr.Markdown("## Sign in + Auto‑Build Spaces\n\n..."
+                )
     login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
     models_md = gr.Markdown()
-
-    # Implicitly injects profile (and token) when calling these
-    demo.load(show_profile,        inputs=None, outputs=status_md)
+    demo.load(show_profile, inputs=None, outputs=status_md)
     demo.load(list_private_models, inputs=None, outputs=models_md)
     login_btn.click(show_profile,        inputs=None, outputs=status_md)
     login_btn.click(list_private_models, inputs=None, outputs=models_md)
 
-    # CONTROLS
     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)
 
-    # CHAT + OUTPUTS
     chatbot   = gr.Chatbot(type="messages")
     user_in   = gr.Textbox(placeholder="Your prompt…", label="Prompt")
     send_btn  = gr.Button("Send")
@@ -185,12 +171,17 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     run_box   = gr.Textbox(label="Run logs",   lines=5, interactive=False)
     preview   = gr.HTML("<p>No Space yet.</p>")
 
-    # Only user inputs here—profile/token are auto‑injected
     send_btn.click(
         fn=handle_user_message,
         inputs=[chatbot, sdk_choice, api_key, grounding],
         outputs=[chatbot, build_box, run_box, preview]
     )
 
-if __name__=="__main__":
-    demo.launch()
+    # — New “Refresh Logs” control for manual edits —
+    refresh_btn = gr.Button("Refresh Logs")
+    def _refresh(profile, token):
+        repo = f"{profile.username}/{profile.username}-auto-space"
+        return fetch_logs(repo, "build"), fetch_logs(repo, "run")
+    refresh_btn.click(_refresh, inputs=[status_md, models_md], outputs=[build_box, run_box])
+
+    demo.launch(server_name="0.0.0.0", server_port=7860)