Update app.py

app.py

@@ -1,29 +1,32 @@
 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
 
-
+# Initialize the MCP client correctly
 try:
     mcp_client = MCPClient(
-        ## Try this working example on the hub:
-        # {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
-        {"url": "https://captain-awesome-alquranchapters.hf.space/gradio_api/mcp/sse"}
+        server_parameters=StdioServerParameters(
+            url="https://captain-awesome-alquranchapters.hf.space/gradio_api/mcp/sse"
+        )
     )
+    
     tools = mcp_client.get_tools()
 
     model = InferenceClientModel()
     agent = CodeAgent(tools=[*tools], model=model)
 
+    # Define Gradio ChatInterface
     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 get chapgters of Quran.",
+        description="This is a simple agent that uses MCP tools to get chapters of the Quran.",
     )
 
     demo.launch()
+
 finally:
-    mcp_client.disconnect()
+    # Properly close the MCP client connection
+    if 'mcp_client' in locals():
+        mcp_client.disconnect()