Update app.py

app.py

@@ -450,10 +450,7 @@ def detectar_conceitos(prompt: str):
 
         
 
-    # ---------- NUTRIÇÃO ----------
-
-       # ---------- NUTRIÇÃO ----------
-
+    # ---------- NUTRIÇÃO ---------
     # Caso específico: "sugestões de treino e alimentação" ou "sugira treino e dieta"
     if re.search(
         r"\b(sugest[õo]es|sugest[ãa]o|sugira)\s+(de\s+)?treino\s+e\s+(alimenta[cç][aã]o|dieta)\b",
@@ -960,11 +957,11 @@ def gerar_plano(idade, sexo, peso, altura, atividade, objetivo, intensidade, n_r
         plano_diario = gerar_plano_diario(plano, n_refeicoes=n_refeicoes, alergias=alergias)
 
         resumo = (
-            f"📊 **Plano Nutricional**\n\n"
+            f"📊 **Plano Nutricional**\n"
             f"- Calorias alvo: {plano.calorias_alvo} kcal\n"
             f"- Proteína: {plano.proteina_g} g\n"
             f"- Carboidratos: {plano.carboidratos_g} g\n"
-            f"- Gorduras: {plano.gorduras_g} g\n\n"
+            f"- Gorduras: {plano.gorduras_g} g\n"
             f"ℹ️ {plano.nota}\n"
         )
 
@@ -1197,6 +1194,7 @@ def extrair_dados_usuario(prompt_norm: str):
         dados["alergias"] = list(set(alergias_encontradas))
 
     dados["lesoes"]=detectar_lesoes(prompt_norm)
+    dados["intencao_treino"]=detectar_objetivos(prompt_norm, [])
 
     return dados
    
@@ -1231,7 +1229,7 @@ def formatar_resposta_humana(resposta_final: dict) -> str:
     system_prompt = (
         "Você é um personal trainer e nutricionista virtual. "
         "Explique o resultado abaixo em português, de forma simples, motivadora "
-        "e prática, como se estivesse conversando com o aluno.\n\n"
+        "e prática, como se estivesse conversando com o aluno.\n"
     )
     dados_json = json.dumps(resposta_final, ensure_ascii=False, indent=2)
     entrada = system_prompt + dados_json
@@ -1285,11 +1283,11 @@ def responder(prompt: str):
             subtipo = treino_conceitos[0].get("subtipo", "generico")
 
             musculos_alvo = detectar_musculos(prompt_norm)
-            objetivos = detectar_objetivos(prompt_norm)
+            objetivos = dados_usuario.get('intencao_treino')
 
             # 🔸 Caso 1: Pergunta conceitual (ex: “o que é hipertrofia?”)
             if subtipo == "pergunta":
-                prompt_llm = f"Você é um treinador experiente. Responda de forma breve e direta:\n\n\"{prompt_norm}\""
+                prompt_llm = f"Você é um treinador experiente. Responda de forma breve e direta:\n\"{prompt_norm}\""
                 inputs = tokenizer(prompt_llm, return_tensors="pt")
                 output = model.generate(**inputs, max_new_tokens=256, do_sample=False)
                 resposta = tokenizer.decode(output[0], skip_special_tokens=True)