Update app.py

app.py

@@ -1,26 +1,29 @@
-import os
 import pathlib
+import os
 import gradio as gr
 from mcp import StdioServerParameters
-from smolagents import MCPClient, CodeAgent
-from smolagents.models.openai import OpenAIChatModel   # ← add this
+from smolagents import MCPClient
+from smolagents.agents import OpenAIAgent   # <-- use OpenAI-aware agent
 
 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's message to the MCP-powered agent and return reply."""
+    """Send the user prompt to an OpenAI-backed agent with MCP tools."""
+    # Spin up the MCP tool server
     params = StdioServerParameters(command="python", args=[str(SERVER_PATH)])
 
-    # Launch the MCP server and create a CodeAgent that uses the REAL model object
+    # Connect and get tools
     with MCPClient(params) as tools:
-        agent = CodeAgent(tools=tools, model=openai_model)
+        # You can pass model name; defaults to gpt-4o if not provided
+        agent = OpenAIAgent(
+            tools=tools,
+            model=os.getenv("OPENAI_MODEL", "gpt-4o-preview"),
+            temperature=0.3,
+        )
         answer = agent.run(message)
 
+    # Update chat history (messages format)
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": answer})
     return history, history
@@ -28,7 +31,8 @@ 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")
+    chatbot = gr.Chatbot(type="messages", label="Enterprise SQL Agent")
+
     textbox = gr.Textbox(
         placeholder="Ask: Who are my inactive Northeast customers?", show_label=False
     )