import gradio as gr
import os

from mcp import StdioServerParameters
from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClient 


try:
    mcp_client = MCPClient(
        {"url": "https://nlarchive-mcp-sentiment.hf.space/gradio_api/mcp/sse"} # This is the MCP Server we created in the previous section
    )
    tools = mcp_client.get_tools()
    model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
    agent = CodeAgent(tools=[*tools], model=model)

    demo = gr.ChatInterface(
        fn=lambda message, history: str(agent.run(message)),
        type="messages",
        examples=["I had a terrible experience with this product. It stopped working after just two days, and the customer service was completely unhelpful. The instructions were confusing, and the build quality felt cheap and flimsy. Honestly, I regret wasting my money on this. Would not recommend to anyone."],
        title="Agent Sentiment Analysis",
        description="This is a simple agent use an MCP server to analyze sentiment",
    )

    demo.launch()
finally:
    mcp_client.disconnect()