MarvinRoque commited on
Commit
7cc4258
·
1 Parent(s): 5f3d4d3
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -104,7 +104,7 @@ concept_keywords = {
104
  "hipertrofia": ["hipertrofia", "ganhar massa", "massa muscular"],
105
  "força": ["forca", "força", "ganho de força", "explosividade"],
106
  "resistência": ["resistencia", "resistência", "condicionamento", "cardio"],
107
- "dieta": ["dieta", "alimentacao", "alimentação", "plano alimentar", "nutrição", "nutricao"],
108
  "suplementos": ["suplemento", "suplementos", "creatina", "whey", "proteina", "proteína", "bcaa", "pre treino", "pré treino", "pos treino", "pós treino"],
109
  "recuperação": ["recuperacao", "recuperação", "descanso", "sono", "alongamento", "aquecimento"],
110
  "lesões": ["lesao", "lesão", "lesoes", "lesões", "joelho", "ombro", "lombar", "coluna", "tendinite", "fisioterapia", "reabilitação", "reabilitacao"],
@@ -612,9 +612,6 @@ def responder(prompt: str):
612
  concept_matches = detectar_conceitos(prompt_emb, prompt_norm)
613
  keyword_bonus = min(len(concept_matches) * KEYWORD_WEIGHT, MAX_KEYWORD_BONUS)
614
 
615
- # músculos
616
- musculos_alvo = detectar_musculos(prompt_norm)
617
-
618
  # score final
619
  score = max_fitness + keyword_bonus
620
 
@@ -624,7 +621,7 @@ def responder(prompt: str):
624
  conceitos_detectados = [c for c, _ in concept_matches]
625
  intenções = {
626
  "treino": any(c in conceitos_detectados for c in ["treino", "hipertrofia", "força", "resistência", "estratégias", "cardio"]),
627
- "nutricao": any(c in conceitos_detectados for c in ["dieta", "suplementos"]),
628
  "recuperacao": any(c in conceitos_detectados for c in ["recuperação", "lesões"]),
629
  }
630
 
@@ -635,6 +632,8 @@ def responder(prompt: str):
635
  # -------------------------------
636
  if intenções["treino"]:
637
  # detectar lesões
 
 
638
  lesoes = detectar_lesoes(prompt_norm)
639
  objetivos = detectar_objetivos(prompt_norm)
640
  tipo, dados = detectar_intencao(prompt_norm, musculos_alvo)
 
104
  "hipertrofia": ["hipertrofia", "ganhar massa", "massa muscular"],
105
  "força": ["forca", "força", "ganho de força", "explosividade"],
106
  "resistência": ["resistencia", "resistência", "condicionamento", "cardio"],
107
+ "dieta": ["dieta", "alimentacao", "alimentação", "plano alimentar", "nutrição", "nutricao","emagrecer", "perder peso", "cutting", "secar"],
108
  "suplementos": ["suplemento", "suplementos", "creatina", "whey", "proteina", "proteína", "bcaa", "pre treino", "pré treino", "pos treino", "pós treino"],
109
  "recuperação": ["recuperacao", "recuperação", "descanso", "sono", "alongamento", "aquecimento"],
110
  "lesões": ["lesao", "lesão", "lesoes", "lesões", "joelho", "ombro", "lombar", "coluna", "tendinite", "fisioterapia", "reabilitação", "reabilitacao"],
 
612
  concept_matches = detectar_conceitos(prompt_emb, prompt_norm)
613
  keyword_bonus = min(len(concept_matches) * KEYWORD_WEIGHT, MAX_KEYWORD_BONUS)
614
 
 
 
 
615
  # score final
616
  score = max_fitness + keyword_bonus
617
 
 
621
  conceitos_detectados = [c for c, _ in concept_matches]
622
  intenções = {
623
  "treino": any(c in conceitos_detectados for c in ["treino", "hipertrofia", "força", "resistência", "estratégias", "cardio"]),
624
+ "nutricao": any(c in conceitos_detectados for c in ["dieta", "suplementos", "ganhar peso", "perder peso", "cutting", "bulking", "ganhar massa", "definir"]),
625
  "recuperacao": any(c in conceitos_detectados for c in ["recuperação", "lesões"]),
626
  }
627
 
 
632
  # -------------------------------
633
  if intenções["treino"]:
634
  # detectar lesões
635
+ # músculos
636
+ musculos_alvo = detectar_musculos(prompt_norm)
637
  lesoes = detectar_lesoes(prompt_norm)
638
  objetivos = detectar_objetivos(prompt_norm)
639
  tipo, dados = detectar_intencao(prompt_norm, musculos_alvo)