Update app.py

app.py

@@ -1,48 +1,82 @@
 import gradio as gr
-from huggingface_hub import create_repo
+from huggingface_hub import create_repo, list_models
 
-with gr.Blocks(title="HF Space Creator") as demo:
-    login_btn  = gr.LoginButton(variant="huggingface", size="lg")
-    status_md  = gr.Markdown("*Not logged in.*")
+def show_profile(profile: gr.OAuthProfile | None) -> str:
+    """Update login status."""
+    if profile is None:
+        return "*Not logged in.*"
+    return f"✅ Logged in as **{profile.username}**"
 
-    def show_profile(profile: gr.OAuthProfile | None) -> str:
-        return "*Not logged in.*" if profile is None \
-            else f"✅ Logged in as **{profile.username}**"
+def list_private_models(
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken | None
+) -> str:
+    """Demo: list user’s private/public models."""
+    if oauth_token is None or profile is None:
+        return "Please log in to see your models."
+    models = [
+        f"{model.id} ({'private' if model.private else 'public'})"
+        for model in list_models(author=profile.username, token=oauth_token.token)
+    ]
+    if not models:
+        return "No models found."
+    return "Models:\n\n" + "\n - ".join(models)
 
-    demo.load(show_profile, inputs=None, outputs=status_md)
+def create_space(
+    repo_name: str,
+    oauth_token: gr.OAuthToken | None
+) -> tuple[str, str, str]:
+    """Create (or get) a Hugging Face Space and return its URL/logs/iframe."""
+    if oauth_token is None:
+        return "", "⚠️ Please log in first.", "<p>No Space created yet.</p>"
+    username = oauth_token.user_info["username"]
+    repo_id = f"{username}/{repo_name}"
+    create_repo(
+        repo_id=repo_id,
+        token=oauth_token.token,
+        exist_ok=True,
+        repo_type="space"
+    )
+    url    = f"https://huggingface.co/spaces/{repo_id}"
+    logmsg = f"✅ Space ready: {url}"
+    iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
+    return repo_id, logmsg, iframe
+
+with gr.Blocks(title="HF OAuth + Space Creator") as demo:
+    gr.Markdown(
+        "## Sign in with Hugging Face + Create a Space\n\n"
+        "1. Click **Sign in**.\n"
+        "2. Enter a name and **Create Space**.\n\n"
+        "---"
+    )
+
+    # — Login UI —
+    login_btn     = gr.LoginButton(variant="huggingface", size="lg")
+    status_md     = gr.Markdown("*Not logged in.*")
+    models_md     = gr.Markdown()
+
+    # Status & model list on load & after login/logout
+    demo.load(show_profile,       inputs=None, outputs=status_md)
     login_btn.click(show_profile, inputs=None, outputs=status_md)
 
-    repo_name      = gr.Textbox(label="New Space name", placeholder="your-repo-name")
-    create_btn     = gr.Button("Create Space", interactive=False)
-    session_id     = gr.Textbox(value="", visible=False)
-    logs           = gr.Textbox(label="Logs", interactive=False, lines=5)
-    preview_iframe = gr.HTML("<p>No Space created yet.</p>")
+    demo.load(list_private_models,       inputs=None, outputs=models_md)
+    login_btn.click(list_private_models, inputs=None, outputs=models_md)
 
-    def enable_create(profile: gr.OAuthProfile | None):
-        return gr.update(interactive=(profile is not None))
+    # — Create Space UI —
+    repo_name     = gr.Textbox(label="New Space name", placeholder="my-space-name")
+    create_btn    = gr.Button("Create Space", interactive=False)
+    session_id    = gr.Textbox(visible=False)
+    logs          = gr.Textbox(label="Logs", interactive=False, lines=3)
+    preview_iframe= gr.HTML("<p>No Space created yet.</p>")
 
-    demo.load(enable_create, inputs=None, outputs=create_btn)
-    login_btn.click(enable_create, inputs=None, outputs=create_btn)
+    # Enable the Create button once logged in
+    def enable_create(profile: gr.OAuthProfile | None) -> gr.Update:
+        return gr.update(interactive=profile is not None)
 
-    def create_space(
-        name: str,
-        oauth_token: gr.OAuthToken | None
-    ) -> tuple[str,str,str]:
-        if oauth_token is None:
-            return "", "⚠️ Please sign in first.", "<p>No Space created yet.</p>"
-        username = oauth_token.user_info["username"]
-        repo_id = f"{username}/{name}"
-        create_repo(
-            repo_id=repo_id,
-            token=oauth_token.token,
-            exist_ok=True,
-            repo_type="space"
-        )
-        url    = f"https://huggingface.co/spaces/{repo_id}"
-        logmsg = f"✅ Created or found Space: {url}"
-        iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
-        return repo_id, logmsg, iframe
+    demo.load(enable_create,       inputs=None, outputs=create_btn)
+    login_btn.click(enable_create, inputs=None, outputs=create_btn)
 
+    # Wire up create_space; Gradio injects the token for you
     create_btn.click(
         fn=create_space,
         inputs=[repo_name],