kennethduong commited on
Commit
84b7a52
·
1 Parent(s): 18d704f

debug budget

Browse files
budget_suggestion/budget_suggestion.py CHANGED
@@ -6,14 +6,14 @@ load_dotenv()
6
 
7
  API_KEY = os.getenv('API_KEY')
8
 
9
- categories = ['Ăn uống', 'Nhà ở', 'Di chuyển', 'Giáo dục', 'Hóa đơn tiện ích']
10
 
11
  genai.configure(api_key=API_KEY)
12
  budget_model = genai.GenerativeModel(
13
  model_name='gemini-2.0-flash',
14
  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ọ'
15
  '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'
16
- '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 tiện ích: 1000000'
17
  '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'
18
  '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'
19
  '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'
 
6
 
7
  API_KEY = os.getenv('API_KEY')
8
 
9
+ categories = ['Ăn uống', 'Nhà ở', 'Di chuyển', 'Giáo dục', 'Hóa đơn & Tiện ích']
10
 
11
  genai.configure(api_key=API_KEY)
12
  budget_model = genai.GenerativeModel(
13
  model_name='gemini-2.0-flash',
14
  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ọ'
15
  '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'
16
+ '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 & Tiện ích: 1000000'
17
  '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'
18
  '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'
19
  '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'