Update app.py

app.py

@@ -1,29 +1,24 @@
 import gradio as gr
 
-from mcp.client.stdio import StdioServerParameters
-from smolagents import InferenceClientModel, CodeAgent, ToolCollection
+from smolagents import InferenceClientModel, CodeAgent
 from smolagents.mcp_client import MCPClient
 
-
-try:
-    mcp_client = MCPClient(
-        ## Try this working example on the hub:
-        # {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
-        {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
-    )
+# Use a 'with' block to ensure proper disconnection
+with MCPClient({"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}) as mcp_client:
+    # Fetch the available tools from the remote MCP server
     tools = mcp_client.get_tools()
 
+    # Set up the agent with tools and a default language model
     model = InferenceClientModel()
-    agent = CodeAgent(tools=[*tools], model=model)
+    agent = CodeAgent(tools=tools, model=model)
 
+    # Launch Gradio interface
     demo = gr.ChatInterface(
-        fn=lambda message, history: str(agent.run(message)),
+        fn=lambda message, history: (str(agent.run(message)), history),
         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.",
+        description="This is a simple agent that uses MCP tools to answer questions."
     )
 
     demo.launch()
-finally:
-    mcp_client.disconnect()