Update app.py

app.py

@@ -64,35 +64,26 @@ async def smart_agent(questions_data: List[Dict[str, Any]]) -> List[Dict[str, An
         llm=llm,
     )
 
-    async def process_question(item):
+    answers_payload = []
+
+    for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
-        if not task_id or not question_text:
-            print(f"Skipping invalid item: {item}")
-            return {"task_id": task_id, "submitted_answer": "Invalid question."}
-    
+        if not task_id or question_text is None:
+            print(f"Skipping item with missing task_id or question: {item}")
+            continue
         try:
-            # Retry up to 3 times if connection closes
-            response = await llamaindex_agent.run(question_text)
-            print(f"Answered Task {task_id}: {response[:50]}...")
-            return {"task_id": task_id, "submitted_answer": response}
-            # for attempt in range(3):
-            #     try:
-            #         response = await llamaindex_agent.run(question_text)
-            #         print(f"Answered Task {task_id}: {response[:50]}...")
-            #         return {"task_id": task_id, "submitted_answer": response}
-            #     except Exception as e:
-            #         if "client has been closed" in str(e) and attempt < 2:
-            #             print(f"Retrying task {task_id} (attempt {attempt + 2})...")
-            #             continue
-            #         raise e
+            submitted_answer = await  llamaindex_agent(question_text)
+            answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:
-            print(f"Error on Task {task_id}: {e}")
-            return {"task_id": task_id, "submitted_answer": f"AGENT ERROR: {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}"})
     
-    #tasks = [process_question(item) for item in questions_data]
-    answers_payload = process_question(questions_data[0])
-    #answers_payload = await asyncio.gather(*tasks)
+        if not answers_payload:
+            print("Agent did not produce any answers to submit.")
+            return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
+
     
     print("✅ All tasks completed.")
     return answers_payload
@@ -149,7 +140,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 3. Run your Agent
     results_log = []
-    answers_payload = []
+    #answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
     # for item in questions_data:
     #     task_id = item.get("task_id")