Upload folder using huggingface_hub

app.py

@@ -73,6 +73,9 @@ huge_llm = ChatOpenAI(
 checkpointer = MemorySaver()
 long_term_store = InMemoryStore()
 
+# Global MCP Client to avoid process leaks
+GLOBAL_MCP_CLIENT = None
+
 # Schemas for Memory
 class UserPreferences(BaseModel):
     """Updated user preferences and project context."""
@@ -187,15 +190,14 @@ async def get_hf_logs(repo_id: str, log_type: Literal["build", "run"] = "run"):
     headers = {"Authorization": f"Bearer {token}"}
     
     try:
-        # Since logs are SSE, we just take a snapshot of recent entries
-        # In a real scenario, we might want to use a streaming client, but for tool use, a snippet is better.
+        # Snap snapshot of recent logs
         response = requests.get(url, headers=headers, timeout=10)
-        # We only take the last 2000 characters to avoid context overflow
         return response.text[-2000:]
     except Exception as e:
         return f"Error fetching logs: {e}"
 
 async def get_all_tools():
+    global GLOBAL_MCP_CLIENT
     # 1. Custom Tools
     custom_tools = [
         Tool(
@@ -216,29 +218,33 @@ async def get_all_tools():
     ]
     
     # 2. MCP Tools using MultiServerMCPClient
-    mcp_config = {
-        "jules": {
-            "transport": "stdio",
-            "command": "python3",
-            "args": ["mcp/mcp_jules.py"],
-            "env": os.environ.copy()
-        },
-        "github": {
-            "transport": "stdio",
-            "command": "npx",
-            "args": ["-y", "@modelcontextprotocol/server-github"],
-            "env": os.environ.copy()
+    if GLOBAL_MCP_CLIENT is None:
+        mcp_config = {
+            "jules": {
+                "transport": "stdio",
+                "command": "python3",
+                "args": ["mcp/mcp_jules.py"],
+                "env": os.environ.copy()
+            },
+            "github": {
+                "transport": "stdio",
+                "command": "npx",
+                "args": ["-y", "@modelcontextprotocol/server-github"],
+                "env": os.environ.copy()
+            }
         }
-    }
-    
+        try:
+            logger.info("Initializing GLOBAL MCP client...")
+            GLOBAL_MCP_CLIENT = MultiServerMCPClient(mcp_config)
+        except Exception as e:
+            logger.error(f"Failed to initialize MCP client: {e}")
+            return custom_tools
+
     try:
-        logger.info("Connecting to MCP servers...")
-        mcp_client = MultiServerMCPClient(mcp_config)
-        mcp_tools = await mcp_client.get_tools()
-        logger.info(f"Successfully loaded {len(mcp_tools)} tools from MCP servers.")
+        mcp_tools = await GLOBAL_MCP_CLIENT.get_tools()
         return custom_tools + mcp_tools
     except Exception as e:
-        logger.error(f"Failed to load MCP tools: {e}", exc_info=True)
+        logger.error(f"Failed to load MCP tools: {e}")
         return custom_tools
 
 # LangGraph Nodes
@@ -611,8 +617,6 @@ async def handle_monitor_session(session_id):
                 else: log_entries.append(f"[{t}] ACTIVITY: {json.dumps(act)[:100]}...")
             log_text = "\n\n".join(log_entries)
             
-            # Simple stuck detection: if no activity in 15 mins (if we had actual timestamps to compare)
-            # For now, just show status
             if status == "AWAITING_USER_FEEDBACK" or status == "AWAITING_PLAN_APPROVAL":
                  status = f"⚠️ {status}"
         
@@ -651,7 +655,11 @@ async def handle_test():
         jules_list = next((t for t in tools if t.name == "list_sources"), None)
         if jules_list:
             j_res = await jules_list.ainvoke({})
-            sources = j_res.get('sources', []) if isinstance(j_res, dict) else []
+            sources = []
+            if isinstance(j_res, dict): sources = j_res.get('sources', [])
+            elif isinstance(j_res, str):
+                 try: sources = json.loads(j_res).get('sources', [])
+                 except: pass
             results.append(f"Jules API (List Sources): OK (Found {len(sources)})")
         else: results.append("Jules API: Tool not found.")
     except Exception as e: results.append(f"MCP Servers: FAILED ({e})")
@@ -715,13 +723,12 @@ with gr.Blocks() as demo:
         
         nudge_output = gr.Textbox(label="Nudge Result")
         
-        # Automated monitoring timer
         auto_timer = gr.Timer(value=30)
         auto_timer.tick(handle_monitor_session, inputs=[mon_session_id], outputs=[activity_log, session_state, last_activity])
 
         async def start_comm_and_monitor(repo, prof, space, sdk, deploy, token_var):
             res, sid = await handle_jules_comm(repo, prof, space, sdk, deploy, token_var)
-            return res, sid, sid # Update comm_output, mon_session_id and its internal state
+            return res, sid, sid
         
         comm_btn.click(start_comm_and_monitor, [repo_input, hf_profile_in, hf_space_in, hf_sdk_in, hf_deploy_check, hf_token_var_in], [comm_output, mon_session_id])
         refresh_btn.click(handle_monitor_session, inputs=[mon_session_id], outputs=[activity_log, session_state, last_activity])