Update app.py

app.py

@@ -33,14 +33,60 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 model.eval()
 
-PROMPT_TEMPLATE = """You are a GAIA final-answer extractor.
-Your job is to extract only what comes after 'Final Answer:' in the text below.
-If no 'Final Answer:' is present, infer the most likely short final answer (one line only).
+PROMPT_TEMPLATE = """You are an expert final-answer extractor working for the GAIA benchmark.you are facing a GAIA questions exam.
+the exam's results evaluator is an AI model that use the exact matching method to evaluate your answers, so you must format your Final Answer in the most consice and useful format.
+    think in logic steps to solve the GAIA benchmark problems
+Your task is to read the raw output of an AI agent and extract the final concise answer.
 
-Text:
-{text}
+### INSTRUCTIONS:
+1. Look for the  "Final Answer:" or "final answer:".
+2. Extract only what comes after it .
+3. If there is no explicit "Final Answer:", infer what you think is the final answer.
+   
+4. Format the answer according to GAIA Benchmark rules:
+   - Numbers → return as-is (e.g., 42)
+   - Dates → format as YYYY-MM-DD
+   - Lists → comma-separated, e.g., "Paris, London"
+   - Short words or names → return exactly that
+5. DO NOT include  any other text, reasoning, or punctuation beyond the final concise answer.
+DON'T take out your output of the logging or error information 
+## EXAMPLES:
+[
+  {{
+    "question": "What is the name of the fourth planet in the solar system?",
+    "agent_response": "The fourth planet from the Sun comes right after Earth. It is known for its reddish color due to the iron oxide on its surface.",
+    "reasoning": "The planet described as the 'red planet' is Mars.",
+    Final Answer: Mars
+  }},
+  {{
+    "question": "Order the following planets from smallest to largest by diameter: Earth, Mercury, Neptune, Venus.",
+    "agent_response": "The smallest is Mercury, followed by Venus, then Earth, and the largest among these four is Neptune.",
+    "reasoning": "The correct order is Mercury < Venus < Earth < Neptune.",
+    Final Answer: ["Mercury", "Venus", "Earth", "Neptune"]
+  }},
+  {{
+    "question": "In which year did NASA launch the Voyager 1 spacecraft?",
+    "agent_response": "It was launched in the late 1970s, a few weeks after Voyager 2, specifically in September 1977.",
+    "reasoning": "Launch year is 1977.",
+    Final Answer: "1977"
+  }},
+  {{
+    "question": "If a book costs $12 and a pen costs $3, what is the total cost of buying 5 books and 4 pens?",
+    "agent_response": "The books cost $60 in total, and the pens cost $12, so the total is their sum.",
+    "reasoning": "60 + 12 = 72.",
+    Final Answer: "72"
+  }},
+  {{
+    "question": "Which of the following animals is a mammal? [Crocodile, Dolphin, Falcon]",
+    "agent_response": "The crocodile is a reptile, the falcon is a bird, while the dolphin breathes air and nurses its young.",
+    "reasoning": "The only mammal in the list is the dolphin.",
+    Final Answer: "Dolphin"
+  }}
+]
+---
 
-Return only the final answer (no extra words).
+Now extract the GAIA final answer from this text:
+{text}
 """
 
 def extract_final_answer(raw: str) -> str: