Update app.py

app.py

@@ -22,8 +22,11 @@ class BasicAgent:
         question = [HumanMessage(content=question)]
         final_answer = self.agent.invoke({"messages": question})
         print(f"Agent returning fixed answer: {final_answer}")
-        return final_answer["messages"][-1].content[14:]
-
+        text = final_answer["messages"][-1].content
+        if "FINAL ANSWER: " in text:
+            return text.partition("FINAL ANSWER: ")[2].strip()
+        return text.strip()
+        
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,