Update app.py

app.py

@@ -1,19 +1,41 @@
 import os
 import openai
+from openai import OpenAI
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, tool
+
+from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Configure OpenAI SDK ---
+# --- Configure OpenAI SDK & Client ---
 openai_api_key = os.getenv("OPENAI_API_KEY")
 if not openai_api_key:
-    raise RuntimeError("Please set OPENAI_API_KEY in your Space secrets.")
+    raise RuntimeError("Please set OPENAI_API_KEY in your Space secrets or env!")
 openai.api_key = openai_api_key
-openai_model_id = os.getenv("OPENAI_MODEL_ID", "gpt-4.1")
+client = OpenAI()  # official OpenAI client
+
+# Optional override for model id
+OPENAI_MODEL_ID = os.getenv("OPENAI_MODEL_ID", "gpt-4.1")
+
+# --- Custom smolagents Model Wrapper around OpenAI client ---
+class OpenAIModelWrapper:
+    """
+    Adapts the OpenAI client.responses.create API to smolagents BaseModel interface.
+    """
+    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
+        resp = self.client.responses.create(
+            model=self.model_id,
+            input=prompt
+        )
+        return resp.output_text
 
 # --- Tool Definitions ---
 
@@ -51,7 +73,7 @@ summarize_tool = summarize_query
 # --- ReACT + Scratchpad + Auto-Retry Prompt ---
 
 instruction_prompt = """
-You are a ReACT agent with three tools: 
+You are a ReACT agent with three tools:
  • DuckDuckGoSearchTool(query: str)
  • wikipedia_search(page: str)
  • summarize_query(query: str)
@@ -65,21 +87,22 @@ Internally, for each question:
    Record new Observation.
 5. Thought: integrate observations.
 
-Finally, output your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
-
+Finally, output exactly one line:
+FINAL ANSWER: [your concise answer]
 
+Rules:
+- Numbers: digits only.
+- Lists: comma-separated, no extra punctuation.
+- Strings: no filler words.
 """
 
-# --- Build the CodeAgent with the same model ---
+# --- Build the CodeAgent with the OpenAIModelWrapper ---
 
-llm = LiteLLMModel(
-    model_id=openai_model_id,
-    api_key=openai_api_key
-)
+llm_wrapper = OpenAIModelWrapper(model_id=OPENAI_MODEL_ID, client=client)
 
 smart_agent = CodeAgent(
     tools=[search_tool, wiki_tool, summarize_tool],
-    model=llm
+    model=llm_wrapper
 )
 
 # --- Wrap in BasicAgent for Gradio ---
@@ -106,6 +129,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     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.raise_for_status()
@@ -113,6 +137,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Error fetching questions: {e}", None
 
+    # 2. Run agent on each question
     logs, payload = [], []
     for item in questions:
         tid = item.get("task_id")
@@ -126,6 +151,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not payload:
         return "Agent did not produce any answers.", pd.DataFrame(logs)
 
+    # 3. Submit answers
     submission = {"username": username, "agent_code": agent_code, "answers": payload}
     try:
         post = requests.post(f"{DEFAULT_API_URL}/submit", json=submission, timeout=60)
@@ -142,6 +168,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Submission Failed: {e}", pd.DataFrame(logs)
 
+# --- Gradio Interface ---
+
 with gr.Blocks() as demo:
     gr.Markdown("# SmolAgent GAIA Runner 🚀")
     gr.Markdown("""
@@ -152,7 +180,7 @@ with gr.Blocks() as demo:
 4. Click **Run Evaluation & Submit All Answers**.
 """)
     gr.LoginButton()
-    run_btn = gr.Button("Run Evaluation & Submit All Answers")
+    run_btn    = gr.Button("Run Evaluation & Submit All Answers")
     status_out = gr.Textbox(label="Status", lines=5, interactive=False)
     table_out  = gr.DataFrame(label="Questions & Answers", wrap=True)