Update app.py

app.py

@@ -1,26 +1,33 @@
 import gradio as gr
-import os
 
-from smolagents import InferenceClientModel, CodeAgent, MCPClient
+from mcp.client.stdio import StdioServerParameters
+from smolagents import ToolCollection, CodeAgent
+from smolagents import CodeAgent, InferenceClientModel
+from smolagents.mcp_client import MCPClient
 
+model = InferenceClientModel()
 
 try:
     mcp_client = MCPClient(
-        {"url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"}
+        {"url": "https://abidlabs-mcp-tools2.hf.space/gradio_api/mcp/sse"}
     )
+
     tools = mcp_client.get_tools()
+    agent = CodeAgent(tools=[*tools], model=model)
 
-    model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
-    agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
+    def call_agent(message, history):
+        return str(agent.run(message))
 
     demo = gr.ChatInterface(
-        fn=lambda message, history: str(agent.run(message)),
+        fn=call_agent,
         type="messages",
-        examples=["Analyze the sentiment of the following text 'This is awesome'"],
+        examples=["Prime factorization of 68"],
         title="Agent with MCP Tools",
         description="This is a simple agent that uses MCP tools to answer questions.",
     )
 
     demo.launch()
+except Exception as e:
+    raise e
 finally:
-    mcp_client.disconnect()
+    mcp_client.stop()