Update mcp_server

mcp_server

@@ -3,27 +3,25 @@ from mcp import StdioServerParameters
 from smolagents import MCPClient, CodeAgent
 
 
-def respond(prompt, history=[]):
-    """Send the user prompt to the MCP‑powered agent and return the reply."""
-    # Launch the MCP server as a subprocess (stdio transport)
+def respond(message, history):
+    """Send the message to the MCP‑powered agent and return its reply."""
     params = StdioServerParameters(command="python", args=["mcp_server.py"])
     with MCPClient(params) as tools:
-        # Replace "gpt-4" with your preferred local or API‑based LLM
-        agent = CodeAgent(tools=tools, model="gpt-4")
-        result = agent.run(prompt)
-    history.append((prompt, result))
+        agent = CodeAgent(tools=tools, model="gpt-4")  # replace with your model or OpenAI wrapper
+        answer = agent.run(message)
+    history.append((message, answer))
     return history, history
 
 
 with gr.Blocks() as demo:
-    chat_history = gr.State([])
-
+    state = gr.State([])
     chatbot = gr.Chatbot(label="Enterprise SQL Agent")
-    prompt_box = gr.Textbox(
+
+    textbox = gr.Textbox(
         show_label=False,
-        placeholder="Ask questions like: 'Who are my inactive Northeast customers?'",
+        placeholder="Ask: Who are my inactive Northeast customers?",
     )
-    prompt_box.submit(respond, [prompt_box, chat_history], [chatbot, chat_history])
+    textbox.submit(respond, [textbox, state], [chatbot, state])
 
     gr.Markdown(
         """