Update app.py

app.py

@@ -61,11 +61,11 @@ from smolagents.mcp_client import MCPClient
 from transformers import AutoModelForCausalLM
 import torch
 
-async def main():
+try:
     mcp_client = MCPClient({"url": "https://captain-awesome-alquranchapters.hf.space/gradio_api/mcp/sse"})
     await mcp_client.connect()  # or use async context manager if supported
 
-    tools = await mcp_client.get_tools()
+    tools = mcp_client.get_tools()
 
     model_id = "unsloth/Llama-3.2-1B"
     model = AutoModelForCausalLM.from_pretrained(
@@ -73,6 +73,7 @@ async def main():
         torch_dtype=torch.bfloat16,
         device_map="auto"
     )
+    
 
     agent = CodeAgent(tools=tools, model=model)
 
@@ -81,15 +82,15 @@ async def main():
         return str(response)
 
     demo = gr.ChatInterface(
-        fn=gradio_fn,
+        fn=lambda message, history: str(agent.run(message)),
         type="messages",
         title="Agent with MCP Tools",
         description="This is a simple agent that uses MCP tools to get chapters of the Quran.",
     )
+    
+    demo.launch()
 
-    demo.launch(share=True)
-
-    await mcp_client.disconnect()
+    # demo.launch(share=True)
 
-if __name__ == "__main__":
-    asyncio.run(main())
+finally:
+    mcp_client.disconnect()