Update app.py

app.py

@@ -2,7 +2,6 @@ import re
 import json
 import time
 import importlib.metadata
-
 import gradio as gr
 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
@@ -28,7 +27,7 @@ def list_private_models(
     ]
     return "No models found." if not models else "Models:\n\n" + "\n - ".join(models)
 
-# — UTILITIES — 
+# — UTILITIES —
 
 def get_sdk_version(sdk_choice: str) -> str:
     pkg = "gradio" if sdk_choice == "gradio" else "streamlit"
@@ -38,13 +37,10 @@ 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()
 
-# — HF SPACE LOGGING — 
+# — HF SPACE LOGGING —
 
 def _get_space_jwt(repo_id: str):
     url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
@@ -69,28 +65,18 @@ def fetch_logs(repo_id: str, level: str) -> str:
                     continue
     return "\n".join(lines)
 
-# — CORE LOOP — 
+# — CORE LOOP —
 
 def handle_user_message(
     history,
     sdk_choice: str,
     gemini_api_key: str,
     grounding_enabled: bool,
-    space_suffix: str,
     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>"
-        )
-
-    username = profile.username
-    repo_name = f"{username}-{space_suffix}"
-    repo_id   = f"{username}/{repo_name}"
+        return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
 
     client = genai.Client(api_key=gemini_api_key)
     system_msg = {
@@ -105,8 +91,9 @@ def handle_user_message(
     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"
 
-    # Try up to 5 times to generate & push working code
+    build_logs = run_logs = ""
     for _ in range(5):
         tools = [Tool(google_search=GoogleSearch())] if grounding_enabled else []
         cfg   = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
@@ -117,15 +104,15 @@ def handle_user_message(
             config=cfg
         )
 
-        raw_code = resp.text
-        code = extract_code(raw_code)
+        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 with dynamic SDK version
+        # write dynamic README
         sdk_version = get_sdk_version(sdk_choice)
         readme = f"""---
 title: Wuhp Auto Space
@@ -143,75 +130,45 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
         with open(readme_fn, "w") as f:
             f.write(readme)
 
-        # Write requirements
+        # 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)
 
-        # Push to Hugging Face
-        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")
 
-        # 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():
+        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
 
         chat.append({
             "role":"user",
             "content":(
-                f"Build logs:\n{build}\n\n"
-                f"Run logs:\n{run}\n\n"
+                f"Build logs:\n{build_logs}\n\n"
+                f"Run logs:\n{run_logs}\n\n"
                 "Please 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}"
-    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."
-    return messages, placeholder, placeholder, iframe
-
-# — REFRESH LOGS — 
+    iframe   = f'<iframe src="https://huggingface.co/spaces/{repo_id}" width="100%" height="500px"></iframe>'
+    return messages, build_logs, run_logs, iframe
 
-def refresh_logs(
-    space_suffix: str,
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-):
-    if profile is None or oauth_token is None:
-        return "⚠️ Please log in.", "⚠️ Please log in."
-    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"
-                "1. Sign in  2. Enter your prompt  3. Watch code, README, requirements, logs, and preview\n\n---")
+    gr.Markdown("## Sign in + Auto‑Build Spaces\n\n1. Sign in  2. Enter your prompt  3. Watch code, README, requirements, logs, and preview\n\n---")
 
-    # LOGIN controls
+    # LOGIN & MODEL LISTING
     login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
     models_md = gr.Markdown()
@@ -220,33 +177,35 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
     login_btn.click(show_profile,        inputs=None, outputs=status_md)
     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.")
+    # 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>")
+    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,
-        inputs=[chatbot, sdk_choice, api_key, grounding, space_suffix],
+        inputs=[chatbot, sdk_choice, api_key, grounding],
         outputs=[chatbot, build_box, run_box, preview]
     )
 
-    # Manual log refresh
+    # — Refresh Logs button —
+    def _refresh(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None):
+        if not profile or not oauth_token:
+            return "", ""
+        repo = f"{profile.username}/{profile.username}-auto-space"
+        return fetch_logs(repo, "build"), fetch_logs(repo, "run")
+
     refresh_btn = gr.Button("Refresh Logs")
-    refresh_btn.click(
-        fn=refresh_logs,
-        inputs=[space_suffix],
-        outputs=[build_box, run_box]
-    )
+    # Gradio will auto‑inject `profile` and `oauth_token` here.
+    refresh_btn.click(_refresh, inputs=None, outputs=[build_box, run_box])
 
     demo.launch(server_name="0.0.0.0", server_port=7860)