import gradio as gr
from mcp import StdioServerParameters
from smolagents import MCPClient, CodeAgent


def respond(message, history):
    """Send the user's message to the MCP-powered agent and return its reply."""
    params = StdioServerParameters(command="python", args=["mcp_server.py"])
    with MCPClient(params) as tools:
        agent = CodeAgent(tools=tools, model="gpt-4")  # swap for your LLM
        answer = agent.run(message)
    history.append((message, answer))
    return history, history


with gr.Blocks() as demo:
    chat_state = gr.State([])
    chatbot = gr.Chatbot(label="Enterprise SQL Agent")

    textbox = gr.Textbox(
        show_label=False,
        placeholder="Ask: Who are my inactive Northeast customers?",
    )
    textbox.submit(respond, [textbox, chat_state], [chatbot, chat_state])

    gr.Markdown(
        """
        ### Example Prompts
        - Who are my Northeast customers who haven’t ordered in 6 months?
        - List customers sorted by last order date.
        - Find clients from the West with recent orders.

        Built with MCP, smolagents, and Gradio
        """
    )

demo.launch()