Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import requests
 from huggingface_hub import create_repo, list_models, upload_file
 
+# — USER INFO & MODEL LISTING — 
+
 def show_profile(profile: gr.OAuthProfile | None) -> str:
     if profile is None:
         return "*Not logged in.*"
@@ -9,7 +11,7 @@ def show_profile(profile: gr.OAuthProfile | None) -> str:
 
 def list_private_models(
     profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
+    oauth_token: gr.OAuthToken  | None
 ) -> str:
     if profile is None or oauth_token is None:
         return "Please log in to see your models."
@@ -17,14 +19,37 @@ def list_private_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)
+    return (
+        "No models found."
+        if not models
+        else "Models:\n\n" + "\n - ".join(models)
+    )
+
+# — BUTTON‑ENABLING HELPERS — 
+
+def enable_create(
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken | None
+) -> gr.Update:
+    """Enable Create Space once logged in."""
+    return gr.update(interactive=profile is not None)
+
+def enable_repo_actions(
+    repo_id: str,
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken  | None
+) -> gr.Update:
+    """Enable Upload & Get Logs once space exists and logged in."""
+    return gr.update(interactive=bool(repo_id and profile and oauth_token))
+
+# — CORE ACTIONS — 
 
 def create_space(
     repo_name: str,
     sdk: str,
     profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
-) -> tuple[str, str, str]:
+    oauth_token: gr.OAuthToken  | None
+) -> tuple[str,str,str]:
     if profile is None or oauth_token is None:
         return "", "⚠️ Please log in first.", "<p>No Space created yet.</p>"
     repo_id = f"{profile.username}/{repo_name}"
@@ -40,20 +65,14 @@ def create_space(
     iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
     return repo_id, logmsg, iframe
 
-def enable_button(
-    profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None,
-    repo_id: str
-):
-    return gr.update(interactive=bool(profile and oauth_token and repo_id))
-
 def upload_file_to_space(
-    file,               # dropped FileData hint
+    file,               # Gradio passes a Python tempfile-like object
     path_in_repo: str,
     repo_id: str,
-    oauth_token: gr.OAuthToken | None
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken  | None
 ) -> str:
-    if oauth_token is None:
+    if profile is None or oauth_token is None:
         return "⚠️ Please log in first."
     if not repo_id:
         return "⚠️ Please create a Space first."
@@ -71,17 +90,19 @@ def upload_file_to_space(
 def get_space_logs(
     repo_id: str,
     profile: gr.OAuthProfile | None,
-    oauth_token: gr.OAuthToken | None
+    oauth_token: gr.OAuthToken  | None
 ) -> str:
-    if profile is None or oauth_token is None:
+    if profile is None or oauth_token is None or not repo_id:
         return "⚠️ Please log in and create a Space first."
     endpoint = f"https://huggingface.co/api/spaces/{repo_id}/logs"
-    headers = {"Authorization": f"Bearer {oauth_token.token}"}
-    resp = requests.get(endpoint, headers=headers)
+    headers  = {"Authorization": f"Bearer {oauth_token.token}"}
+    resp     = requests.get(endpoint, headers=headers)
     if resp.status_code != 200:
         return f"❌ Failed to fetch logs ({resp.status_code})"
-    entries = resp.json()
-    return "\n".join(e.get("text", "") for e in entries)
+    entries  = resp.json()
+    return "\n".join(e.get("text","") for e in entries)
+
+# — BUILD THE UI — 
 
 with gr.Blocks(title="HF OAuth + Space Manager") as demo:
     gr.Markdown(
@@ -94,37 +115,34 @@ with gr.Blocks(title="HF OAuth + Space Manager") as demo:
     )
 
     # — LOGIN & MODEL LIST —
-    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)
-    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)
+    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)
+    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 —
     repo_name    = gr.Textbox(label="New Space name", placeholder="my-space")
     sdk_selector = gr.Radio(
-        choices=["gradio", "streamlit"],
+        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)
+    create_logs  = gr.Textbox(label="Create Logs", interactive=False, lines=3)
     preview      = gr.HTML("<p>No Space created yet.</p>")
 
-    demo.load(enable_button,
-              inputs=[login_btn, login_btn, session_id],
-              outputs=[create_btn])
-    login_btn.click(enable_button,
-                    inputs=[login_btn, login_btn, session_id],
-                    outputs=[create_btn])
+    # enable create_btn after login
+    demo.load(enable_create,       inputs=[login_btn], outputs=[create_btn])
+    login_btn.click(enable_create, inputs=[login_btn], outputs=[create_btn])
 
     create_btn.click(
         fn=create_space,
-        inputs=[repo_name, sdk_selector, login_btn, login_btn],
-        outputs=[session_id, logs, preview]
+        inputs=[repo_name, sdk_selector, login_btn],
+        outputs=[session_id, create_logs, preview]
     )
 
     # — UPLOAD FILES —
@@ -133,16 +151,16 @@ with gr.Blocks(title="HF OAuth + Space Manager") as demo:
     upload_btn    = gr.Button("Upload File", interactive=False)
     upload_logs   = gr.Textbox(label="Upload Logs", interactive=False, lines=2)
 
-    session_id.change(
-        fn=enable_button,
-        inputs=[login_btn, login_btn, session_id],
-        outputs=[upload_btn]
-    )
-    login_btn.click(
-        fn=enable_button,
-        inputs=[login_btn, login_btn, session_id],
-        outputs=[upload_btn]
-    )
+    # enable upload_btn after space exists & login
+    demo.load(enable_repo_actions, 
+              inputs=[session_id, login_btn], 
+              outputs=[upload_btn])
+    login_btn.click(enable_repo_actions,
+                    inputs=[session_id, login_btn],
+                    outputs=[upload_btn])
+    session_id.change(enable_repo_actions,
+                      inputs=[session_id, login_btn],
+                      outputs=[upload_btn])
 
     upload_btn.click(
         fn=upload_file_to_space,
@@ -154,20 +172,19 @@ with gr.Blocks(title="HF OAuth + Space Manager") as demo:
     logs_btn      = gr.Button("Get Space Logs", interactive=False)
     space_logs_md = gr.Textbox(label="Space Logs", interactive=False, lines=10)
 
-    session_id.change(
-        fn=enable_button,
-        inputs=[login_btn, login_btn, session_id],
-        outputs=[logs_btn]
-    )
-    login_btn.click(
-        fn=enable_button,
-        inputs=[login_btn, login_btn, session_id],
-        outputs=[logs_btn]
-    )
+    demo.load(enable_repo_actions, 
+              inputs=[session_id, login_btn], 
+              outputs=[logs_btn])
+    login_btn.click(enable_repo_actions,
+                    inputs=[session_id, login_btn],
+                    outputs=[logs_btn])
+    session_id.change(enable_repo_actions,
+                      inputs=[session_id, login_btn],
+                      outputs=[logs_btn])
 
     logs_btn.click(
         fn=get_space_logs,
-        inputs=[session_id, login_btn, login_btn],
+        inputs=[session_id, login_btn],
         outputs=[space_logs_md]
     )