MarvinRoque commited on
Commit
ff4da06
·
1 Parent(s): 59a1971

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -652,9 +652,9 @@ def extrair_dados_usuario(prompt_norm: str):
652
  if re.search(r"\b(iniciante|começando|primeira vez)\b", prompt_norm):
653
  dados["nivel_usuario"] = "iniciante"
654
  elif re.search(r"\b(intermedi[áa]rio|m[ée]dio|regular)\b", prompt_norm):
655
- dados["nivel_usuario"] = "intermediário"
656
  elif re.search(r"\b(avançado|experiente|veterano|treino pesado há anos)\b", prompt_norm):
657
- dados["nivel_usuario"] = "avançado"
658
 
659
  return dados
660
 
@@ -755,17 +755,24 @@ def responder(prompt: str):
755
  nivel = dados_usuario.get("nivel_usuario", "").lower()
756
  atividade = dados_usuario.get("atividade", "").lower()
757
 
 
 
758
  # Mapa de budget
759
  if nivel == "iniciante":
760
  budget = 40 if dias <= 4 else 50
761
- elif nivel == "intermediário" or nivel == "intermedio":
 
762
  budget = 60 if dias < 4 else 50
763
- elif nivel == "avançado":
 
 
764
  budget = 75
765
  else:
766
  # Caso não detecte o nível, define um padrão razoável
767
  budget = 50
768
 
 
 
769
  try:
770
  treino_semana = gerar_split(
771
  sexo=dados_usuario["sexo"],
 
652
  if re.search(r"\b(iniciante|começando|primeira vez)\b", prompt_norm):
653
  dados["nivel_usuario"] = "iniciante"
654
  elif re.search(r"\b(intermedi[áa]rio|m[ée]dio|regular)\b", prompt_norm):
655
+ dados["nivel_usuario"] = "intermedario"
656
  elif re.search(r"\b(avançado|experiente|veterano|treino pesado há anos)\b", prompt_norm):
657
+ dados["nivel_usuario"] = "avancado"
658
 
659
  return dados
660
 
 
755
  nivel = dados_usuario.get("nivel_usuario", "").lower()
756
  atividade = dados_usuario.get("atividade", "").lower()
757
 
758
+ print(dias, nivel, atividade)
759
+
760
  # Mapa de budget
761
  if nivel == "iniciante":
762
  budget = 40 if dias <= 4 else 50
763
+ print('iniciante')
764
+ elif nivel == "intermediario" or nivel == "intermedio":
765
  budget = 60 if dias < 4 else 50
766
+ print('intermediario')
767
+ elif nivel == "avancado":
768
+ print('avancado')
769
  budget = 75
770
  else:
771
  # Caso não detecte o nível, define um padrão razoável
772
  budget = 50
773
 
774
+ print(budget)
775
+
776
  try:
777
  treino_semana = gerar_split(
778
  sexo=dados_usuario["sexo"],