Update app.py

app.py

@@ -136,15 +136,20 @@ class BasicAgent:
         for event in events:
             listMessages.extend(event["messages"])
 
-        print(f"messages: {listMessages}/n")
+        print(f"messages: {listMessages}\n")
 
         answer = listMessages[-1].content
         
         #answer = supervisor.invoke(messages)
-        print(f"Answer: {answer}/n")
+        print(f"Answer: {answer}\n")
+
+        start_index = answer.find("FINAL ANSWER: ")
+        substring=""
+        if start_index != -1:
+            substring = answer[start_index:start_index+14] 
         
-        final_answer = answer.removeprefix("FINAL ANSWER: ") #re.search(r"FINAL ANSWER:\s*(.*)", answer)
-        print(f"Agent returning answer: {final_answer}/n")
+        final_answer = substring #answer.removeprefix("FINAL ANSWER: ") #re.search(r"FINAL ANSWER:\s*(.*)", answer)
+        print(f"Agent returning answer: {final_answer}\n")
         return final_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):