Update app.py

app.py

@@ -1,10 +1,29 @@
 import gradio as gr
 import json, time
-from huggingface_hub import create_repo, upload_file, constants
+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                 # Gemini Python SDK
+from google import genai
 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}**"
+
+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)
+
 # — HELPERS FOR HF SPACE LOGS —
 
 def _get_space_jwt(repo_id: str):
@@ -23,7 +42,9 @@ def fetch_logs(repo_id: str, level: str):
             if raw.startswith(b"data: "):
                 try:
                     ev = json.loads(raw[len(b"data: "):].decode())
-                    lines.append(f"[{ev.get('timestamp','')}] {ev.get('data','')}")
+                    ts  = ev.get("timestamp","")
+                    txt = ev.get("data","")
+                    lines.append(f"[{ts}] {txt}")
                 except:
                     continue
     return "\n".join(lines)
@@ -31,19 +52,18 @@ def fetch_logs(repo_id: str, level: str):
 # — CORE LOOP: send prompt & (iteratively) deploy —
 
 def handle_user_message(
-    history,              # list of {"role","content"} dicts
+    history,               # list of {"role","content"} dicts
     sdk_choice: str,
     gemini_api_key: str,
     grounding_enabled: bool,
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
-    # require login
     if profile is None or oauth_token is None:
         return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
 
-    genai_client = genai.Client(api_key=gemini_api_key)
-    chat = [{
+    client = genai.Client(api_key=gemini_api_key)
+    chat   = [{
         "role":"system",
         "content":(
             f"You are an AI assistant that writes a HuggingFace Space using the "
@@ -52,58 +72,46 @@ def handle_user_message(
         )
     }] + history
 
-    code_filename   = "app.py" if sdk_choice=="gradio" else "streamlit_app.py"
-    config_filename = "README.md"
-    reqs_filename   = "requirements.txt"
-    repo_id         = f"{profile.username}/{profile.username}-auto-space"
+    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):
-        # build tools for grounding
         tools = []
         if grounding_enabled:
             tools.append(Tool(google_search=GoogleSearch()))
-        config = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
+        cfg = GenerateContentConfig(tools=tools, response_modalities=["TEXT"])
 
-        # call Gemini
-        resp = genai_client.models.generate_content(
+        resp    = client.models.generate_content(
             model="gemini-2.5-flash-preview-04-17",
             contents=[m["content"] for m in chat],
-            config=config
+            config=cfg
         )
-        ai_code = resp.text
-        chat.append({"role":"assistant", "content": ai_code})
-
-        # write code file
-        with open(code_filename, "w") as f:
-            f.write(ai_code)
+        code    = resp.text
+        chat.append({"role":"assistant","content":code})
 
-        # write README.md with correct app_file
-        readme = f"""---
+        # write files
+        with open(code_fn,   "w") as f: f.write(code)
+        with open(readme_fn, "w") as f:
+            f.write(f"""---
 title: Wuhp Auto Space
 emoji: 🐢
 colorFrom: red
 colorTo: pink
 sdk: {sdk_choice}
 sdk_version: 1.44.1
-app_file: {code_filename}
+app_file: {code_fn}
 pinned: false
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
-"""
-        with open(config_filename, "w") as f:
-            f.write(readme)
-
-        # write requirements.txt
-        if sdk_choice == "streamlit":
-            reqs = "streamlit\npandas\n"
-        else:
-            reqs = "gradio\npandas\n"
-        with open(reqs_filename, "w") as f:
-            f.write(reqs)
-
-        # create/update Space and upload all three files
+""")
+        reqs = "pandas\n" + ("streamlit\n" if sdk_choice=="streamlit" else "gradio\n")
+        with open(reqs_fn, "w") as f: f.write(reqs)
+
+        # deploy
         create_repo(
             repo_id=repo_id,
             token=oauth_token.token,
@@ -111,7 +119,7 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
             repo_type="space",
             space_sdk=sdk_choice
         )
-        for fn in (code_filename, config_filename, reqs_filename):
+        for fn in (code_fn, readme_fn, reqs_fn):
             upload_file(
                 path_or_fileobj=fn,
                 path_in_repo=fn,
@@ -120,15 +128,12 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
                 repo_type="space"
             )
 
-        # fetch logs
         build_logs = fetch_logs(repo_id, "build")
         run_logs   = fetch_logs(repo_id, "run")
 
-        # stop if no errors
         if "ERROR" not in build_logs.upper() and "ERROR" not in run_logs.upper():
             break
 
-        # feed errors back
         chat.append({
             "role":"user",
             "content":(
@@ -139,25 +144,25 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
         })
         time.sleep(2)
 
-    # prepare outputs for gr.Chatbot(type="messages")
-    messages = [{"role":m["role"], "content":m["content"]} for m in chat if m["role"]!="system"]
+    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 —
 
-def show_profile(profile):
-    return f"✅ Logged in as **{profile.username}**" if profile else "*Not logged in.*"
-
 with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
     gr.Markdown("## Sign in + Auto‑Build Spaces\n\n"
-                "1. Sign in\n2. Enter your prompt\n3. Watch the code, README, and requirements deploy & debug\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()
 
-    # LOGIN
-    login_btn = gr.LoginButton(variant="huggingface", size="lg")
-    status_md = gr.Markdown("*Not logged in.*")
-    demo.load(show_profile,         inputs=None, outputs=status_md)
-    login_btn.click(show_profile,   inputs=None, outputs=status_md)
+    demo.load(show_profile,        inputs=[login_btn], outputs=status_md)
+    demo.load(list_private_models, inputs=[login_btn], outputs=models_md)
+    login_btn.click(show_profile,        inputs=[login_btn], outputs=status_md)
+    login_btn.click(list_private_models, inputs=[login_btn], outputs=models_md)
 
     # CONTROLS
     sdk_choice = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK template")
@@ -166,7 +171,7 @@ with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
 
     # CHAT + OUTPUTS
     chatbot   = gr.Chatbot(type="messages")
-    user_in   = gr.Textbox(placeholder="e.g. Create me a CSV inspector…", label="Prompt")
+    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)
@@ -175,7 +180,7 @@ with gr.Blocks(title="HF Space Auto‑Builder (Gradio & Streamlit)") as demo:
 
     send_btn.click(
         fn=handle_user_message,
-        inputs=[chatbot, sdk_choice, api_key, grounding, gr.State(), gr.State()],
+        inputs=[chatbot, sdk_choice, api_key, grounding, login_btn],
         outputs=[chatbot, build_box, run_box, preview]
     )