Update app.py

app.py

@@ -8,6 +8,7 @@ from smolagents.tools import Tool
 import time
 import openai
 from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type
+import random
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -16,10 +17,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 # --- Retry Helper for Agent Call ---
-def safe_agent_call(agent, question, retries=5, wait_time=60):
+def safe_agent_call(agent, question, retries=5, wait_time=20):
     """
-    Safely call the agent with retry on rate limit errors (HTTP 429).
-    Retries up to `retries` times, waits `wait_time` seconds between attempts.
+    Helper function to safely call the agent with retry on rate limit errors (HTTP 429).
     """
     for attempt in range(retries):
         try:
@@ -34,6 +34,7 @@ def safe_agent_call(agent, question, retries=5, wait_time=60):
                 raise e
     raise Exception(f"Failed after {retries} retries due to repeated rate limit errors.")
 
+
 # --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self):
@@ -112,22 +113,16 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
         try:
             submitted_answer = safe_agent_call(agent, question_text)
-            answers_payload.append({
-                "task_id": task_id,
-                "submitted_answer": submitted_answer,
-            })
-            results_log.append({
-                "Task ID": task_id,
-                "Question": question_text,
-                "Submitted Answer": submitted_answer,
-            })
+            answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+            
+            sleep_time = random.uniform(25, 40)
+            print(f"[Rate Limit Control] Sleeping {sleep_time:.1f} seconds after answering to avoid hitting rate limits...")
+            time.sleep(sleep_time)
+    
         except Exception as e:
             print(f"Error running agent on task {task_id}: {e}")
-            results_log.append({
-                "Task ID": task_id,
-                "Question": question_text,
-                "Submitted Answer": f"AGENT ERROR: {e}",
-            })
+            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")