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
 
@@ -69,17 +70,18 @@ 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."""
     try:
-        # Wait for a token before proceeding
-        await token_bucket.wait(1)
-        submitted_answer = await agent(question_text)
+        # 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)
         results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
-        return {"task_id"
-: task_id, "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 token_bucket.wait(1)
+            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}