Update mcp_server

mcp_server

@@ -1,51 +1,37 @@
-from mcp.server.fastmcp import FastMCP
-import sqlite3
-import json
+import gradio as gr
+from mcp import StdioServerParameters
+from smolagents import MCPClient, CodeAgent
 
-# Initialize MCP server
-mcp = FastMCP("EnterpriseData")
 
-# Create in-memory SQLite database
-conn = sqlite3.connect(":memory:", check_same_thread=False)
-cursor = conn.cursor()
+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)
+    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))
+    return history, history
 
-# Create Customers table
-cursor.execute("""
-CREATE TABLE Customers (
-    CustomerID INTEGER PRIMARY KEY AUTOINCREMENT,
-    Name TEXT,
-    Region TEXT,
-    LastOrderDate TEXT
-)
-""")
 
-# Seed sample data
-cursor.executemany(
-    "INSERT INTO Customers (Name, Region, LastOrderDate) VALUES (?, ?, ?)",
-    [
-        ("Acme Corp", "Northeast", "2024-12-01"),
-        ("Beta Inc", "West", "2025-06-01"),
-        ("Gamma Co", "Northeast", "2023-09-15"),
-        ("Delta LLC", "South", "2025-03-20"),
-        ("Epsilon Ltd", "Northeast", "2025-07-10")
-    ]
-)
-conn.commit()
+with gr.Blocks() as demo:
+    chat_history = gr.State([])
 
-# Expose SQL tool to the AI
-@mcp.tool()
-def query_database(sql: str) -> str:
-    """Executes raw SQL and returns JSON results."""
-    try:
-        cur = conn.cursor()
-        cur.execute(sql)
-        rows = cur.fetchall()
-        columns = [desc[0] for desc in cur.description or []]
-        results = [dict(zip(columns, row)) for row in rows]
-        return json.dumps(results)
-    except Exception as e:
-        return json.dumps({"error": str(e)})
+    chatbot = gr.Chatbot(label="Enterprise SQL Agent")
+    prompt_box = gr.Textbox(
+        show_label=False,
+        placeholder="Ask questions like: 'Who are my inactive Northeast customers?'",
+    )
+    prompt_box.submit(respond, [prompt_box, chat_history], [chatbot, chat_history])
 
-# Start MCP server
-if __name__ == "__main__":
-    mcp.run(transport="stdio")
+    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.
+        """
+    )
+
+demo.launch()