Add try block

app.py

@@ -5,22 +5,28 @@ from smolagents import ToolCollection, CodeAgent
 from smolagents import CodeAgent, InferenceClientModel
 from smolagents.mcp_client import MCPClient
 
-mcp_client = MCPClient(
-    {"url"="http://localhost:7860/gradio_api/mcp/sse"}
-)
+try:
+    mcp_client = MCPClient(
+        {"url": "http://localhost:7860/gradio_api/mcp/sse"}
+    )
+    tools = mcp_client.get_tools()
 
-tools=mcp_client.get_tools()
+    model = InferenceClientModel()
+    agent = CodeAgent(tools=[*tools], model=model)
 
-model = InferenceClientModel()
-agent = CodeAgent(tools=[*tools], model=model)
+    def call_agent(message, history):
+        return str(agent.run(message))
 
-demo = gr.ChatInterface(
-    fn=lambda message, history: str(agent.run(message)),
-    type="messages",
-    examples=["Prime factorization of 68"],
-    title="Agent with MCP Tools",
-    description="This is a simple agent that uses MCP tools to answer questions.",
-    messages=[],
-)
 
-demo.launch()
+    demo = gr.ChatInterface(
+        fn=lambda message, history: str(agent.run(message)),
+        type="messages",
+        examples=["Prime factorization of 68"],
+        title="Agent with MCP Tools",
+        description="This is a simple agent that uses MCP tools to answer questions.",
+        messages=[],
+    )
+
+    demo.launch()
+finally:
+    mcp_client.close()