Update app.py

app.py

@@ -25,16 +25,11 @@ class BasicAgent:
         if not api_key:
             raise ValueError("GEMINI_API_KEY environment variable not set. Please set it before running.")
         client = genai.Client(api_key=api_key)
-        model = "gemini-2.0-flash"
+        model = "gemini-2.0-flash-lite"
+         
 
         # Define a strict system instruction. Ensure that the output is minimal:
-        system_instruction_text = (
-            "You are a general AI assistant. Answer the question with a precise, final answer only. "
-            "If asked for a number, reply with only that number (without commas or units unless explicitly required). "
-            "If asked for a string, reply with minimal text (do not include articles or abbreviations). "
-            "If the answer is a comma separated list, format accordingly with minimal punctuation. "
-            "Output the answer only, with no extra commentary or explanation."
-        )
+        system_instruction_text = ("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.")
 
         # Build the conversation contents with the user query.
         contents = [