Update app.py

app.py

@@ -2,26 +2,36 @@ import gradio as gr
 from mcp import StdioServerParameters
 from smolagents import MCPClient, CodeAgent
 
-def respond(prompt, history=[]):
-    # Launch MCP subprocess
+
+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")  # Replace with your local or API model
-        result = agent.run(prompt)
-    history.append((prompt, result))
+        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_history = gr.State([])
+    chat_state = gr.State([])
     chatbot = gr.Chatbot(label="Enterprise SQL Agent")
-    textbox = gr.Textbox(show_label=False, placeholder="Ask questions like 'List inactive customers in Northeast'")
-    textbox.submit(respond, [textbox, chat_history], [chatbot, chat_history])
-    
-    gr.Markdown(\"\"\"\n
-    ### 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.
-    \"\"\")
+
+    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()