import gradio as gr

import os
from smolagents import (
    InferenceClientModel,
    CodeAgent,
    MCPClient
)

try:
    mcp_client = MCPClient(
        {
            "url": "https://jahnavidave-mcp-sentiment.hf.space/gradio_api/mcp/sse",
            "transport": "sse",
        }
    )

    tools = mcp_client.get_tools()

    model = InferenceClientModel(
    token=os.getenv("HF_TOKEN")
    )

    agent = CodeAgent(
        tools=[*tools],
        model=model,
        additional_authorized_imports=[
            "json",
            "ast",
            "urllib",
            "base64"
        ]
    )

    demo = gr.ChatInterface(
    fn=lambda message, history: str(agent.run(message)),
    examples=[
        "Analyze the sentiment of: I love this product"
    ],
    title="MCP Sentiment Agent",
    description="Agent using MCP tools with Ollama + Qwen",
    )

    demo.launch()

finally:
    try:
        mcp_client.disconnect()
    except:
        pass