Clean up

app.py

@@ -47,7 +47,7 @@ class BasicAgent:
         # self.model = OpenAIServerModel(model_id="gpt-4o")
 
         # Define your system prompt
-        self.system_prompt = """You are a general AI assistant. I will ask you a question. Finish your answer with only: [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."""
+        self.system_prompt = """You are a general AI assistant. I will ask you a question. Finish your answer with only 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."""
         
         # Create the agent with tools
         self.agent = CodeAgent(
@@ -65,7 +65,8 @@ class BasicAgent:
 
         try:
             # Run the agent with the question
-            formatted_question = f"{self.system_prompt}\n\nQuestion: {question}\n\n"
+            formatted_question = f"{question}\n\nOnly return the exact answer, no explanation." 
+            # formatted_question = f"{self.system_prompt}\n\nQuestion: {question}\n\n"
             answer = self.agent.run(formatted_question)
             answer = str(answer).strip() if answer is not None else "No answer produced."
             # answer = self.agent.run(question)