Update app.py

app.py

@@ -36,14 +36,27 @@ class BasicAgent:
             max_steps=12,
             additional_authorized_imports=["pandas", "numpy"],
         )
+GAIA_SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
+FINAL ANSWER: [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.
+If you are asked for a number, don't use commas to write your number, and don't use units such as $ or percent sign unless specified otherwise.
+If you are asked for a string, don't use articles, don't use abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
+If you are asked for a comma separated list, apply the above rules depending on whether the element to be put in the list is a number or a string."""
 
-    def __call__(self, question: str) -> str:
-        try:
-            return str(self.agent.run(f"{SYSTEM_HINT}\n\nQuestion: {question}"))
-        except Exception as e:
-            print(f"Agent failed on question: {e}")
-            return "unable to answer"
 
+def __call__(self, question: str) -> str:
+    try:
+        raw = str(self.agent.run(f"{GAIA_SYSTEM_PROMPT}\n\nQuestion: {question}"))
+        # Извлекаем только то, что после "FINAL ANSWER:"
+        if "FINAL ANSWER:" in raw:
+            answer = raw.split("FINAL ANSWER:")[-1].strip()
+            # Убираем точку в конце, если есть
+            answer = answer.rstrip(".").strip()
+            return answer
+        return raw.strip()
+    except Exception as e:
+        print(f"Agent failed: {e}")
+        return "unable to answer"