Update app.py

app.py

@@ -29,6 +29,7 @@ token_bucket = Limiter(rate=TOKEN_BUCKET_REFILL_RATE, capacity=TOKEN_BUCKET_CAPA
 async def check_n_load_attach(session: aiohttp.ClientSession, task_id: str, api_url: str = DEFAULT_API_URL) -> Optional[str]:
     file_url = f"{api_url}/files/{task_id}"
     try:
+
         async with session.get(file_url, timeout=15) as response:
             if response.status == 200:
                 # Get filename from Content-Disposition
@@ -185,24 +186,21 @@ 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")
             if not task_id or question_text is None:
                 print(f"Skipping item with missing task_id or question: {item}")
                 continue
-#            if "1ht" in question_text.lower() or "attach" in question_text.lower():
             file_path = await check_n_load_attach(session, task_id)
-            result = await process_question(agent, question_text, task_id, file_path, results_log)
+            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:
             if result:
                 answers_payload.append(result)
-            else:
-                print(f"Skipping not related question: {task_id}")
-                results_log.append({
-                    "Task ID": task_id,
-                    "Question": question_text,
-                    "Submitted Answer": "Question skipped - not related"
-                })
 
         if not answers_payload:
             print("Agent did not produce any answers to submit.")