fix: prompting and answers

app.py

@@ -29,11 +29,9 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        full_answer = messages['messages'][-1].content
-        if "Final answer:" in answer:
-            answer = full_answer.split("Final answer:")[-1].strip()
+        # full_answer = messages['messages'][-1].content
         # print(f"Agent response: {full_answer}")
-        return answer
+        return answer[14:]
         # return full_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
@@ -99,6 +97,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             continue
         try:
             submitted_answer = agent(question_text)
+            if "Final answer:" in submitted_answer:
+                submitted_answer = submitted_answer.split("Final answer:")[-1].strip()
             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: