Update app.py

app.py

@@ -11,8 +11,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-        # Use GPT-3.5-turbo for quota-friendly operation
-        model = OpenAIServerModel(model_id="gpt-3.5-turbo")
+        # Use GPT-4o; ensure your API key has access
+        model = OpenAIServerModel(model_id="gpt-4o")
         final_tool = FinalAnswerTool()
         self.agent = CodeAgent(
             model=model,
@@ -25,24 +25,14 @@ class BasicAgent:
         return self.agent.run(question)
 
 
-def extract_username(profile):
-    """
-    Extract username from the Hugging Face OAuthProfile or dict.
-    """
-    if not profile:
-        return None
-    # If profile is a dict (Gradio may return dict), extract keys
-    if isinstance(profile, dict):
-        return profile.get('username') or profile.get('name') or profile.get('login') or profile.get('id')
-    # Otherwise, assume object with attributes
-    return getattr(profile, 'username', None) or getattr(profile, 'name', None) or getattr(profile, 'login', None)
-
-
 def run_and_submit_all(profile):
-    """
-    Fetch all questions, run the BasicAgent, submit answers, and return status and DataFrame.
-    """
-    username = extract_username(profile)
+    # Extract username
+    username = None
+    if profile:
+        if isinstance(profile, dict):
+            username = profile.get('username') or profile.get('name') or profile.get('login') or profile.get('id')
+        else:
+            username = getattr(profile, 'username', None) or getattr(profile, 'name', None)
     if not username:
         return "Please login to Hugging Face with the login button.", None
 
@@ -54,7 +44,7 @@ def run_and_submit_all(profile):
     except Exception as e:
         return f"Error fetching questions: {e}", None
 
-    # Run agent on each question
+    # Run agent
     agent = BasicAgent()
     results = []
     payload = []
@@ -73,7 +63,7 @@ def run_and_submit_all(profile):
     if not payload:
         return "Agent returned no answers.", pd.DataFrame(results)
 
-    # Prepare and submit
+    # Submit
     submission = {
         'username': username,
         'agent_code': f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
@@ -96,10 +86,12 @@ def run_and_submit_all(profile):
 
 
 def test_random_question(profile):
-    """
-    Fetch a random question and return question and agent answer.
-    """
-    username = extract_username(profile)
+    username = None
+    if profile:
+        if isinstance(profile, dict):
+            username = profile.get('username') or profile.get('name')
+        else:
+            username = getattr(profile, 'username', None) or getattr(profile, 'name', None)
     if not username:
         return "Please login to Hugging Face with the login button.", ""
     try:
@@ -120,9 +112,11 @@ with gr.Blocks() as demo:
         3. Use **Run Evaluation & Submit All Answers** or **Test Random Question**.
         """
     )
+
     login = gr.LoginButton()
     run_btn = gr.Button("Run Evaluation & Submit All Answers")
     test_btn = gr.Button("Test Random Question")
+
     status_out = gr.Textbox(label="Status / Result", lines=5, interactive=False)
     table_out = gr.DataFrame(label="Full Results Table", wrap=True)
     question_out = gr.Textbox(label="Random Question", lines=3, interactive=False)