Update app.py

app.py

@@ -1,23 +1,23 @@
 import os
 import json
+import gradio as gr
 from huggingface_hub import create_repo, list_models, upload_file, constants
 from huggingface_hub.utils import build_hf_headers, get_session, hf_raise_for_status
-import gradio as gr
 from google import genai
 from google.genai import types
 
 # --- Globals ---
-client = None  # Will hold Gemini client
-chat = None    # Will hold Gemini chat session
+client = None
+chat = None
 
-# --- System prompt for Gemini ---
+# --- System instruction for Gemini ---
 system_instruction = (
     "You are a helpful assistant that writes, debugs, and pushes code to Hugging Face Spaces. "
-    "Treat Hugging Face as a sandbox: create spaces, upload code, and debug via function calls. "
-    "Respond in JSON with {success, data, message}."
+    "Treat Spaces as a sandbox: create, upload, debug. "
+    "Use function calling for logs and respond in JSON {success, data, message}."
 )
 
-# --- Function declarations for logs (behind the scenes) ---
+# --- Function declarations for logs ---
 get_build_logs_decl = {
     "name": "get_build_logs",
     "description": "Fetches build logs for a Space",
@@ -28,118 +28,93 @@ get_container_logs_decl = {
     "description": "Fetches container logs for a Space",
     "parameters": {"type":"object","properties":{"repo_id":{"type":"string"}},"required":["repo_id"]}
 }
-tools = [types.Tool(function_declarations=[get_build_logs_decl, get_container_logs_decl])]
+tools = [ types.Tool(function_declarations=[get_build_logs_decl, get_container_logs_decl]) ]
 
-# --- Core Hugging Face functions ---
-def _fetch_space_logs_level(repo_id: str, level: str):
+# --- HF helpers ---
+def create_space_backend(username, hf_token, repo_name, sdk):
+    repo_id = f"{username}/{repo_name}"
+    create_repo(repo_id=repo_id, token=hf_token, exist_ok=True, repo_type="space", space_sdk=sdk)
+    return repo_id
+
+def fetch_logs(repo_id, level):
     jwt_url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
     r = get_session().get(jwt_url, headers=build_hf_headers())
     hf_raise_for_status(r)
     jwt = r.json()["token"]
-    logs_url = f"https://api.hf.space/v1/{repo_id}/logs/{level}"
-    records = []
-    with get_session().get(logs_url, headers=build_hf_headers(token=jwt), stream=True) as resp:
+    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:
         hf_raise_for_status(resp)
         for raw in resp.iter_lines():
-            if not raw.startswith(b"data: "): continue
-            try:
-                event = json.loads(raw[len(b"data: "):].decode())
-                records.append({"timestamp": event.get("timestamp"), "message": event.get("data")})
-            except:
-                pass
-    return records
+            if raw.startswith(b"data: "):
+                try:
+                    ev=json.loads(raw[len(b"data: "):].decode())
+                    lines.append({"timestamp":ev.get("timestamp"),"message":ev.get("data")})
+                except: pass
+    return lines
 
-# --- Backend: create HF Space ---
-def create_space_backend(username: str, hf_token: str, repo_name: str, sdk: str) -> str:
-    repo_id = f"{username}/{repo_name}"
-    create_repo(
-        repo_id=repo_id,
-        token=hf_token,
-        exist_ok=True,
-        repo_type="space",
-        space_sdk=sdk
-    )
-    return repo_id
-
-# --- Chat init & respond handlers ---
-
-def init_chat(repo_name: str, sdk: str, api_key: str, hf_profile: gr.OAuthProfile, hf_token: gr.OAuthToken):
+# --- Chat init & respond ---
+def init_chat(repo_name, sdk, gemini_key, hf_profile, hf_token):
     global client, chat
-    # Validate inputs
-    if not api_key:
-        return {"success": False, "data": None, "message": "Missing Gemini API key."}, ""
+    # Validate
     if hf_profile is None or hf_token is None:
-        return {"success": False, "data": None, "message": "Please sign in with Hugging Face."}, ""
-    # Create HF sandbox space
+        return {"success":False,"data":None,"message":"Please sign in with HF."}, ""
+    if not gemini_key:
+        return {"success":False,"data":None,"message":"Missing Gemini API key."}, ""
+    # create space
     repo_id = create_space_backend(hf_profile.username, hf_token.token, repo_name, sdk)
     os.environ["HF_TOKEN"] = hf_token.token
-    # Init Gemini client & chat
-    client = genai.Client(api_key=api_key)
+    # init Gemini
+    client = genai.Client(api_key=gemini_key)
     chat = client.chats.create(
         model="gemini-2.5-flash-preview-04-17",
-        config=types.GenerateContentConfig(
-            system_instruction=system_instruction,
-            tools=tools,
-            temperature=0
-        )
+        config=types.GenerateContentConfig(system_instruction=system_instruction, tools=tools, temperature=0)
     )
-    return {"success": True, "data": None, "message": f"Sandbox ready: {repo_id}"}, repo_id
+    return {"success":True,"data":None,"message":f"Sandbox ready: {repo_id}"}, repo_id
 
-
-def chatbot_respond(message: str, history: list, repo_id: str, api_key: str):
+def chatbot_respond(message, history, repo_id, gemini_key):
     global chat
     if chat is None:
         history.append((None, "Error: chat not initialized."))
         return history
-    response = chat.send_message(message)
-    part = response.candidates[0].content.parts[0]
+    resp = chat.send_message(message)
+    part = resp.candidates[0].content.parts[0]
     if part.function_call:
-        fn = part.function_call
-        args = json.loads(fn.args)
-        result = _fetch_space_logs_level(repo_id, "build" if fn.name=="get_build_logs" else "run")
-        response2 = chat.send_message("", function_response={fn.name: result})
-        reply = response2.candidates[0].content.parts[0].text
+        fn=part.function_call
+        args=json.loads(fn.args)
+        level = "build" if fn.name=="get_build_logs" else "run"
+        logs=fetch_logs(repo_id, level)
+        resp2 = chat.send_message("", function_response={fn.name:logs})
+        reply=resp2.candidates[0].content.parts[0].text
     else:
-        reply = part.text
+        reply=part.text
     history.append((message, reply))
     return history
 
-# --- UI: Chatbot + Sidebar ---
-with gr.Blocks(title="HF Code Sandbox Chat") as demo:
-    # Top bar: HF login
-    with gr.Row():
-        login_btn = gr.LoginButton("Sign in with HF", variant="huggingface")
-        login_status = gr.Markdown("*Not signed in.*")
-        login_btn.click(
-            lambda profile: f"✅ Logged in as {profile.username}" if profile else "*Not signed in.*",
-            inputs=[login_btn], outputs=[login_status]
-        )
+# --- UI ---
+with gr.Blocks() as demo:
+    gr.Markdown("# HF Code Sandbox Chat")
+    # login
+    login_btn = gr.LoginButton("Sign in with HF", variant="huggingface")
+    login_status = gr.Markdown("*Not signed in.*")
+    models_md = gr.Markdown()
+    login_btn.click(lambda p: show_profile(p), inputs=[login_btn], outputs=[login_status])
+    login_btn.click(lambda p, t: list_private_models(p,t), inputs=[login_btn, login_btn.token], outputs=[models_md])
 
     with gr.Row():
-        # Sidebar
         with gr.Column(scale=2):
-            gr.Markdown("### 🏗️ Create New Space Sandbox")
-            api_key = gr.Textbox(label="Gemini API Key", placeholder="sk-...", type="password")
-            repo_name = gr.Textbox(label="Space Name", placeholder="my-sandbox")
-            sdk_selector = gr.Radio(label="SDK", choices=["gradio","streamlit"], value="gradio")
-            create_btn = gr.Button("Initialize Sandbox")
-            create_status = gr.JSON(label="Initialization Status")
+            gr.Markdown("## Setup Sandbox")
+            gemini_key = gr.Textbox(label="Gemini API Key", type="password")
+            repo_name = gr.Textbox(label="Space Name")
+            sdk = gr.Radio(choices=["gradio","streamlit"], label="SDK", value="gradio")
+            init_btn = gr.Button("Initialize Sandbox")
+            init_status = gr.JSON()
             repo_store = gr.State("")
-            create_btn.click(
-                init_chat,
-                inputs=[repo_name, sdk_selector, api_key, login_btn, login_btn],
-                outputs=[create_status, repo_store]
-            )
-
-        # Chat area
+            init_btn.click(init_chat, inputs=[repo_name, sdk, gemini_key, login_btn, login_btn.token], outputs=[init_status, repo_store])
         with gr.Column(scale=8):
-            chatbot = gr.Chatbot()
-            user_input = gr.Textbox(show_label=False, placeholder="Ask the sandbox to write or debug code...")
-            user_input.submit(
-                chatbot_respond,
-                inputs=[user_input, chatbot, repo_store, api_key],
-                outputs=[chatbot]
-            )
+            chatbot = gr.Chatbot(type="messages")
+            user_input = gr.Textbox(show_label=False, placeholder="Ask to write/debug code...")
+            user_input.submit(chatbot_respond, inputs=[user_input, chatbot, repo_store, gemini_key], outputs=[chatbot])
 
 if __name__ == "__main__":
     demo.launch()