Update app.py

app.py

@@ -1,21 +1,26 @@
+import os
 import pathlib
 import gradio as gr
 from mcp import StdioServerParameters
 from smolagents import MCPClient, CodeAgent
+from smolagents.models.openai import OpenAIChatModel   # ← add this
 
-# Absolute path to the MCP server script that lives next to this file
 SERVER_PATH = pathlib.Path(__file__).with_name("mcp_server.py")
 
+# Build an OpenAI model object once, reuse for all requests
+# Ensure OPENAI_API_KEY is set in your HF Space secrets
+openai_model = OpenAIChatModel(model="gpt-4o")  # or "gpt-4o-mini", "gpt-3.5-turbo"
+                                                # choose any model your key can access
 
 def respond(message: str, history: list):
-    """Send the user message to the MCP‑powered agent and return the reply."""
-    # Launch the MCP server as a subprocess via stdio
+    """Send the user's message to the MCP-powered agent and return reply."""
     params = StdioServerParameters(command="python", args=[str(SERVER_PATH)])
+
+    # Launch the MCP server and create a CodeAgent that uses the REAL model object
     with MCPClient(params) as tools:
-        agent = CodeAgent(tools=tools, model="gpt-4")  # change to your LLM if needed
+        agent = CodeAgent(tools=tools, model=openai_model)
         answer = agent.run(message)
 
-    # Update chat history using the new Gradio 'messages' format
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": answer})
     return history, history
@@ -23,11 +28,9 @@ def respond(message: str, history: list):
 
 with gr.Blocks(title="Enterprise SQL Agent") as demo:
     chat_state = gr.State([])
-
     chatbot = gr.Chatbot(label="Enterprise SQL Agent", type="messages")
     textbox = gr.Textbox(
-        placeholder="Ask: Who are my inactive Northeast customers?",
-        show_label=False,
+        placeholder="Ask: Who are my inactive Northeast customers?", show_label=False
     )
     textbox.submit(respond, [textbox, chat_state], [chatbot, chat_state])
 
@@ -38,7 +41,7 @@ with gr.Blocks(title="Enterprise SQL Agent") as demo:
         - List customers sorted by last order date.
         - Find clients from the West with recent orders.
 
-        _Built with MCP, smolagents, and Gradio_
+        _Built with MCP, smolagents, Gradio, and OpenAI_
         """
     )