Update app.py

app.py

@@ -14,18 +14,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         
-        openai_key_present = bool(os.getenv("OPENAI_API_KEY"))
-        print("OPENAI_API_KEY loaded:", openai_key_present)
-        if not openai_key_present:
-            raise ValueError("OPENAI_API_KEY environment variable not set.")
+        api_key = os.getenv("OPENAI_API_KEY")
+        if not api_key:
+            raise ValueError("OPENAI_API_KEY not found. Please set it as a Hugging Face secret.")
         
-        # Check if Hugging Face token is set
-        hf_token = os.getenv("HF_API_TOKEN")
-        if not hf_token:
-            raise ValueError("HF_API_TOKEN environment variable not set.")
-        
-        # Initialize model
-        model = OpenAIServerModel(model_id="gpt-3.5-turbo")
+        model = OpenAIServerModel(model_id="gpt-4o", api_key=api_key)
         
         # Initialize tools
         search_tool = DuckDuckGoSearchTool()