Update app.py

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 
 import spaces
 from agent import graph
-from langchain_core.messages import HumanMessage
+from langchain_core.messages import HumanMessage, SystemMessage
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -27,7 +27,14 @@ class BasicAgent:
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         answer = graph.invoke({
-            "messages": [HumanMessage(content=question)],
+            "messages": [SystemMessage(content = """
+    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 comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+    If you are asked for a string, don't use articles, neither 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 of whether the element to be put in the list is a number or a string.
+    """), HumanMessage(content=question)],
             "file_path": None,
             "url": None,
             "task_id": None