Update app.py

app.py

@@ -63,9 +63,6 @@ async def check_n_load_attach(session: aiohttp.ClientSession, task_id: str, api_
                         await file.write(chunk)
                 print(f"File downloaded successfully: {local_file_path}")
                 return local_file_path
-            else:
-                print(f"No attachment found for task {task_id}")
-                return None
     except aiohttp.ClientError as e:
         print(f"Error downloading attachment for task {task_id}: {str(e)}")
         return None
@@ -108,7 +105,7 @@ async def process_question(agent, question_text: str, task_id: str, file_path: O
     """Process a single question with global rate limiting and retry logic."""
     submitted_answer = None
     max_retries = 5
-    retry_delay = 30  # Initial retry delay in seconds (increased for Gemini's 15s suggestion)
+    retry_delay = 30  # Initial retry delay in seconds
 
     for attempt in range(max_retries):
         try:
@@ -118,7 +115,7 @@ 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=120  # Increased timeout for audio processing
+                timeout=300  # 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]}...")
@@ -135,7 +132,7 @@ 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 120 seconds"
+            submitted_answer =<JsonHighlight> f"AGENT ERROR: Timeout after 300 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
@@ -183,7 +180,6 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
         answers_payload = []
         print(f"Running agent on {len(questions_data)} questions...")
 
-        tasks = []
         for item in questions_data:
             task_id = item.get("task_id")
             question_text = item.get("question")
@@ -191,11 +187,7 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
                 print(f"Skipping item with missing task_id or question: {item}")
                 continue
             file_path = await check_n_load_attach(session, task_id)
-            tasks.append(process_question(agent, question_text, task_id, file_path, results_log))
-
-        # Wait for all tasks to complete
-        results = await asyncio.gather(*tasks)
-        for result in results:
+            result = await process_question(agent, question_text, task_id, file_path, results_log)
             if result:
                 answers_payload.append(result)