rewrite

app.py

@@ -1,26 +1,29 @@
 import gradio as gr
+
 from mcp.client.stdio import StdioServerParameters
 from smolagents import InferenceClientModel, CodeAgent, ToolCollection
 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": "http://localhost:7860/gradio_api/mcp/sse"},
-  )
-  tools = mcp_client.get_tools()
+    mcp_client = MCPClient(
+        ## Try this working example on the hub:
+        # {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
+        {"url": "http://localhost:7860/gradio_api/mcp/sse"}
+    )
+    tools = mcp_client.get_tools()
+
+    model = InferenceClientModel()
+    agent = CodeAgent(tools=[*tools], model=model)
 
-  model = InferenceClientModel()
-  agent = CodeAgent(tools=[*tools], model=model)
+    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.",
+    )
 
-  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",
-  )
-  demo.launch()
+    demo.launch()
 finally:
-  mcp_client.disconnect()  
+    mcp_client.disconnect()