Update app.py

app.py

@@ -27,7 +27,7 @@ def enable_create(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
-    """Enable Create Space once logged in."""
+    # Called with no inputs→Gradio auto‑injects profile & token
     return gr.update(interactive=profile is not None)
 
 def enable_repo_actions(
@@ -35,7 +35,7 @@ def enable_repo_actions(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ):
-    """Enable Upload & Get Logs once space exists and logged in."""
+    # Called with inputs=[session_id]→Gradio injects profile & token
     return gr.update(interactive=bool(repo_id and profile and oauth_token))
 
 # — CORE ACTIONS —
@@ -46,7 +46,7 @@ def create_space(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ) -> tuple[str, str, str]:
-    if profile is None or oauth_token is None:
+    if not profile or not oauth_token:
         return "", "⚠️ Please log in first.", "<p>No Space created yet.</p>"
     repo_id = f"{profile.username}/{repo_name}"
     create_repo(
@@ -62,17 +62,17 @@ def create_space(
     return repo_id, logmsg, iframe
 
 def upload_file_to_space(
-    file,               # Gradio passes a Python tempfile-like object
+    file,
     path_in_repo: str,
     repo_id: str,
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ) -> str:
-    if profile is None or oauth_token is None:
+    if not profile or not oauth_token:
         return "⚠️ Please log in first."
     if not repo_id:
         return "⚠️ Please create a Space first."
-    if file is None:
+    if not file:
         return "⚠️ No file selected."
     upload_file(
         path_or_fileobj=file.name,
@@ -88,15 +88,15 @@ def get_space_logs(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ) -> str:
-    if profile is None or oauth_token is None or not repo_id:
+    if not profile or not oauth_token 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)
+    resp = requests.get(
+        f"https://huggingface.co/api/spaces/{repo_id}/logs",
+        headers={"Authorization": f"Bearer {oauth_token.token}"}
+    )
     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)
+    return "\n".join(entry.get("text","") for entry in resp.json())
 
 # — BUILD THE UI —
 
@@ -114,15 +114,16 @@ with gr.Blocks(title="HF OAuth + Space Manager") as demo:
     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)
+    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)"
     )
@@ -131,13 +132,13 @@ with gr.Blocks(title="HF OAuth + Space Manager") as demo:
     create_logs  = gr.Textbox(label="Create Logs", interactive=False, lines=3)
     preview      = gr.HTML("<p>No Space created yet.</p>")
 
-    # 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])
+    # enable create_btn once logged in
+    demo.load(enable_create,      inputs=None,           outputs=[create_btn])
+    login_btn.click(enable_create, inputs=None,          outputs=[create_btn])
 
     create_btn.click(
         fn=create_space,
-        inputs=[repo_name, sdk_selector, login_btn],
+        inputs=[repo_name, sdk_selector],  # only non‑OAuth inputs
         outputs=[session_id, create_logs, preview]
     )
 
@@ -147,20 +148,20 @@ 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)
 
-    # enable upload_btn after space exists & login
-    demo.load(enable_repo_actions, 
-              inputs=[session_id, login_btn], 
+    # enable upload_btn once space exists & logged in
+    demo.load(enable_repo_actions,
+              inputs=[session_id],  # only non‑OAuth inputs
               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],
+                      inputs=[session_id],
                       outputs=[upload_btn])
+    login_btn.click(enable_repo_actions,
+                    inputs=[session_id],
+                    outputs=[upload_btn])
 
     upload_btn.click(
         fn=upload_file_to_space,
-        inputs=[file_uploader, path_in_repo, session_id, login_btn],
+        inputs=[file_uploader, path_in_repo, session_id],
         outputs=[upload_logs]
     )
 
@@ -168,19 +169,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)
 
-    demo.load(enable_repo_actions, 
-              inputs=[session_id, login_btn], 
+    demo.load(enable_repo_actions,
+              inputs=[session_id],
               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],
+                      inputs=[session_id],
                       outputs=[logs_btn])
+    login_btn.click(enable_repo_actions,
+                    inputs=[session_id],
+                    outputs=[logs_btn])
 
     logs_btn.click(
         fn=get_space_logs,
-        inputs=[session_id, login_btn],
+        inputs=[session_id],
         outputs=[space_logs_md]
     )