Update app.py

app.py

@@ -30,7 +30,7 @@ get_container_logs_decl = {
 }
 tools = [types.Tool(function_declarations=[get_build_logs_decl, get_container_logs_decl])]
 
-# --- Core Hugging Face functions (unchanged) ---
+# --- Core Hugging Face functions ---
 def _fetch_space_logs_level(repo_id: str, level: str):
     jwt_url = f"{constants.ENDPOINT}/api/spaces/{repo_id}/jwt"
     r = get_session().get(jwt_url, headers=build_hf_headers())
@@ -45,10 +45,11 @@ def _fetch_space_logs_level(repo_id: str, level: str):
             try:
                 event = json.loads(raw[len(b"data: "):].decode())
                 records.append({"timestamp": event.get("timestamp"), "message": event.get("data")})
-            except: pass
+            except:
+                pass
     return records
 
-# --- Backends for creating space & uploading files ---
+# --- 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(
@@ -60,7 +61,7 @@ def create_space_backend(username: str, hf_token: str, repo_name: str, sdk: str)
     )
     return repo_id
 
-# --- Chat initialization & handlers ---
+# --- Chat init & respond 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
@@ -68,11 +69,10 @@ def init_chat(repo_name: str, sdk: str, api_key: str, hf_profile: gr.OAuthProfil
         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
+    # Create HF sandbox space
     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
+    # Init Gemini client & chat
     client = genai.Client(api_key=api_key)
     chat = client.chats.create(
         model="gemini-2.5-flash-preview-04-17",
@@ -86,21 +86,16 @@ def init_chat(repo_name: str, sdk: str, api_key: str, hf_profile: gr.OAuthProfil
 
 
 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
+    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]
-    # Handle function calls
     if part.function_call:
         fn = part.function_call
         args = json.loads(fn.args)
-        if fn.name == "get_build_logs":
-            result = _fetch_space_logs_level(repo_id, "build")
-        else:
-            result = _fetch_space_logs_level(repo_id, "run")
+        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
     else:
@@ -108,7 +103,7 @@ def chatbot_respond(message: str, history: list, repo_id: str, api_key: str):
     history.append((message, reply))
     return history
 
-# --- UI: Chatbot with Sidebar for Space Creation ---
+# --- UI: Chatbot + Sidebar ---
 with gr.Blocks(title="HF Code Sandbox Chat") as demo:
     # Top bar: HF login
     with gr.Row():
@@ -116,11 +111,11 @@ with gr.Blocks(title="HF Code Sandbox Chat") as demo:
         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=login_status
+            inputs=[login_btn], outputs=[login_status]
         )
 
     with gr.Row():
-        # Sidebar for sandbox setup
+        # Sidebar
         with gr.Column(scale=2):
             gr.Markdown("### 🏗️ Create New Space Sandbox")
             api_key = gr.Textbox(label="Gemini API Key", placeholder="sk-...", type="password")
@@ -128,11 +123,10 @@ with gr.Blocks(title="HF Code Sandbox Chat") as demo:
             sdk_selector = gr.Radio(label="SDK", choices=["gradio","streamlit"], value="gradio")
             create_btn = gr.Button("Initialize Sandbox")
             create_status = gr.JSON(label="Initialization Status")
-            # Hidden store for repo_id
-            repo_store = gr.Variable("")
+            repo_store = gr.State("")
             create_btn.click(
                 init_chat,
-                inputs=[repo_name, sdk_selector, api_key, login_btn.profile, login_btn.token],
+                inputs=[repo_name, sdk_selector, api_key, login_btn, login_btn],
                 outputs=[create_status, repo_store]
             )
 
@@ -143,7 +137,7 @@ with gr.Blocks(title="HF Code Sandbox Chat") as demo:
             user_input.submit(
                 chatbot_respond,
                 inputs=[user_input, chatbot, repo_store, api_key],
-                outputs=chatbot
+                outputs=[chatbot]
             )
 
 if __name__ == "__main__":