import os
import gradio as gr
from smolagents import CodeAgent, InferenceClientModel
from smolagents.mcp_client import MCPClient

# --- Use a valid provider and API key if needed ---
model = InferenceClientModel(
    model="tiiuae/falcon-7b-instruct",
    provider="hf-inference",
    api_key=os.getenv("HF_TOKEN")  # Optional, use if your model requires auth
)

try:
    mcp_client = MCPClient({
        "url": "https://abidlabs-mcp-tools2.hf.space/gradio_api/mcp/sse"
    })

    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=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.",
    )

    demo.launch()

except Exception as e:
    raise e

# Removed mcp_client.stop() — MCPClient has no such method