Update agent.py

agent.py

@@ -12,7 +12,16 @@ class GaiaAgent:
         self.api_url = "https://agents-course-unit4-scoring.hf.space"
 
     def clean(self, text):
-        return text.strip().replace("Final Answer:", "").replace("\n", "").replace(".", "").strip()
+        text = text.strip()
+        text = re.sub(r"Final Answer:\s*", "", text, flags=re.IGNORECASE)
+        text = re.sub(r"(?i)(the answer is|best move is|response is|it is|this is|answer:)\s*", "", text)
+        text = re.sub(r"\s*\(.*?\)", "", text)  # remove comments in brackets
+        text = re.sub(r"^\W+|\W+$", "", text)  # remove leading/trailing punctuation
+        lines = text.splitlines()
+        text = lines[0] if lines else text
+        # Handle numeric extraction if mixed in text
+        match = re.match(r"^.*?(\$?\d+(\.\d{1,2})?).*", text)
+        return match.group(1).strip() if match else text.strip()
 
     def fetch_file(self, task_id):
         try:
@@ -26,7 +35,7 @@ class GaiaAgent:
         res = self.client.chat.completions.create(
             model=model,
             messages=[
-                {"role": "system", "content": "You are a precise assistant. Think step by step and return only the final answer in the correct format."},
+                {"role": "system", "content": "You are a precise assistant. Think step by step and return only the final answer in the correct format. Avoid any explanation."},
                 {"role": "user", "content": prompt + "\n\nFinal Answer:"}
             ],
             temperature=0.0,