cooldown on error + smarter again

app.py

@@ -99,27 +99,30 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
                 
             submitted_answer = agent(question_text, additional_data=additional_data)
                 
-            # cooldown_attempts = 2
-            # while cooldown_attempts > 0:
-            #     try:
-            #         submitted_answer = agent(question_text, additional_data=additional_data)
-            #         break
-            #     except RateLimitError as e:
-            #         print(f"cooldown due to {e}")
-            #         sleep(61)
-            #         cooldown_attempts -= 1
-            #         if cooldown_attempts == 0:
-            #             raise e
+            cooldown_attempts = 2
+            while cooldown_attempts > 0:
+                try:
+                    submitted_answer = agent(question_text, additional_data=additional_data)
+                    break
+                except RateLimitError as e:
+                    print(f"cooldown due to {e}")
+                    sleep(65)
+                    cooldown_attempts -= 1
+                    if cooldown_attempts == 0:
+                        raise e
 
             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})
 
-            # cooldown
-            sleep(60)
+            sleep(10)
+
         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}"})
 
+             # cooldown
+             sleep(65)
+
     if not answers_payload:
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)