MickMick102 commited on
Commit
3b0c9f6
·
1 Parent(s): be04392

fix: update tax deduction limit and adjust premium calculation logic in get_user_info function

Browse files
Files changed (2) hide show
  1. app.py +13 -4
  2. backend/utils.py +1 -1
app.py CHANGED
@@ -337,7 +337,7 @@ def get_user_info(current_persona, mode: str = DEFAULT_PERSONA_MODE):
337
  age = current_persona.get("Age")
338
  gender = current_persona.get("Gender")
339
  mode_value = mode if mode in PERSONA_MODES else DEFAULT_PERSONA_MODE
340
- tax_deduct = min(0.15 * income, 200000)
341
 
342
  info_lines = [
343
  f"Mode: {mode_value}",
@@ -355,16 +355,25 @@ def get_user_info(current_persona, mode: str = DEFAULT_PERSONA_MODE):
355
 
356
  has_policy = random.choice(["yes", "no"])
357
  if has_policy == "yes" and isinstance(age, int):
358
- sum_assured = income * round(random.uniform(1.5, 5), 1)
359
  plan = random.choice(["จ่าย 8 ปี", "จ่ายถึง 60"])
360
- final_premium = calculate_be_together_premium(age, sum_assured, plan, gender)
 
361
  premium_text = final_premium if isinstance(final_premium, str) else f"{final_premium:,.2f} บาท"
 
 
 
 
 
 
 
362
  info_lines.extend(
363
  [
364
  "ประวัติการถือประกัน :",
365
  "BT smart retirement:",
366
  f" - แผน: {plan}",
367
- f" - ทุนเอาประกัน: {sum_assured:,.0f} บาท",
 
368
  f" - เบี้ยประกันต่อปี: {premium_text}",
369
  ]
370
  )
 
337
  age = current_persona.get("Age")
338
  gender = current_persona.get("Gender")
339
  mode_value = mode if mode in PERSONA_MODES else DEFAULT_PERSONA_MODE
340
+ tax_deduct = min(0.15 * income, 300000)
341
 
342
  info_lines = [
343
  f"Mode: {mode_value}",
 
355
 
356
  has_policy = random.choice(["yes", "no"])
357
  if has_policy == "yes" and isinstance(age, int):
358
+ # sum_assured = income * round(random.uniform(1.5, 5), 1)
359
  plan = random.choice(["จ่าย 8 ปี", "จ่ายถึง 60"])
360
+ final_premium = random.randrange(20, int(tax_deduct * 1.5 / 1000)) * 1000
361
+ # final_premium = calculate_be_together_premium(age, sum_assured, plan, gender)
362
  premium_text = final_premium if isinstance(final_premium, str) else f"{final_premium:,.2f} บาท"
363
+ if age > 35:
364
+ purchase_year = random.randint(1,11)
365
+ elif age > 25:
366
+ purchase_year = random.randint(1,2)
367
+ else:
368
+ purchase_year = 1
369
+
370
  info_lines.extend(
371
  [
372
  "ประวัติการถือประกัน :",
373
  "BT smart retirement:",
374
  f" - แผน: {plan}",
375
+ # f" - เงินเอาประกัน: {sum_assured:,.0f} บาท",
376
+ f" - ถือมาแล้ว: {purchase_year} ปี",
377
  f" - เบี้ยประกันต่อปี: {premium_text}",
378
  ]
379
  )
backend/utils.py CHANGED
@@ -43,7 +43,7 @@ class PersonaState:
43
 
44
  nationalities = ["Thai"]
45
  occupations = ["พนักงานบริษัท", "เจ้าของธุรกิจ", "ฟรีแลนซ์", "แม่บ้าน/พ่อบ้าน","ข้าราชการ","พ่อค้า/เเม่ค้า", "Influencer"]
46
- traits = ["ใจดี", "จริงจัง", "ชอบพูดคุย", "ขี้สงสัย", "ชอบวางแผน", "รักครอบครัว", "ชอบความท้าทาย", "รอบคอบ", "หงุหงิดง่าย", "ใจร้อน","หัวอ่อน","ego"]
47
  interests = ["yes", 'no', 'neutral']
48
 
49
  gender = random.choice(genders)
 
43
 
44
  nationalities = ["Thai"]
45
  occupations = ["พนักงานบริษัท", "เจ้าของธุรกิจ", "ฟรีแลนซ์", "แม่บ้าน/พ่อบ้าน","ข้าราชการ","พ่อค้า/เเม่ค้า", "Influencer"]
46
+ traits = ["ใจดี", "จริงจัง", "ชอบพูดคุย", "ขี้สงสัย", "ชอบวางแผน", "รักครอบครัว", "ชอบความท้าทาย", "รอบคอบ", "หงุหงิดง่าย", "ใจร้อน","หัวอ่อน","ego"]
47
  interests = ["yes", 'no', 'neutral']
48
 
49
  gender = random.choice(genders)