modify suggest budget

budget_suggestion/budget_suggestion.py

@@ -6,19 +6,23 @@ load_dotenv()
 
 API_KEY = os.getenv('API_KEY')
 
+categories = ['Ăn uống', 'Nhà ở', 'Di chuyển', 'Giáo dục', 'Hóa đơn và tiện ích']
+
 genai.configure(api_key=API_KEY)
 budget_model = genai.GenerativeModel(
     model_name='gemini-2.0-flash',
     system_instruction='Bạn hãy giúp người dùng chia tỷ lệ ngân sách cho từng mục chi tiêu trong tháng này dựa vào thu nhập của họ'
     'Ví dụ: tôi có thu nhập 10000000 VND, tôi dành khoảng 3 triệu cho ăn uống, 3 triệu cho nhà ở, 2 triệu cho tiết kiệm và 2 triệu còn lại cho các mục khác'
-    'Hãy trả lời theo dạng: Ăn uống: 3000000 VND, Nhà ở: 3000000 VND, Tiết kiệm: 2000000 VND, Các mục khác: 2000000 VND'
+    'Hãy trả lời theo dạng: Ăn uống: 3000000, Nhà ở: 2500000, Di chuyển: 500000, Giáo dục: 2000000, Hóa đơn và tiện ích: 1000000'
     'Chỉ trả lời theo dạng trên mà không kèm theo từ nào khác, chỉ gồm những danh mục trên'
     'Hãy trả lời với bối cảnh là cá nhân đang sinh sống tại thành phố Hà Nội'
+    'Chú ý, mức ngân sách chi tiêu thường giao động không quá 90% thu nhập của người dùng'
 )
 
 def budget_suggestion(income):
     prompt = (
         f"Người dùng có thu nhập là {income} VND, hãy giúp họ chia tỷ lệ ngân sách cho từng mục chi tiêu trong tháng này"
+        f"Chia ngân sách theo các danh mục sau: {', '.join(categories)} "
     )
     try:
         response = budget_model.generate_content(