a

app.py

@@ -245,13 +245,13 @@ def detectar_objetivo(texto: str) -> str:
     return "hipertrofia"  # padrão se nada for detectado
 
 
-def montar_treino(musculos_alvo, budget=45, objetivo="hipertrofia"):
+def montar_treino(musculos_alvo, budget=45, objetivos=["hipertrofia"]):
     treino = []
     custo_total = 0
     usados = set()
     musculos_cobertos = set()
 
-    # 1️⃣ Definição de faixas de repetições por objetivo
+    # 1️⃣ Faixas de repetições por objetivo
     faixas_reps = {
         "hipertrofia": (6, 15),
         "forca": (2, 5),
@@ -260,26 +260,25 @@ def montar_treino(musculos_alvo, budget=45, objetivo="hipertrofia"):
     }
 
     def escolher_reps(objetivo):
-        """Escolhe reps dentro da faixa do objetivo"""
         faixa = faixas_reps.get(objetivo, (8, 12))
         return random.randint(*faixa)
 
-    def add_exercicio(ex, variacao, series):
+    def add_exercicio(ex, variacao, series, objetivo_escolhido):
         nonlocal custo_total
         custo_ex = variacao["custo"] * series
-        reps = escolher_reps(objetivo)
+        reps = escolher_reps(objetivo_escolhido)
 
         if ex["nome"] in usados:
             return False
         if custo_total + custo_ex <= budget:
 
             descricao_final = variacao["descricao"]
-            # Se for explosividade e o exercício não é marcado como pliométrico
-            if objetivo == "explosividade" and not ex.get("pliometrico", False):
+            # Regras específicas para explosividade
+            if objetivo_escolhido == "explosividade" and not ex.get("pliometrico", False):
                 if ex.get("equipamento") == "peso_livre":
                     descricao_final += " (executar com carga moderada e máxima velocidade)"
                 else:
-                    return False  # descarta máquinas, elásticos etc. no treino explosivo
+                    return False  # descarta máquinas, elásticos etc.
 
             treino.append({
                 "nome": ex["nome"],
@@ -288,7 +287,8 @@ def montar_treino(musculos_alvo, budget=45, objetivo="hipertrofia"):
                 "reps": reps,
                 "custo_total": custo_ex,
                 "custo_unit": variacao["custo"],
-                "video": variacao["video"]
+                "video": variacao["video"],
+                "objetivo": objetivo_escolhido
             })
             custo_total += custo_ex
             usados.add(ex["nome"])
@@ -296,19 +296,12 @@ def montar_treino(musculos_alvo, budget=45, objetivo="hipertrofia"):
             return True
         return False
 
-    # 2️⃣ Selecionar 1 multiarticulado principal (custo = 5)
+    # 2️⃣ Multiarticulado principal
     candidatos_multi = []
     for ex in exercicios_db:
         if any(m in ex["musculos"] for m in musculos_alvo):
             for v in ex["variacoes"]:
                 if v["custo"] == 5:
-                    # Se for explosividade, só aceita pliométrico
-                    if objetivo == "explosividade":
-                        if ex.get("pliometrico", False):
-                            candidatos.append((ex, v, cobertura))
-                        elif ex.get("equipamento") == "peso_livre":
-                            candidatos.append((ex, v, cobertura, "explosivo"))
-
                     cobertura = len(set(ex["musculos"]) & set(musculos_alvo))
                     candidatos_multi.append((ex, v, cobertura))
 
@@ -317,17 +310,15 @@ def montar_treino(musculos_alvo, budget=45, objetivo="hipertrofia"):
         melhor_cobertura = candidatos_multi[0][2]
         top = [c for c in candidatos_multi if c[2] == melhor_cobertura]
         ex, variacao, _ = random.choice(top)
-        add_exercicio(ex, variacao, series=4)
+        obj_escolhido = random.choice(objetivos)
+        add_exercicio(ex, variacao, series=4, objetivo_escolhido=obj_escolhido)
 
-    # 3️⃣ Garantir pelo menos 1 exercício para cada músculo alvo
+    # 3️⃣ Garantir pelo menos 1 exercício por músculo
     for alvo in musculos_alvo:
         if alvo not in musculos_cobertos:
             candidatos = []
             for ex in exercicios_db:
                 if alvo in ex["musculos"] and ex["nome"] not in usados:
-                    # Se for explosividade, só aceita pliométrico
-                    if objetivo == "explosividade" and not ex.get("pliometrico", False):
-                        continue
                     for v in ex["variacoes"]:
                         candidatos.append((ex, v))
             if candidatos:
@@ -335,29 +326,29 @@ def montar_treino(musculos_alvo, budget=45, objetivo="hipertrofia"):
                 top_custo = candidatos[0][1]["custo"]
                 top = [c for c in candidatos if c[1]["custo"] == top_custo]
                 ex, variacao = random.choice(top)
-                add_exercicio(ex, variacao, series=3)
+                obj_escolhido = random.choice(objetivos)
+                add_exercicio(ex, variacao, series=3, objetivo_escolhido=obj_escolhido)
 
-    # 4️⃣ Completar com multiarticulados leves (custo 2–4), se sobrar budget
+    # 4️⃣ Completar treino respeitando objetivos múltiplos
     if custo_total < budget:
         candidatos = []
         for ex in exercicios_db:
             if any(m in ex["musculos"] for m in musculos_alvo) and ex["nome"] not in usados:
-                # Se for explosividade, só aceita pliométrico
-                if objetivo == "explosividade" and not ex.get("pliometrico", False):
-                    continue
                 for v in ex["variacoes"]:
                     if 2 <= v["custo"] <= 4:
                         candidatos.append((ex, v))
         candidatos.sort(key=lambda x: x[1]["custo"], reverse=True)
 
         for ex, variacao in candidatos:
-            if add_exercicio(ex, variacao, series=3):
+            obj_escolhido = random.choice(objetivos)
+            if add_exercicio(ex, variacao, series=3, objetivo_escolhido=obj_escolhido):
                 continue
             if custo_total >= budget:
                 break
 
     return treino, custo_total
 
+
 # -------------------------
 # Função principal
 # -------------------------