Update app.py

app.py

@@ -1,21 +1,25 @@
 import gradio as gr
 import os
+from mcp.client.stdio import StdioServerParameters
+from smolagents import ToolCollection, CodeAgent
+from smolagents import CodeAgent, InferenceClientModel
+from smolagents.mcp_client import MCPClient
 
-from mcp import StdioServerParameters
-from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClient
-
+model = InferenceClientModel(token=os.getenv("mcp2")
 
 try:
     mcp_client = MCPClient(
-        {"url": "https://wlchee-mcp-sentiment.hf.space/gradio_api/mcp/sse"} # This is the MCP Server we created in the previous section
+        {"url": "https://wlchee-mcp-sentiment.hf.space/gradio_api/mcp/sse"}
     )
-    tools = mcp_client.get_tools()
 
-    model = InferenceClientModel(token=os.getenv("mcp2"))
+    tools = mcp_client.get_tools()
     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)),
+        fn=call_agent,
         type="messages",
         examples=["Prime factorization of 68"],
         title="Agent with MCP Tools",
@@ -23,5 +27,7 @@ try:
     )
 
     demo.launch()
+except Exception as e:
+    raise e
 finally:
-    mcp_client.disconnect()
+    mcp_client.stop()