Update app.py

app.py

@@ -1,9 +1,4 @@
-import gradio as gr
-import os
-
-from smolagents import InferenceClientModel, CodeAgent, MCPClient
-
-
+mcp_client = None
 try:
     mcp_client = MCPClient(
         {"url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"}
@@ -11,7 +6,11 @@ try:
     tools = mcp_client.get_tools()
 
     model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
-    agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
+    agent = CodeAgent(
+        tools=[*tools],
+        model=model,
+        additional_authorized_imports=["json", "ast", "urllib", "base64"]
+    )
 
     demo = gr.ChatInterface(
         fn=lambda message, history: str(agent.run(message)),
@@ -22,5 +21,8 @@ try:
     )
 
     demo.launch()
+except Exception as e:
+    print("❌ MCP connection failed:", e)
 finally:
-    mcp_client.disconnect()
+    if mcp_client:
+        mcp_client.disconnect()