Update app.py

app.py

@@ -86,6 +86,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     # 3. Run your Agent
     results_log = []
     answers_payload = []
+    submitted_answer = ""
     print(f"Running agent on {len(questions_data)} questions...")
     for item in questions_data:
         task_id = item.get("task_id")
@@ -96,7 +97,12 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         try:
             input_state = {"messages": [HumanMessage(content=question_text)]}
             answer = agent.invoke(input_state)
-            submitted_answer = answer['messages'][-1].content
+            final_answer = answer['messages'][-1].content
+            if "Final Answer:" in final_answer:
+                submitted_answer = final_answer.split("Final Answer:")[1].strip()
+                print(final_answer)
+            else:
+                print("No Final Answer found in the response.")
             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: