Update app.py

app.py

@@ -2,37 +2,42 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-import openai
+from openai import OpenAI
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# --- Initialize OpenAI client once ---
+key = os.getenv("OPENAIAPIKEY")
+if not key:
+    raise ValueError("Please set OPENAIAPIKEY in your Space secrets!")
+client = OpenAI(api_key=key)
+
+
 # --- Simple OpenAI‑based Agent ---
 class GaiaAgent:
     def __init__(self, model_name: str = "gpt-4"):
-        # Load your API key (no underscore)
-        key = os.getenv("OPENAIAPIKEY")
-        if not key:
-            raise ValueError("Please set OPENAIAPIKEY in your Space secrets!")
-        openai.api_key = key
+        self.client = client
         self.model = model_name
 
     def __call__(self, question: str) -> str:
         """
         Sends the question to OpenAI and returns the assistant's response.
         """
-        prompt = [
-            {"role": "system", "content": "You are a helpful assistant answering GAIA Level 1 questions. Be concise and return just the final answer."},
+        messages = [
+            {"role": "system", "content": "You are a helpful assistant answering GAIA Level 1 questions. Return only the final answer."},
             {"role": "user",   "content": question}
         ]
-        resp = openai.ChatCompletion.create(
+        resp = self.client.chat.completions.create(
             model=self.model,
-            messages=prompt,
+            messages=messages,
             temperature=0.0,
             max_tokens=256,
         )
+        # The new client returns a dict‑like object
         return resp.choices[0].message.content.strip()
 
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches GAIA questions, runs GaiaAgent on each, submits answers,
@@ -102,9 +107,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     return status, pd.DataFrame(results)
 
+
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
-    gr.Markdown("# GAIA Level 1 Agent (OpenAI‑only)")
+    gr.Markdown("# GAIA Level 1 Agent (new OpenAI client)")
     gr.Markdown(
         """
         1. Add your OpenAI key (no underscore) as `OPENAIAPIKEY` in Space secrets.