Update app.py

app.py

@@ -11,16 +11,23 @@ from smolagents import DuckDuckGoSearchTool, CodeAgent, LiteLLMModel
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+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 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."""
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
         print("Initializing BasicAgent.")
-        model_id = "gemini/gemini-2.0-flash-live-001" 
+        model_id = "gemini/gemini-2.0-flash-lite" 
 
         self.model = LiteLLMModel(
             model_id=model_id,
             api_key=os.environ.get('GOOGLE_API_KEY')
+            system_prompt=SYSTEM_PROMPT
         )
 
         search_tool = DuckDuckGoSearchTool()