Update app.py

app.py

@@ -109,8 +109,9 @@ async def process_question(agent, question_text: str, task_id: str, file_path: O
     submitted_answer = None
     max_retries = 5
     retry_delay = 30  # Initial retry delay in seconds
+    atimeout = 200
 
-    for attempt in range(max_retries):
+    for attempt in range(max_retries +1):
         try:
             while not token_bucket.consume(1):
                 print(f"Rate limit reached for task {task_id}. Waiting to retry...")
@@ -118,13 +119,13 @@ async def process_question(agent, question_text: str, task_id: str, file_path: O
             print(f"Processing task {task_id} (attempt {attempt + 1})...")
             submitted_answer = await asyncio.wait_for(
                 agent(question_text, file_path),
-                timeout=100  # Increased timeout for audio processing
+                timeout=atimeout  # Increased timeout for audio processing
             )
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
             print(f"Completed task {task_id} with answer: {submitted_answer[:50]}...")
             
             #Addl sleep
-            await asyncio.sleep(60)
+            await asyncio.sleep(retry_delay)
             return {"task_id": task_id, "submitted_answer": submitted_answer}
         except aiohttp.ClientResponseError as e:
             if e.status == 429:
@@ -138,15 +139,15 @@ async def process_question(agent, question_text: str, task_id: str, file_path: O
                 print(f"Failed task {task_id}: {submitted_answer}")
                 return None
         except asyncio.TimeoutError:
-            submitted_answer = f"AGENT ERROR: Timeout after 300 seconds"
+            submitted_answer = f"AGENT ERROR: Timeout after {atimeout} seconds"
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
             print(f"Failed task {task_id}: {submitted_answer}")
             return None
         except Exception as e:
             submitted_answer = f"AGENT ERROR: {e}"
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
-            print(f"Failed task {task_id}: {submitted_answer}")
-            return None
+    print(f"Failed task {task_id}: {submitted_answer}")
+    return None
 
 async def run_and_submit_all(profile: gr.OAuthProfile | None):
     """