Update app.py

app.py

@@ -12,33 +12,37 @@ def list_private_models(
     oauth_token: gr.OAuthToken | None
 ) -> str:
     """Demo: list user’s private/public models."""
-    if oauth_token is None or profile is None:
+    if profile is None or oauth_token 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)
+    return "No models found." if not models \
+        else "Models:\n\n" + "\n - ".join(models)
 
 def create_space(
     repo_name: str,
+    sdk: str,
+    profile: gr.OAuthProfile | None,
     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:
+    """
+    Create (or get) a Hugging Face Space with the chosen SDK template,
+    returning (repo_id, logs, iframe).
+    """
+    if profile is None or 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}"
+    repo_id = f"{profile.username}/{repo_name}"
     create_repo(
         repo_id=repo_id,
         token=oauth_token.token,
         exist_ok=True,
-        repo_type="space"
+        repo_type="space",
+        space_sdk=sdk
     )
     url    = f"https://huggingface.co/spaces/{repo_id}"
-    logmsg = f"✅ Space ready: {url}"
+    logmsg = f"✅ Space ready: {url} (SDK: {sdk})"
     iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
     return repo_id, logmsg, iframe
 
@@ -46,40 +50,41 @@ 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"
+        "2. Pick a template, 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)
+    login_btn  = gr.LoginButton(variant="huggingface", size="lg")
+    status_md  = gr.Markdown("*Not logged in.*")
+    models_md  = gr.Markdown()
 
-    demo.load(list_private_models,       inputs=None, outputs=models_md)
-    login_btn.click(list_private_models, inputs=None, outputs=models_md)
+    demo.load(show_profile,             inputs=None, outputs=status_md)
+    login_btn.click(show_profile,       inputs=None, outputs=status_md)
+    demo.load(list_private_models,      inputs=None, outputs=models_md)
+    login_btn.click(list_private_models,inputs=None, outputs=models_md)
 
     # — Create Space UI —
     repo_name      = gr.Textbox(label="New Space name", placeholder="my-space-name")
+    sdk_selector   = gr.Radio(
+        choices=["gradio", "streamlit"],
+        value="gradio",
+        label="Space template (SDK)"
+    )
     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>")
 
-    # Enable the Create button once logged in
     def enable_create(profile: gr.OAuthProfile | None):
         return gr.update(interactive=profile is not None)
 
     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],
+        inputs=[repo_name, sdk_selector],
         outputs=[session_id, logs, preview_iframe]
     )