Update app.py

app.py

@@ -1,13 +1,33 @@
 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
 from google import genai
 from google.genai.types import Tool, GenerateContentConfig, GoogleSearch
-import gradio as gr
 
-# — UTILITIES — 
+# — 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)
+
+# — UTILITIES —
 
 def get_sdk_version(sdk_choice: str) -> str:
     pkg = "gradio" if sdk_choice == "gradio" else "streamlit"
@@ -17,13 +37,10 @@ def get_sdk_version(sdk_choice: str) -> str:
         return "UNKNOWN"
 
 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()
 
-# — HF SPACE LOGGING — 
+# — HF SPACE LOGGING —
 
 def _get_space_jwt(repo_id: str):
     url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
@@ -48,7 +65,7 @@ def fetch_logs(repo_id: str, level: str) -> str:
                     continue
     return "\n".join(lines)
 
-# — CORE LOOP — 
+# — CORE LOOP —
 
 def handle_user_message(
     history,
@@ -58,7 +75,7 @@ def handle_user_message(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
-    if not profile or not oauth_token:
+    if profile is None or oauth_token is None:
         return history + [{"role":"assistant","content":"⚠️ Please log in first."}], "", "", "<p>No Space yet.</p>"
 
     client = genai.Client(api_key=gemini_api_key)
@@ -146,23 +163,26 @@ 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..."
-                )
+    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 & MODEL LISTING
     login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
     models_md = gr.Markdown()
-    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")
@@ -177,11 +197,15 @@ with gr.Blocks(title="HF Space Auto‑Builder") as demo:
         outputs=[chatbot, build_box, run_box, preview]
     )
 
-    # — New “Refresh Logs” control for manual edits —
-    refresh_btn = gr.Button("Refresh Logs")
-    def _refresh(profile, token):
+    # — 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.click(_refresh, inputs=[status_md, models_md], outputs=[build_box, run_box])
+
+    refresh_btn = gr.Button("Refresh Logs")
+    # 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)