MarvinRoque commited on
Commit
94b04a7
·
1 Parent(s): a099739

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -1328,6 +1328,7 @@ def responder(prompt: str):
1328
  # 🔸 Caso 2: Treino semanal (split)
1329
  elif subtipo == "split":
1330
  tipo, dados = detectar_intencao(prompt_norm, musculos_alvo, dados_usuario)
 
1331
  dias = dados if isinstance(dados, int) else 4 # padrão 4 dias
1332
 
1333
  # Budget dinâmico baseado no nível
@@ -1361,9 +1362,21 @@ def responder(prompt: str):
1361
  "core", "peito", "ombro", "triceps", "dorsal", "trapezio",
1362
  "biceps", "antebracos", "deltoide_frontal", "deltoide_lateral",
1363
  "deltoide_posterior", "romboides", "lombar"]
 
 
 
 
 
 
 
 
 
 
 
 
1364
  treino, custo = montar_treino(
1365
  musculos,
1366
- budget=75,
1367
  objetivos=objetivos,
1368
  lesoes=dados_usuario.get("lesoes", []),
1369
  )
 
1328
  # 🔸 Caso 2: Treino semanal (split)
1329
  elif subtipo == "split":
1330
  tipo, dados = detectar_intencao(prompt_norm, musculos_alvo, dados_usuario)
1331
+ print(dias)
1332
  dias = dados if isinstance(dados, int) else 4 # padrão 4 dias
1333
 
1334
  # Budget dinâmico baseado no nível
 
1362
  "core", "peito", "ombro", "triceps", "dorsal", "trapezio",
1363
  "biceps", "antebracos", "deltoide_frontal", "deltoide_lateral",
1364
  "deltoide_posterior", "romboides", "lombar"]
1365
+
1366
+ # Budget dinâmico baseado no nível
1367
+ nivel = dados_usuario.get("nivel_usuario", "").lower()
1368
+ if nivel == "iniciante":
1369
+ budget = 40 if dias <= 4 else 50
1370
+ elif nivel in ["intermediario", "intermedio"]:
1371
+ budget = 60 if dias < 4 else 50
1372
+ elif nivel == "avancado":
1373
+ budget = 75
1374
+ else:
1375
+ budget = 50
1376
+
1377
  treino, custo = montar_treino(
1378
  musculos,
1379
+ budget=budget,
1380
  objetivos=objetivos,
1381
  lesoes=dados_usuario.get("lesoes", []),
1382
  )