Update mcp_server

mcp_server

@@ -1,37 +1,41 @@
-import gradio as gr
-from mcp import StdioServerParameters
-from smolagents import MCPClient, CodeAgent
-
-
-def respond(message, history):
-    """Send the message to the MCP-powered agent and return its answer."""
-    params = StdioServerParameters(command="python", args=["mcp_server.py"])
-    with MCPClient(params) as tools:
-        agent = CodeAgent(tools=tools, model="gpt-4")  # swap for your model
-        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.
-        """
-    )
-
-    gr.Markdown("Built with MCP, smolagents, and Gradio")
-
-demo.launch()
+from mcp.server.fastmcp import FastMCP
+import sqlite3, json
+
+mcp = FastMCP("EnterpriseData")
+
+conn = sqlite3.connect(":memory:", check_same_thread=False)
+cur = conn.cursor()
+
+cur.execute("""
+CREATE TABLE Customers (
+    CustomerID INTEGER PRIMARY KEY AUTOINCREMENT,
+    Name TEXT,
+    Region TEXT,
+    LastOrderDate TEXT
+)
+""")
+cur.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()
+
+@mcp.tool()
+def query_database(sql: str) -> str:
+    """Run a SQL query and return JSON."""
+    try:
+        cur.execute(sql)
+        cols = [d[0] for d in cur.description or []]
+        rows = [dict(zip(cols, r)) for r in cur.fetchall()]
+        return json.dumps(rows)
+    except Exception as e:
+        return json.dumps({"error": str(e)})
+
+if __name__ == "__main__":
+    mcp.run(transport="stdio")