strip final answer

app.py

@@ -4,9 +4,28 @@ import requests
 import inspect
 import pandas as pd
 import traceback
+import re
 from huggingface_hub import login
 from tools import create_memory_safe_workflow, get_file_type, write_bytes_to_temp_dir, AgentState, extract_final_answer, run_agent
 
+import re
+
+def strip_final_answer(text):
+    """
+    Removes 'FINAL ANSWER:' (case-insensitive) and all following whitespace from the start of the string.
+    Returns the remainder of the string.
+    """
+    # The regex matches 'FINAL ANSWER:', optional colon, and all whitespace after it
+    return re.sub(r'^\s*FINAL ANSWER:\s*', '', text, flags=re.IGNORECASE)
+
+# Example usage:
+s = "FINAL ANSWER: Joe Torre"
+print(strip_final_answer(s))  # Output: Joe Torre
+
+s2 = "  FINAL ANSWER:   Jane Doe"
+print(strip_final_answer(s2))  # Output: Jane Doe
+
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -105,7 +124,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             task_id = hf_questions[r]['task_id']
             question_text = hf_questions[r]['question']
             full_answer = run_agent(agent, s)
-            submitted_answer = extract_final_answer(full_answer[-1].content)
+            submitted_answer = strip_final_answer(extract_final_answer(full_answer[-1].content))
             print(f"\n\nQuestion {r+1} Answer: {submitted_answer}\n\n")
             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})