Update app.py

app.py

@@ -1,7 +1,7 @@
-import os, json
+import json, os
 import gradio as gr
 from huggingface_hub import (
-    create_repo, list_models, upload_file, constants
+    create_repo, list_models, upload_file, list_repo_files, constants
 )
 from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status
 from google import genai
@@ -35,13 +35,7 @@ def create_space(repo_name, sdk, profile, token):
     if not (profile and token):
         return "", "⚠️ Please log in first.", "<p>No Space created yet.</p>"
     rid = f"{profile.username}/{repo_name}"
-    create_repo(
-        repo_id=rid,
-        token=token.token,
-        exist_ok=True,
-        repo_type="space",
-        space_sdk=sdk
-    )
+    create_repo(rid, token=token.token, exist_ok=True, repo_type="space", space_sdk=sdk)
     url = f"https://huggingface.co/spaces/{rid}"
     return rid, f"✅ Space ready: {url}", f'<iframe src="{url}" width="100%" height="400px"></iframe>'
 
@@ -52,18 +46,11 @@ def upload_file_to_space(file, path, repo_id, profile, token):
         return "⚠️ Please create a Space first."
     if not file:
         return "⚠️ No file selected."
-    upload_file(
-        path_or_fileobj=file.name,
-        path_in_repo=path,
-        repo_id=repo_id,
-        token=token.token,
-        repo_type="space"
-    )
+    upload_file(file.name, path, repo_id, token=token.token, repo_type="space")
     return f"✅ Uploaded `{path}` to `{repo_id}`"
 
 def _fetch_space_logs_level(repo_id, level):
-    jwt_url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
-    r = get_session().get(jwt_url, headers=build_hf_headers())
+    r = get_session().get(f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt", headers=build_hf_headers())
     hf_raise_for_status(r)
     jwt = r.json()["token"]
     url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
@@ -86,12 +73,11 @@ def get_container_logs(repo_id, profile, token):
         return "⚠️ Please log in and create a Space first."
     return _fetch_space_logs_level(repo_id, "run")
 
-
 # — GEMINI FUNCTION DECLARATIONS —
 
 func_decls = [
     {
-      "name":"create_space","description":"Create or get a Hugging Face Space",
+      "name":"create_space","description":"Create/get a HF Space",
       "parameters":{
         "type":"object",
         "properties":{
@@ -102,7 +88,7 @@ func_decls = [
       }
     },
     {
-      "name":"list_files","description":"List files in the Space",
+      "name":"list_files","description":"List files in Space",
       "parameters":{
         "type":"object",
         "properties":{"repo_id":{"type":"string"}},
@@ -136,8 +122,8 @@ def process_message(profile, token, user_msg,
         client = genai.Client(api_key=gemini_key)
         cfg = types.GenerateContentConfig(
             system_instruction=(
-                "You are a HF Spaces admin. "
-                "Use functions to create spaces, list files, and fetch logs."
+                "You are a HF Spaces admin. Use functions to create spaces, "
+                "list files, and fetch logs."
             ),
             temperature=0,
             tools=[ types.Tool(function_declarations=func_decls) ]
@@ -154,20 +140,18 @@ def process_message(profile, token, user_msg,
     if part.function_call:
         name = part.function_call.name
         args = json.loads(part.function_call.args)
-        if name=="create_space":
+        if name == "create_space":
             rid, log, iframe = create_space(
                 args["repo_name"], args["sdk"], profile, token
             )
             session["repo_id"] = rid
             result = {"log": log, "iframe": iframe}
-        elif name=="list_files":
-            files = list_repo_files(
-                session["repo_id"], token=token.token, repo_type="space"
-            )
-            result = {"files": "\n".join(files)}
-        elif name=="get_build_logs":
+        elif name == "list_files":
+            fl = list_repo_files(session["repo_id"], token=token.token, repo_type="space")
+            result = {"files": "\n".join(fl)}
+        elif name == "get_build_logs":
             result = {"log": get_build_logs(session["repo_id"], profile, token)}
-        elif name=="get_run_logs":
+        elif name == "get_run_logs":
             result = {"log": get_container_logs(session["repo_id"], profile, token)}
         else:
             result = {"log": f"⚠️ Unknown function {name}"}
@@ -185,7 +169,6 @@ def process_message(profile, token, user_msg,
     else:
         final = part.text
 
-    # update panels
     if "iframe" in result: session["iframe"] = result["iframe"]
     if "log"    in result: session["log"]    = result["log"]
     if "files"  in result: session["files"]  = result["files"]
@@ -203,25 +186,19 @@ def sync_manual(profile, token, session):
                 "⚠️ Cannot sync manual changes.",
                 session.get("files",""),
                 session)
-    files = list_repo_files(
-        session["repo_id"], token=token.token, repo_type="space"
-    )
-    session["files"] = "\n".join(files)
+    fl = list_repo_files(session["repo_id"], token=token.token, repo_type="space")
+    session["files"] = "\n".join(fl)
     session["log"]   = "🔄 Manual changes synced."
     return (session.get("iframe",""),
             session["log"],
             session["files"],
             session)
 
-def capture_login(profile, oauth_token):
-    return profile, oauth_token
-
-
-# — BUILD THE UI —
+# — BUILD UI —
 
 with gr.Blocks(css="""
-#sidebar {background:#f0f0f0;padding:1rem;border-right:1px solid #ccc;}
-#main    {padding:1rem;}
+#sidebar { background: #f2f2f2; padding: 1rem; border-right:1px solid #ccc; }
+#main    { padding: 1rem; }
 """) as demo:
 
     with gr.Row():
@@ -229,17 +206,14 @@ with gr.Blocks(css="""
         with gr.Column(elem_id="sidebar", scale=1):
             gr.Markdown("### 🔑 HF Login & Config")
             login_btn     = gr.LoginButton(variant="huggingface", size="sm")
-            profile_state = gr.State()
-            token_state   = gr.State()
+            profile_state = gr.State(None)
+            token_state   = gr.State(None)
 
-            login_btn.click(capture_login,
-                            inputs=[login_btn],
-                            outputs=[profile_state, token_state])
+            # capture both outputs
+            login_btn.click(None, [], [profile_state, token_state])
 
             status_md = gr.Markdown("*Not logged in.*")
-            profile_state.change(show_profile,
-                                 inputs=[profile_state],
-                                 outputs=[status_md])
+            profile_state.change(show_profile, inputs=[profile_state], outputs=[status_md])
 
             models_md = gr.Markdown()
             profile_state.change(list_private_models,
@@ -248,8 +222,7 @@ with gr.Blocks(css="""
 
             gemini_key   = gr.Textbox(label="Gemini API Key", type="password")
             sidebar_repo = gr.Textbox(label="Space name", placeholder="my-space")
-            sidebar_sdk  = gr.Radio(["gradio","streamlit"],
-                                    value="gradio", label="SDK")
+            sidebar_sdk  = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
 
             gr.Markdown("---")
             confirm_btn = gr.Button("🔄 Confirm Manual Changes")
@@ -259,19 +232,16 @@ with gr.Blocks(css="""
             tabs = gr.Tabs()
             with tabs:
                 with gr.TabItem("💬 Chat"):
-                    chatbox    = gr.Chatbot(type="pairs")
-                    user_input = gr.Textbox(show_label=False,
-                                            placeholder="Ask the LLM…")
+                    chatbox    = gr.Chatbot(type="tuples")
+                    user_input = gr.Textbox(show_label=False, placeholder="Ask the LLM…")
                     send_btn   = gr.Button("Send")
                 with gr.TabItem("🛠️ Manual"):
                     gr.Markdown("#### Create a Space")
-                    repo_m   = gr.Textbox(label="Name")
-                    sdk_m    = gr.Radio(["gradio","streamlit"],
-                                        value="gradio", label="SDK")
+                    repo_m     = gr.Textbox(label="Name")
+                    sdk_m      = gr.Radio(["gradio","streamlit"], value="gradio", label="SDK")
                     create_btn = gr.Button("Create Space")
                     sess_id    = gr.Textbox(visible=False)
-                    log_c      = gr.Textbox(label="Log",
-                                            interactive=False, lines=2)
+                    log_c      = gr.Textbox(label="Log", interactive=False, lines=2)
                     preview    = gr.HTML("<p>No Space yet.</p>")
 
                     create_btn.click(create_space,
@@ -280,37 +250,30 @@ with gr.Blocks(css="""
                                      outputs=[sess_id, log_c, preview])
 
                     gr.Markdown("#### Upload File")
-                    path_in_repo = gr.Textbox(label="Path in Repo",
-                                              value="app.py")
-                    file_uploader = gr.File()
-                    up_btn       = gr.Button("Upload File")
-                    log_u        = gr.Textbox(label="Log",
-                                              interactive=False, lines=2)
-
-                    up_btn.click(upload_file_to_space,
-                                 inputs=[file_uploader, path_in_repo,
-                                         sess_id,
-                                         profile_state, token_state],
-                                 outputs=[log_u])
+                    path        = gr.Textbox(label="Path", value="app.py")
+                    file_u      = gr.File()
+                    upload_btn  = gr.Button("Upload File")
+                    log_u       = gr.Textbox(label="Log", interactive=False, lines=2)
+
+                    upload_btn.click(upload_file_to_space,
+                                     inputs=[file_u, path, sess_id,
+                                             profile_state, token_state],
+                                     outputs=[log_u])
 
                     gr.Markdown("#### Fetch Logs")
                     b_btn = gr.Button("Build Logs")
                     r_btn = gr.Button("Run Logs")
-                    log_b = gr.Textbox(label="Build",
-                                       interactive=False, lines=5)
-                    log_r = gr.Textbox(label="Run",
-                                       interactive=False, lines=5)
+                    log_b = gr.Textbox(label="Build", interactive=False, lines=5)
+                    log_r = gr.Textbox(label="Run",   interactive=False, lines=5)
 
                     b_btn.click(get_build_logs,
-                                inputs=[sess_id,
-                                        profile_state, token_state],
+                                inputs=[sess_id, profile_state, token_state],
                                 outputs=[log_b])
                     r_btn.click(get_container_logs,
-                                inputs=[sess_id,
-                                        profile_state, token_state],
+                                inputs=[sess_id, profile_state, token_state],
                                 outputs=[log_r])
 
-            # Persistent panels
+            # persistent panels
             gr.Markdown("---")
             iframe_out = gr.HTML(label="🖼️ Preview")
             log_out    = gr.Textbox(label="📋 Latest Log", lines=4)