Update app.py

app.py

@@ -6,9 +6,11 @@ import gradio as gr
 from google import genai
 from google.genai import types
 
-# --- Initialize Gemini client ---
-client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
-# System prompt for Gemini
+# --- Globals ---
+client = None  # Will hold Gemini client
+chat = None    # Will hold Gemini chat session
+
+# --- System prompt 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. "
@@ -19,28 +21,14 @@ system_instruction = (
 get_build_logs_decl = {
     "name": "get_build_logs",
     "description": "Fetches build logs for a Space",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "repo_id": {"type": "string"}
-        },
-        "required": ["repo_id"]
-    }
+    "parameters": {"type":"object","properties":{"repo_id":{"type":"string"}},"required":["repo_id"]}
 }
 get_container_logs_decl = {
     "name": "get_container_logs",
     "description": "Fetches container logs for a Space",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "repo_id": {"type": "string"}
-        },
-        "required": ["repo_id"]
-    }
+    "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 (unchanged) ---
 def _fetch_space_logs_level(repo_id: str, level: str):
@@ -53,46 +41,39 @@ def _fetch_space_logs_level(repo_id: str, level: str):
     with get_session().get(logs_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
+            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
+                records.append({"timestamp": event.get("timestamp"), "message": event.get("data")})
+            except: pass
     return records
 
-# Space management (hidden behind scenes)
-def create_space_backend(username, token, repo_name, sdk):
+# --- Backends for creating space & uploading files ---
+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=token,
+        token=hf_token,
         exist_ok=True,
         repo_type="space",
         space_sdk=sdk
     )
     return repo_id
 
-def upload_file_backend(repo_id, token, file_path, path_in_repo):
-    upload_file(
-        path_or_fileobj=file_path,
-        path_in_repo=path_in_repo,
-        repo_id=repo_id,
-        token=token,
-        repo_type="space"
-    )
-    return True
-
-# --- Chat and interaction ---
-chat = None
-
-def init_chat(repo_id, api_key):
-    global chat
-    os.environ["GEMINI_API_KEY"] = api_key
+# --- Chat initialization & handlers ---
+def init_chat(repo_name: str, sdk: str, api_key: str, hf_profile: gr.OAuthProfile, hf_token: gr.OAuthToken):
+    global client, chat
+    # Validate inputs
+    if not api_key:
+        return {"success": False, "data": None, "message": "Missing Gemini API key."}, ""
+    if hf_profile is None or hf_token is None:
+        return {"success": False, "data": None, "message": "Please sign in with Hugging Face."}, ""
+    # Create HF space sandbox
+    repo_id = create_space_backend(hf_profile.username, hf_token.token, repo_name, sdk)
+    # Set HF token for log streaming
+    os.environ["HF_TOKEN"] = hf_token.token
+    # Init Gemini client and chat
+    client = genai.Client(api_key=api_key)
     chat = client.chats.create(
         model="gemini-2.5-flash-preview-04-17",
         config=types.GenerateContentConfig(
@@ -101,12 +82,15 @@ def init_chat(repo_id, api_key):
             temperature=0
         )
     )
-    return {"success": True, "data": None, "message": f"Sandbox ready: {repo_id}"}
+    return {"success": True, "data": None, "message": f"Sandbox ready: {repo_id}"}, repo_id
+
 
-def chatbot_respond(message, history, repo_id, api_key):
-    global chat
-    if chat is None:
-        init_chat(repo_id, api_key)
+def chatbot_respond(message: str, history: list, repo_id: str, api_key: str):
+    global chat, client
+    if not chat:
+        # Should not happen if initialized properly
+        history.append((None, "Error: chat not initialized."))
+        return history
     response = chat.send_message(message)
     part = response.candidates[0].content.parts[0]
     # Handle function calls
@@ -126,28 +110,31 @@ def chatbot_respond(message, history, repo_id, api_key):
 
 # --- UI: Chatbot with Sidebar for Space Creation ---
 with gr.Blocks(title="HF Code Sandbox Chat") as demo:
-    # Top login button
+    # Top bar: HF login
     with gr.Row():
         login_btn = gr.LoginButton(variant="huggingface", label="Sign in with HF")
-        status = gr.Markdown("*Not signed in.*")
+        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=None,
-            outputs=status
+            inputs=None, outputs=login_status
         )
 
     with gr.Row():
-        # Sidebar
+        # Sidebar for sandbox setup
         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")
+            sdk_selector = gr.Radio(label="SDK", choices=["gradio","streamlit"], value="gradio")
             create_btn = gr.Button("Initialize Sandbox")
-            create_status = gr.JSON(label="Init Status")
-            create_btn.click(init_chat, inputs=[repo_name, api_key], outputs=create_status)
-            # hidden store for repo_id
+            create_status = gr.JSON(label="Initialization Status")
+            # Hidden store for repo_id
             repo_store = gr.Variable("")
+            create_btn.click(
+                init_chat,
+                inputs=[repo_name, sdk_selector, api_key, login_btn.profile, login_btn.token],
+                outputs=[create_status, repo_store]
+            )
 
         # Chat area
         with gr.Column(scale=8):