regex for final answer

app.py

@@ -728,6 +728,17 @@ def max_object_in_video(video_url: str, object_label: str = "bird") -> str:
 
     return f"Maximum {object_label} count in a single frame: {max_count}"
 
+def extract_final_answer(output: str) -> str:
+    # Try to extract answer after [YOUR FINAL ANSWER] or Final Answer:
+    match = re.search(r"\[YOUR FINAL ANSWER\]\s*(.+)", output)
+    if match:
+        return match.group(1).strip()
+    match = re.search(r"Final Answer:\s*(.+)", output)
+    if match:
+        return match.group(1).strip()
+    # Fallback: return the whole output if no match
+    return output.strip()
+
 
 ##-- Tool Discovery ---
 # Use @tool for each function.
@@ -765,8 +776,8 @@ tool_descriptions = "\n".join(f"- {tool.name}: {tool.description}" for tool in t
 # --- System Prompt for the Agent ---
 
 system_prompt = f"""
-You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: [YOUR FINAL ANSWER].
-YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+You are a general AI assistant. I will ask you a question. Report your thoughts as required, and finish your answer with the following template: [YOUR FINAL ANSWER].
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings, and should not include any explanations, thoughts, or tool calls. Don't give output anything else except the final answer.
 First check if you can answer the question by yourself without the need for a tool. If you can answer without hallucination then answer the question directly. If you cannot, use the tools provided to answer the question.
 
 You also have access to a set of tools, which you can use to answer the question, if you can't answer the question directly. The available tools are:
@@ -944,7 +955,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         try:
             # full_prompt = f"{system_prompt}\n Input Question: {question_text}"
             # submitted_answer = agent.run(full_prompt)
-            submitted_answer = agent.run(question_text)
+            submitted_answer_raw = agent.run(question_text)
+            submitted_answer = extract_final_answer(submitted_answer_raw)
             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})
         except Exception as e: