Update app.py

app.py

@@ -30,7 +30,15 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        return answer[14:]
+        
+        # Extract just the final answer
+        if "FINAL ANSWER:" in answer:
+            # Split by "FINAL ANSWER:" and take everything after it
+            final_answer = answer.split("FINAL ANSWER:", 1)[1].strip()
+            return final_answer
+        else:
+            # If for some reason the model didn't follow the format, return as is
+            return answer
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):