Update app.py

app.py

@@ -12,11 +12,12 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
-    def __init__(self):
+    def __init__(self, openai_key):
+        self.openai_key = openai_key
         print("BasicAgent initialized.")
         # Initialize the model
         #model = HfApiModel()
-        model = OpenAIServerModel(model_id="gpt-4.1")
+        model = OpenAIServerModel(model_id="gpt-4.1", api_key=self.openai_key)
         # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
         # Initialize Agent
@@ -30,7 +31,7 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+def run_and_submit_all(profile: gr.OAuthProfile | None, openai_key: str):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -51,7 +52,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = BasicAgent(openai_key)
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None