Update app.py

app.py

@@ -46,14 +46,27 @@ class ZephyrAPI:
 
     def __call__(self, question: str, scratchpad: str = "") -> str:
         prompt = f"""<|system|>
-                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.
+                You are a helpful AI assistant that can answer questions using tools if needed.
+
+                Use the following reasoning format to answer questions:
+
+                Thought: [your reasoning]
+                Action: [tool name]
+                Action Input: [JSON-encoded input arguments for the tool]
+                
+                After you see an Observation from a tool, continue reasoning:
+                
+                Observation: [tool output]
+                Thought: [continue reasoning]
+                FINAL ANSWER: [your final answer]
+
+                Answer rules:
+                - 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.
                 
-                Available tools:
+                You have access to the following tools:
                 {self.tool_descriptions}
                 
                 <|user|>
@@ -61,6 +74,7 @@ class ZephyrAPI:
                 
                 <|assistant|>
                 {scratchpad}"""
+        
         payload = {
             "inputs": prompt,
             "parameters": {