Update app.py

app.py

@@ -4,6 +4,7 @@ from openai import OpenAI
 import gradio as gr
 import requests
 import pandas as pd
+
 from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
 
 # --- Constants ---
@@ -14,25 +15,9 @@ 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
-OPENAI_MODEL_ID = os.getenv("OPENAI_MODEL_ID", "gpt-4.1")
-
-# --- Model Wrapper with __call__ ---
-class OpenAIModelWrapper:
-    """
-    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
+client = OpenAI()
 
-    def __call__(self, prompt: str) -> str:
-        resp = self.client.responses.create(
-            model=self.model_id,
-            input=prompt
-        )
-        return resp.output_text
+OPENAI_MODEL_ID = os.getenv("OPENAI_MODEL_ID", "gpt-4.1")
 
 # --- Tool Definitions ---
 
@@ -67,7 +52,7 @@ search_tool    = DuckDuckGoSearchTool()
 wiki_tool      = wikipedia_search
 summarize_tool = summarize_query
 
-# --- ReACT + Scratchpad + Auto-Retry Prompt ---
+# --- ReACT + Scratchpad + Auto-Retry Instruction Prompt ---
 
 instruction_prompt = """
 You are a ReACT agent with three tools:
@@ -93,16 +78,31 @@ Rules:
 - Strings: no filler words.
 """
 
-# --- Build the CodeAgent with the callable wrapper ---
+# --- Model wrapper to satisfy CodeAgent's expected interface ---
+
+class OpenAIModelWrapper:
+    def __init__(self, model_id: str, client: OpenAI):
+        self.model_id = model_id
+        self.client = client
+
+    def __call__(self, prompt: str, **kwargs) -> str:
+        # ignore kwargs like stop_sequences, temperature, etc.
+        resp = self.client.responses.create(
+            model=self.model_id,
+            input=prompt
+        )
+        return resp.output_text
 
 llm_wrapper = OpenAIModelWrapper(model_id=OPENAI_MODEL_ID, client=client)
 
+# --- Build the CodeAgent ---
+
 smart_agent = CodeAgent(
     tools=[search_tool, wiki_tool, summarize_tool],
     model=llm_wrapper
 )
 
-# --- Wrap in BasicAgent for Gradio ---
+# --- BasicAgent wrapper for Gradio ---
 
 class BasicAgent:
     def __init__(self):
@@ -116,19 +116,19 @@ class BasicAgent:
         except Exception as e:
             return f"AGENT ERROR: {e}"
 
-# --- Gradio & Submission Logic ---
+# --- Gradio / Submission Logic ---
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not profile:
         return "Please log in to Hugging Face.", None
-    username = profile.username
-    space_id = os.getenv("SPACE_ID", "")
-    agent = BasicAgent()
+    username   = profile.username
+    space_id   = os.getenv("SPACE_ID", "")
+    agent      = BasicAgent()
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
     # 1. Fetch questions
     try:
-        resp = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
+        resp      = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
         resp.raise_for_status()
         questions = resp.json() or []
     except Exception as e: