Update app.py

app.py

@@ -4,7 +4,6 @@ from openai import OpenAI
 import gradio as gr
 import requests
 import pandas as pd
-
 from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
 
 # --- Constants ---
@@ -15,22 +14,20 @@ openai_api_key = os.getenv("OPENAI_API_KEY")
 if not openai_api_key:
     raise RuntimeError("Please set OPENAI_API_KEY in your Space secrets or env!")
 openai.api_key = openai_api_key
-client = OpenAI()  # official OpenAI client
-
-# Optional override for model id
+client = OpenAI()   # official OpenAI client
 OPENAI_MODEL_ID = os.getenv("OPENAI_MODEL_ID", "gpt-4.1")
 
-# --- Custom smolagents Model Wrapper around OpenAI client ---
+# --- Model Wrapper with __call__ ---
 class OpenAIModelWrapper:
     """
-    Adapts the OpenAI client.responses.create API to smolagents BaseModel interface.
+    Wraps the new OpenAI client.responses.create API so that
+    CodeAgent can call it directly.
     """
     def __init__(self, model_id: str, client: OpenAI):
         self.model_id = model_id
         self.client = client
 
-    def run(self, prompt: str) -> str:
-        # call the new OpenAI SDK endpoint
+    def __call__(self, prompt: str) -> str:
         resp = self.client.responses.create(
             model=self.model_id,
             input=prompt
@@ -96,7 +93,7 @@ Rules:
 - Strings: no filler words.
 """
 
-# --- Build the CodeAgent with the OpenAIModelWrapper ---
+# --- Build the CodeAgent with the callable wrapper ---
 
 llm_wrapper = OpenAIModelWrapper(model_id=OPENAI_MODEL_ID, client=client)