Update app.py

app.py

@@ -40,54 +40,54 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     results_log = []
     answers_payload = []
+    loop = asyncio.get_event_loop()
+
     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:
             continue
 
-    try:
-    # Full prompt with required system instructions
-        system_prompt = (
-            "You are a general AI assistant. I will ask you a question. "
-            "Report your thoughts, and finish your answer with the following template: "
-            "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. "
-            "If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
-            "If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. "
-            "If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.\n\n"
-        )
-        full_prompt = system_prompt + f"Question: {question_text.strip()}"
-    
-        # Run agent with full prompt
-        loop = asyncio.get_event_loop()
-        agent_response = await loop.run_in_executor(None, agent, full_prompt)
-    
-        # Try to extract FINAL ANSWER
-        if "FINAL ANSWER:" in agent_response:
-            _, final_answer = agent_response.rsplit("FINAL ANSWER:", 1)
-            final_answer = final_answer.strip()
-        else:
-            final_answer = agent_response.strip()
-    
-        answers_payload.append({
-            "task_id": task_id,
-            "model_answer": final_answer
-        })
-
-        results_log.append({
-            "Task ID": task_id,
-            "Question": question_text,
-            "Submitted Answer": final_answer
-        })
-
-    except Exception as 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}"
-        })
-
+        try:
+            system_prompt = (
+                "You are a general AI assistant. I will ask you a question. "
+                "Report your thoughts, and finish your answer with the following template: "
+                "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. "
+                "If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
+                "If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. "
+                "If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.\n\n"
+            )
+            full_prompt = system_prompt + f"Question: {question_text.strip()}"
+
+            agent_result = await loop.run_in_executor(None, agent, full_prompt)
+
+            # Try to extract final answer depending on type of result
+            if isinstance(agent_result, dict) and "final_answer" in agent_result:
+                final_answer = str(agent_result["final_answer"]).strip()
+            elif isinstance(agent_result, str) and "FINAL ANSWER:" in agent_result:
+                _, final_answer = agent_result.rsplit("FINAL ANSWER:", 1)
+                final_answer = final_answer.strip()
+            else:
+                final_answer = str(agent_result).strip()
+
+            answers_payload.append({
+                "task_id": task_id,
+                "model_answer": final_answer
+            })
+
+            results_log.append({
+                "Task ID": task_id,
+                "Question": question_text,
+                "Submitted Answer": final_answer
+            })
+
+        except Exception as 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}"
+            })
 
     if not answers_payload:
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
@@ -144,4 +144,4 @@ if __name__ == "__main__":
         print(f"✅ SPACE_ID: https://huggingface.co/spaces/{space_id_startup}")
 
     print("Launching Gradio Interface...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)