Update app.py

app.py

@@ -1,33 +1,51 @@
 import gradio as gr
-
-from mcp.client.stdio import StdioServerParameters
-from smolagents import ToolCollection, CodeAgent
+import os
 from smolagents import CodeAgent, InferenceClientModel
 from smolagents.mcp_client import MCPClient
 
-model = InferenceClientModel()
+# Initialize the model using Hugging Face's serverless inference
+# Ensure you have an HF_TOKEN set in your Space's Secrets
+model = InferenceClientModel("meta-llama/Llama-3.2-3B-Instruct")
+
+mcp_client = None
 
 try:
+    # Initialize the MCP Client pointing to the Gradio SSE endpoint
     mcp_client = MCPClient(
         {"url": "https://abidlabs-mcp-tools2.hf.space/gradio_api/mcp/sse"}
     )
 
+    # Retrieve tools from the MCP server
     tools = mcp_client.get_tools()
-    agent = CodeAgent(tools=[*tools], model=model)
+    
+    # Create the agent with the retrieved tools
+    agent = CodeAgent(tools=[*tools], model=model, add_base_tools=True)
 
     def call_agent(message, history):
-        return str(agent.run(message))
+        try:
+            # Run the agent and return the final answer as a string
+            response = agent.run(message)
+            return str(response)
+        except Exception as error:
+            return f"Agent Error: {str(error)}"
 
+    # Set up the Gradio Chat Interface
     demo = gr.ChatInterface(
         fn=call_agent,
         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.",
+        examples=["What is the prime factorization of 2026?"],
+        title="SKT AI: Agent with MCP Tools",
+        description="This agent uses the Model Context Protocol (MCP) to access external tools.",
     )
 
-    demo.launch()
 except Exception as e:
-    raise e
-finally:
-    mcp_client.stop()
+    print(f"Initialization Error: {e}")
+    # Fallback demo in case of connection issues
+    demo = gr.Interface(fn=lambda x: f"Initialization failed: {e}", inputs="text", outputs="text")
+
+if __name__ == "__main__":
+    try:
+        demo.launch()
+    finally:
+        if mcp_client is not None:
+            mcp_client.stop()