Update app.py

app.py

@@ -4,18 +4,18 @@ import requests
 import pandas as pd
 
 from tools import AnswerTool
-from smolagents import CodeAgent
+from smolagents import CodeAgent, OpenAIServerModel
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-        # Use a custom AnswerTool to handle GAIA Level 1 questions exactly
+        # Initialize CodeAgent with an OpenAI model and custom AnswerTool
+        model = OpenAIServerModel(model_id="gpt-3.5-turbo")
         answer_tool = AnswerTool()
-        # Initialize CodeAgent with only the AnswerTool, no code execution
         self.agent = CodeAgent(
-            model=None,
+            model=model,
             tools=[answer_tool],
             add_base_tools=False,
             max_steps=1,
@@ -23,7 +23,7 @@ class BasicAgent:
         )
 
     def __call__(self, question: str) -> str:
-        # Single-step execution: call the AnswerTool
+        # Execute a single step to call the AnswerTool
         return self.agent.run(question)
 
 
@@ -32,21 +32,14 @@ def run_and_submit_all(username):
     if not username:
         return "Please enter your Hugging Face username.", None
 
-        # Fetch questions
+    # Fetch questions
     try:
         resp = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
-        # Handle rate limiting specifically
         if resp.status_code == 429:
             return "Server rate limited the requests. Please wait a moment and try again.", None
         resp.raise_for_status()
         questions = resp.json()
-    except requests.exceptions.HTTPError as e:
-        # Specific HTTP errors not caught above
-        status_code = getattr(e.response, 'status_code', None)
-        if status_code == 429:
-            return "Server rate limited the requests. Please wait a moment and try again.", None
-        return f"Error fetching questions: {e}", None
-    except requests.exceptions.RequestException as e:
+    except Exception as e:
         return f"Error fetching questions: {e}", None
 
     # Run agent on all questions