Update app.py

app.py

@@ -2,28 +2,25 @@ import pathlib
 import os
 import gradio as gr
 from mcp import StdioServerParameters
-from smolagents import MCPClient
-from smolagents.agents import OpenAIAgent   # <-- use OpenAI-aware agent
+from smolagents import MCPClient, CodeAgent, InferenceClientModel
 
+# Absolute path to sibling mcp_server.py
 SERVER_PATH = pathlib.Path(__file__).with_name("mcp_server.py")
 
+# Lazy‑load a Hugging Face Inference model (defaults to Qwen if no id given)
+HF_MODEL_ID = os.getenv("HF_MODEL_ID", "Qwen/Qwen2.5-Coder-8B-Instruct")  # small & free-tier friendly
+
 
 def respond(message: str, history: list):
-    """Send the user prompt to an OpenAI-backed agent with MCP tools."""
-    # Spin up the MCP tool server
+    """Route the prompt through an MCP‑enabled CodeAgent and return the answer."""
     params = StdioServerParameters(command="python", args=[str(SERVER_PATH)])
 
-    # Connect and get tools
     with MCPClient(params) as tools:
-        # 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,
-        )
+        model = InferenceClientModel(model_id=HF_MODEL_ID)
+        agent = CodeAgent(tools=tools, model=model)
         answer = agent.run(message)
 
-    # Update chat history (messages format)
+    # Update chat in OpenAI‑style messages
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": answer})
     return history, history
@@ -34,7 +31,8 @@ with gr.Blocks(title="Enterprise SQL Agent") as demo:
     chatbot = gr.Chatbot(type="messages", label="Enterprise SQL Agent")
 
     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])
 
@@ -45,7 +43,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, Gradio, and OpenAI_
+        _Powered by smolagents + MCP + Hugging Face Inference API_
         """
     )