Update app.py

app.py

@@ -220,10 +220,27 @@ def extract_final_answer(agent_response):
     return "Unable to determine"
 
 # Replace BasicAgent with your SmolaAgent in the run_and_submit_all function
+import backoff
+import time
+
+# Add backoff decorator for API requests
+@backoff.on_exception(
+    backoff.expo,
+    requests.exceptions.HTTPError,
+    max_tries=5,
+    giveup=lambda e: e.response.status_code != 429,
+    factor=2
+)
+def rate_limited_request(method, url, **kwargs):
+    """Make a request with automatic backoff for rate limited requests"""
+    response = requests.request(method, url, **kwargs)
+    response.raise_for_status()
+    return response
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the SmolaAgent on them, submits all answers,
-    and displays the results.
+    and displays the results with rate limit handling.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
@@ -250,11 +267,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(agent_code)
 
-    # 2. Fetch Questions
+    # 2. Fetch Questions with rate limit handling
     print(f"Fetching questions from: {questions_url}")
     try:
-        response = requests.get(questions_url, timeout=15)
-        response.raise_for_status()
+        response = rate_limited_request("GET", questions_url, timeout=15)
         questions_data = response.json()
         if not questions_data:
              print("Fetched questions list is empty.")
@@ -279,15 +295,17 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         task_id = item.get("task_id")
         question_text = item.get("question")
         
-        # Check if there are files associated with this task
+        # Check if there are files associated with this task with rate limit handling
         try:
             files_url = f"{api_url}/files/{task_id}"
-            files_response = requests.get(files_url, timeout=15)
-            if files_response.status_code == 200:
-                # Save the file and provide its path to the agent
-                # This depends on what format the files are returned in
-                print(f"Task {task_id} has associated files")
-                # Handle files if needed
+            try:
+                files_response = rate_limited_request("GET", files_url, timeout=15)
+                if files_response.status_code == 200:
+                    print(f"Task {task_id} has associated files")
+                    # Handle files if needed
+            except Exception as e:
+                print(f"Error checking for files for task {task_id}: {e}")
+                # Continue even if file check fails
         except Exception as e:
             print(f"Error checking for files for task {task_id}: {e}")
             # Continue even if file check fails
@@ -333,11 +351,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
     print(status_update)
 
-    # 5. Submit
+    # 5. Submit with rate limit handling
     print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
     try:
-        response = requests.post(submit_url, json=submission_data, timeout=60)
-        response.raise_for_status()
+        response = rate_limited_request("POST", submit_url, json=submission_data, timeout=60)
         result_data = response.json()
         final_status = (
             f"Submission Successful!\n"