Update app.py

app.py

@@ -1,9 +1,13 @@
-import re, json, os
+import re
+import json
+import os
+
 import gradio as gr
 from huggingface_hub import (
     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
 from google.genai import types
 
@@ -35,13 +39,15 @@ 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(rid,
-                token=token.token,
-                exist_ok=True,
-                repo_type="space",
-                space_sdk=sdk)
-    url = f"https://huggingface.co/spaces/{rid}"
-    log = f"✅ Space ready: {url} (SDK: {sdk})"
+    create_repo(
+        repo_id=rid,
+        token=token.token,
+        exist_ok=True,
+        repo_type="space",
+        space_sdk=sdk
+    )
+    url    = f"https://huggingface.co/spaces/{rid}"
+    log    = f"✅ Space ready: {url} (SDK: {sdk})"
     iframe = f'<iframe src="{url}" width="100%" height="400px"></iframe>'
     return rid, log, iframe
 
@@ -52,8 +58,13 @@ def write_file(path, content, repo_id, profile, token):
         return "⚠️ Please create a Space first."
     with open(path, "w") as f:
         f.write(content)
-    upload_file(path, path, repo_id,
-                token=token.token, repo_type="space")
+    upload_file(
+        path_or_fileobj=path,
+        path_in_repo=path,
+        repo_id=repo_id,
+        token=token.token,
+        repo_type="space"
+    )
     return f"✅ Wrote and uploaded `{path}`"
 
 def upload_file_to_space(file, path, repo_id, profile, token):
@@ -63,8 +74,13 @@ 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(file.name, path, repo_id,
-                token=token.token, repo_type="space")
+    upload_file(
+        path_or_fileobj=file.name,
+        path_in_repo=path,
+        repo_id=repo_id,
+        token=token.token,
+        repo_type="space"
+    )
     return f"✅ Uploaded `{path}` to `{repo_id}`"
 
 def _fetch_space_logs_level(repo_id, level):
@@ -76,9 +92,7 @@ def _fetch_space_logs_level(repo_id, level):
     jwt = r.json()["token"]
     url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
     lines = []
-    with get_session().get(url,
-                          headers=build_hf_headers(token=jwt),
-                          stream=True) as resp:
+    with get_session().get(url, headers=build_hf_headers(token=jwt), stream=True) as resp:
         hf_raise_for_status(resp)
         for raw in resp.iter_lines():
             if raw.startswith(b"data: "):
@@ -99,44 +113,76 @@ def get_container_logs(repo_id, profile, token):
 # — GEMINI FUNCTION DECLARATIONS —
 
 func_decls = [
-    {
-      "name":"create_space","description":"Create/get a HF Space",
-      "parameters":{...}
-    },
-    {
-      "name":"write_file","description":"Write/overwrite a file in the Space",
-      "parameters":{...}
-    },
-    {
-      "name":"list_files","description":"List files in the Space",
-      "parameters":{...}
-    },
-    {
-      "name":"get_build_logs","description":"Fetch build logs",
-      "parameters":{...}
-    },
-    {
-      "name":"get_run_logs","description":"Fetch run logs",
-      "parameters":{...}
-    }
+    types.FunctionDeclaration(
+        name="create_space",
+        description="Create/get a HF Space",
+        parameters={
+            "type": "object",
+            "properties": {
+                "repo_name": {"type": "string"},
+                "sdk":       {"type": "string", "enum": ["gradio", "streamlit"]},
+            },
+            "required": ["repo_name", "sdk"],
+        }
+    ),
+    types.FunctionDeclaration(
+        name="write_file",
+        description="Write/overwrite a file in the Space",
+        parameters={
+            "type": "object",
+            "properties": {
+                "path":    {"type": "string", "description": "File path in the space"},
+                "content": {"type": "string", "description": "Full text content"},
+            },
+            "required": ["path", "content"],
+        }
+    ),
+    types.FunctionDeclaration(
+        name="list_files",
+        description="List files in the Space",
+        parameters={
+            "type": "object",
+            "properties": {"repo_id": {"type": "string"}},
+            "required": ["repo_id"],
+        }
+    ),
+    types.FunctionDeclaration(
+        name="get_build_logs",
+        description="Fetch build logs",
+        parameters={
+            "type": "object",
+            "properties": {"repo_id": {"type": "string"}},
+            "required": ["repo_id"],
+        }
+    ),
+    types.FunctionDeclaration(
+        name="get_run_logs",
+        description="Fetch container (run) logs",
+        parameters={
+            "type": "object",
+            "properties": {"repo_id": {"type": "string"}},
+            "required": ["repo_id"],
+        }
+    ),
 ]
 
 # — CHAT HANDLER —
 
-def process_message(profile, token, user_msg,
-                    gemini_key, sidebar_repo, sidebar_sdk,
-                    chat_history, session):
-    # Initialize the chat
+def process_message(
+    profile, token, user_msg,
+    gemini_key, sidebar_repo, sidebar_sdk,
+    chat_history, session
+):
+    # Initialize Gemini chat
     if session.get("chat") is None:
         client = genai.Client(api_key=gemini_key)
         cfg = types.GenerateContentConfig(
             system_instruction=(
                 "You are a HF Spaces admin. You can create spaces, "
-                "write files (app.py, requirements.txt, README.md...), "
-                "list files, and fetch logs."
+                "write files, list files, and fetch logs."
             ),
             temperature=0,
-            tools=[ types.Tool(function_declarations=func_decls) ]
+            tools=func_decls,                  # ← pass declarations directly
         )
         session["chat"]     = client.chats.create(
             model="gemini-2.0-flash", config=cfg
@@ -144,7 +190,7 @@ def process_message(profile, token, user_msg,
         session["repo_id"]  = None
         session["messages"]= []
 
-    # Quick create via “call it NAME”
+    # Handle “call it XYZ” before other logic
     if session["repo_id"] is None:
         m = re.search(r"call (?:it )?([A-Za-z0-9_-]+)", user_msg, re.IGNORECASE)
         if m:
@@ -152,18 +198,17 @@ def process_message(profile, token, user_msg,
             rid, log, iframe = create_space(
                 name, sidebar_sdk, profile, token
             )
-            session["repo_id"] = rid
-            session["iframe"]  = iframe
-            session["log"]     = log
-            session["files"]   = ""
+            session.update({
+                "repo_id": rid,
+                "iframe": iframe,
+                "log": log,
+                "files": ""
+            })
             session["messages"].append({
                 "role":"assistant",
                 "content":f"✅ Created Space `{name}`. {log}"
             })
-            return (
-                session["messages"],
-                iframe, log, "", session
-            )
+            return session["messages"], iframe, log, "", session
 
     # Record user message
     session["messages"].append({"role":"user","content":user_msg})
@@ -195,7 +240,9 @@ def process_message(profile, token, user_msg,
 
         elif fn == "list_files":
             fl = list_repo_files(
-                session["repo_id"], token=token.token, repo_type="space"
+                session["repo_id"],
+                token=token.token,
+                repo_type="space"
             )
             result = {"files": "\n".join(fl)}
 
@@ -209,20 +256,18 @@ def process_message(profile, token, user_msg,
                 session["repo_id"], profile, token
             )}
 
-        else:
-            result = {"log": f"⚠️ Unknown function {fn}"}
-
-        # Pass the raw dict so Pydantic can validate it
+        # Send back the *dict* directly (no json.dumps)
         session["chat"].send_message(
             types.Content(
                 role="function",
                 parts=[ types.Part(
                     function_call=part.function_call,
                     function_response=result
-                ) ]
+                )]
             )
         )
         assistant_text = session["chat"].get_history()[-1].parts[0].text
+
     else:
         assistant_text = part.text
 
@@ -232,9 +277,9 @@ def process_message(profile, token, user_msg,
     })
 
     # 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"]
+    for k in ("iframe","log","files"):
+        if k in result:
+            session[k] = result[k]
 
     return (
         session["messages"],
@@ -248,23 +293,13 @@ def process_message(profile, token, user_msg,
 
 def sync_manual(profile, token, session):
     if not (profile and token and session.get("repo_id")):
-        return (
-            session.get("iframe",""),
-            "⚠️ Cannot sync manual changes.",
-            session.get("files",""),
-            session
-        )
+        return session.get("iframe",""), "⚠️ Cannot sync manual changes.", session.get("files",""), session
     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
-    )
+    return session["iframe"], session["log"], session["files"], session
 
 # — BUILD THE UI —
 
@@ -273,107 +308,93 @@ with gr.Blocks(css="""
 #main    { padding:1rem; }
 """) as demo:
 
-    with gr.Row():
-        # Sidebar
-        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(None)
-            token_state   = gr.State(None)
-
-            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])
-
-            models_md = gr.Markdown()
-            profile_state.change(list_private_models,
-                                 inputs=[profile_state, token_state],
-                                 outputs=[models_md])
-
-            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")
-
-            gr.Markdown("---")
-            confirm_btn  = gr.Button("🔄 Confirm Manual Changes")
-
-        # Main area
-        with gr.Column(elem_id="main", scale=3):
-            tabs = gr.Tabs()
-            with tabs:
-                with gr.TabItem("💬 Chat"):
-                    chatbox    = gr.Chatbot(type="messages")
-                    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")
-                    create_btn = gr.Button("Create Space")
-                    sess_id    = gr.Textbox(visible=False)
-                    log_c      = gr.Textbox(label="Log",
-                                            interactive=False, lines=2)
-                    preview    = gr.HTML("<p>No Space yet.</p>")
-
-                    create_btn.click(create_space,
-                                     inputs=[repo_m, sdk_m,
-                                             profile_state, token_state],
-                                     outputs=[sess_id, log_c, preview])
-
-                    gr.Markdown("#### Upload File")
-                    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)
-
-                    b_btn.click(get_build_logs,
-                                inputs=[sess_id,
-                                        profile_state, token_state],
-                                outputs=[log_b])
-                    r_btn.click(get_container_logs,
-                                inputs=[sess_id,
-                                        profile_state, token_state],
-                                outputs=[log_r])
-
-            # Persistent panels
-            gr.Markdown("---")
-            iframe_out = gr.HTML(label="🖼️ Preview")
-            log_out    = gr.Textbox(label="📋 Latest Log",
-                                    lines=4)
-            files_out  = gr.Textbox(label="📚 Files", lines=4)
+    # Sidebar
+    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(None)
+        token_state   = gr.State(None)
+        login_btn.click(None, [], [profile_state, token_state])
+
+        status_md = gr.Markdown("*Not logged in.*")
+        profile_state.change(show_profile, [profile_state], [status_md])
+
+        models_md = gr.Markdown()
+        profile_state.change(list_private_models,
+                             [profile_state, token_state], [models_md])
+
+        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")
+
+        gr.Markdown("---")
+        confirm_btn  = gr.Button("🔄 Confirm Manual Changes")
+
+    # Main area
+    with gr.Column(elem_id="main", scale=3):
+        tabs = gr.Tabs()
+        with tabs:
+            with gr.TabItem("💬 Chat"):
+                chatbox    = gr.Chatbot(type="messages")
+                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")
+                create_btn = gr.Button("Create Space")
+                sess_id    = gr.Textbox(visible=False)
+                log_c      = gr.Textbox(label="Log",
+                                        interactive=False, lines=2)
+                preview    = gr.HTML("<p>No Space yet.</p>")
+
+                create_btn.click(create_space,
+                                 [repo_m, sdk_m, profile_state, token_state],
+                                 [sess_id, log_c, preview])
+
+                gr.Markdown("#### Upload File")
+                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,
+                                 [file_u, path, sess_id, profile_state, token_state],
+                                 [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)
+
+                b_btn.click(get_build_logs,
+                            [sess_id, profile_state, token_state],
+                            [log_b])
+                r_btn.click(get_container_logs,
+                            [sess_id, profile_state, token_state],
+                            [log_r])
+
+        gr.Markdown("---")
+        iframe_out = gr.HTML(label="🖼️ Preview")
+        log_out    = gr.Textbox(label="📋 Latest Log", lines=4)
+        files_out  = gr.Textbox(label="📚 Files", lines=4)
 
     state = gr.State({})
     send_btn.click(process_message,
-                   inputs=[profile_state, token_state,
-                           user_input, gemini_key,
-                           sidebar_repo, sidebar_sdk,
-                           chatbox, state],
-                   outputs=[chatbox, iframe_out,
-                            log_out, files_out, state])
+                   [profile_state, token_state,
+                    user_input, gemini_key,
+                    sidebar_repo, sidebar_sdk,
+                    chatbox, state],
+                   [chatbox, iframe_out, log_out, files_out, state])
 
     confirm_btn.click(sync_manual,
-                      inputs=[profile_state, token_state, state],
-                      outputs=[iframe_out, log_out, files_out, state])
+                      [profile_state, token_state, state],
+                      [iframe_out, log_out, files_out, state])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)