Update app.py

app.py

@@ -45,7 +45,8 @@ async def fetch_questions(session: aiohttp.ClientSession, questions_url: str) ->
     except aiohttp.ClientError as e:
         print(f"Error fetching questions: {e}")
         return None
-    except Exception as e:
+    except
+ Exception as e:
         print(f"An unexpected error occurred fetching questions: {e}")
         return None
 
@@ -68,27 +69,36 @@ async def submit_answers(session: aiohttp.ClientSession, submit_url: str,
 
 async def process_question(agent, question_text: str, task_id: str, results_log: list):
     """Process a single question with global rate limiting."""
+    submitted_answer = None
     try:
         # Retry until a token is available
         while not token_bucket.consume(1):
             print(f"Rate limit reached for task {task_id}. Waiting to retry...")
-            await asyncio.sleep(60 / RATE_LIMIT)  # Wait before        submitted_answer = await agent(question_text)
+            await asyncio.sleep(60 / RATE_LIMIT)
+        submitted_answer = await agent(question_text)
         results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         return {"task_id": task_id, "submitted_answer": submitted_answer}
     except aiohttp.ClientResponseError as e:
         if e.status == 429:
             print(f"Rate limit hit for task {task_id}. Retrying after delay...")
-            await asyncio.sleep(60 / RATE_LIMIT)  # Wait before retry
+            await asyncio.sleep(60 / RATE_LIMIT)
             while not token_bucket.consume(1):
                 await asyncio.sleep(60 / RATE_LIMIT)
-            submitted_answer = await agent(question_text)
-            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
-            return {"task_id": task_id, "submitted_answer": submitted_answer}
+            try:
+                submitted_answer = await agent(question_text)
+                results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+                return {"task_id": task_id, "submitted_answer": submitted_answer}
+            except Exception as retry_e:
+                submitted_answer = f"AGENT ERROR: {retry_e}"
+                results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+                return None
         else:
-            raise
+            submitted_answer = f"AGENT ERROR: {e}"
+            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+            return None
     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}"})
+        submitted_answer = f"AGENT ERROR: {e}"
+        results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         return None
 
 async def run_and_submit_all(profile: gr.OAuthProfile | None):