reformat FINAL ANSWER output

agent.py

@@ -24,6 +24,7 @@ import pytesseract
 import fitz  # PyMuPDF
 
 # Load environment variables from .env file
+# in HF Spaces, the .env file is saved in Variables and secrets in settings
 from dotenv import load_dotenv
 load_dotenv()
 
@@ -265,7 +266,12 @@ class MyAgent:
 
         # Use invoke() instead of run()
         out = graph.invoke(state)
-        return out["messages"][-1].content
+        last_message = out["messages"][-1].content
+        
+        # Extract only the FINAL ANSWER part
+        if "FINAL ANSWER:" in last_message:
+            return last_message.split("FINAL ANSWER:")[-1].strip()
+        return last_message.strip()
     
     def run(self, question: str, file_paths: Optional[List[str]] = None) -> str:
         return self(question, file_paths)