Update app.py

app.py

@@ -8,15 +8,13 @@ from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
 # — USER INFO & MODEL LISTING —
 
 def show_profile(profile: gr.OAuthProfile | None) -> str:
-    if profile is None:
-        return "*Not logged in.*"
-    return f"✅ Logged in as **{profile.username}**"
+    return f"✅ Logged in as **{profile.username}**" if profile else "*Not logged in.*"
 
 def list_private_models(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ) -> str:
-    if profile is None or oauth_token is None:
+    if not (profile and oauth_token):
         return "Please log in to see your models."
     models = [
         f"{m.id} ({'private' if m.private else 'public'})"
@@ -24,7 +22,7 @@ def list_private_models(
     ]
     return "No models found." if not models else "Models:\n\n" + "\n - ".join(models)
 
-# — HELPERS FOR HF SPACE LOGS —
+# — LOG FETCHING —
 
 def _get_space_jwt(repo_id: str):
     url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
@@ -34,33 +32,31 @@ def _get_space_jwt(repo_id: str):
 
 def fetch_logs(repo_id: str, level: str):
     jwt = _get_space_jwt(repo_id)
-    logs_url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
+    url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
     lines = []
-    with get_session().get(logs_url, headers=build_hf_headers(token=jwt), stream=True) as resp:
+    with get_session().get(url, headers=build_hf_headers(token=jwt), stream=True) as resp:
         hf_raise_for_status(resp)
         for raw in resp.iter_lines():
             if raw.startswith(b"data: "):
                 try:
                     ev = json.loads(raw[len(b"data: "):].decode())
-                    ts  = ev.get("timestamp","")
-                    txt = ev.get("data","")
-                    lines.append(f"[{ts}] {txt}")
+                    lines.append(f"[{ev.get('timestamp','')}] {ev.get('data','')}")
                 except:
                     continue
     return "\n".join(lines)
 
-# — CORE LOOP: send prompt & (iteratively) deploy —
+# — HANDLERS —
 
 def handle_user_message(
-    history,               # list of {"role","content"} dicts
+    history,               # list of dicts {"role","content"}
     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:
-        return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
+    if not (profile and 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   = [{
@@ -71,27 +67,32 @@ def handle_user_message(
         )
     }] + history
 
-    code_fn   = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
+    # choose filenames
+    code_fn   = "app.py" if sdk_choice == "gradio" else "streamlit_app.py"
     readme_fn = "README.md"
     reqs_fn   = "requirements.txt"
     repo_id   = f"{profile.username}/{profile.username}-auto-space"
 
     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"])
-
-        resp = client.models.generate_content(
+        # dynamic sdk_version
+        if sdk_choice == "gradio":
+            import gradio as _gr; sdk_version = _gr.__version__
+        else:
+            import streamlit as _st; sdk_version = _st.__version__
+
+        # ask Gemini
+        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
+        code  = resp.text
         chat.append({"role":"assistant","content":code})
 
-        # write code file
+        # write code
         with open(code_fn, "w") as f:
             f.write(code)
 
@@ -103,40 +104,36 @@ 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
+See config reference → https://huggingface.co/docs/hub/spaces-config-reference
 """)
 
-        # write requirements.txt
-        base_reqs = "pandas\n"
-        extra     = "streamlit\n" if sdk_choice=="streamlit" else "gradio\n"
+        # write requirements
+        base = "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
-        )
+            f.write(base + extra)
+
+        # deploy
+        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")
 
+        # fetch logs
         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
 
@@ -150,47 +147,70 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
         })
         time.sleep(2)
 
+    # prepare outputs
     messages = [{"role":m["role"],"content":m["content"]} for m in chat if m["role"]!="system"]
     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 —
-
-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
-    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(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)
+    return messages, build_logs, run_logs, iframe, repo_id
+
+def refresh_build_logs(repo_id: str, profile, oauth_token):
+    if not (profile and oauth_token and repo_id):
+        return "⚠️ Please deploy first."
+    return fetch_logs(repo_id, "build")
+
+def refresh_run_logs(repo_id: str, profile, oauth_token):
+    if not (profile and oauth_token and repo_id):
+        return "⚠️ Please deploy first."
+    return fetch_logs(repo_id, "run")
+
+# — UI —
+
+with gr.Blocks() as demo:
+    gr.Markdown("## HF Space Auto‑Builder\n1. Sign in  2. Prompt  3. Deploy & Debug ►")
+
+    # Login & model list
+    login_btn  = gr.LoginButton("huggingface", size="lg")
+    status_md  = gr.Markdown("*Not logged in.*")
+    models_md  = gr.Markdown()
+    demo.load(show_profile,         None, status_md)
+    demo.load(list_private_models,  None, models_md)
+    login_btn.click(show_profile,        None, status_md)
+    login_btn.click(list_private_models, None, models_md)
+
+    # Controls
+    with gr.Row():
+        sdk_choice = gr.Radio(["gradio","streamlit"], "gradio", label="SDK")
+        api_key    = gr.Textbox(label="Gemini API Key", type="password")
+        grounding  = gr.Checkbox(label="Enable grounding")
+
+    # Chat + outputs
+    chatbot    = gr.Chatbot(type="messages")
+    user_in    = gr.Textbox(label="Prompt", placeholder="e.g. CSV inspector…")
+    send_btn   = gr.Button("Send")
+    build_box  = gr.Textbox(label="Build logs", lines=5)
+    run_box    = gr.Textbox(label="Run logs",   lines=5)
+    preview    = gr.HTML("<p>No Space yet.</p>")
+    state_repo = gr.Textbox(visible=False)
 
-    # 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")
-
-    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>")
-
-    # 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]
+        outputs=[chatbot, build_box, run_box, preview, state_repo]
+    )
+
+    # Manual refresh
+    with gr.Row():
+        refresh_build = gr.Button("Refresh Build Logs")
+        refresh_run   = gr.Button("Refresh Run Logs")
+
+    refresh_build.click(
+        fn=refresh_build_logs,
+        inputs=[state_repo],
+        outputs=build_box
+    )
+    refresh_run.click(
+        fn=refresh_run_logs,
+        inputs=[state_repo],
+        outputs=run_box
     )
 
-if __name__=="__main__":
-    demo.launch()
+demo.launch()