MarvinRoque commited on
Commit
7869fa7
·
1 Parent(s): 20e4c51
Files changed (2) hide show
  1. app.py +1 -1
  2. nutrition.py +2 -0
app.py CHANGED
@@ -695,7 +695,7 @@ def responder(prompt: str):
695
  dias = dados
696
  try:
697
  treino_semana = gerar_split(dias=dias, budget=60, objetivos=objetivos, lesoes=lesoes)
698
- resposta_final["treino"] = treino_semana
699
  except ValueError:
700
  resposta_final["treino"] = f"Não tenho splits configurados para {dias} dias/semana."
701
  elif tipo == "isolado":
 
695
  dias = dados
696
  try:
697
  treino_semana = gerar_split(dias=dias, budget=60, objetivos=objetivos, lesoes=lesoes)
698
+ resposta_final["treino"] = json.dump(treino_semana)
699
  except ValueError:
700
  resposta_final["treino"] = f"Não tenho splits configurados para {dias} dias/semana."
701
  elif tipo == "isolado":
nutrition.py CHANGED
@@ -308,6 +308,7 @@ def sugerir_alimentos_para_refeicao(nome_refeicao, macros_ref, restricoes=None,
308
  and a["macros"]["carboidrato"] >= 10
309
  and a["nome"] not in usados
310
  and a["categoria"] != "oleaginosa"
 
311
  ]
312
  if carb_opcoes:
313
  carb_opcoes = sorted(carb_opcoes, key=lambda x: x["macros"]["calorias"])
@@ -334,6 +335,7 @@ def sugerir_alimentos_para_refeicao(nome_refeicao, macros_ref, restricoes=None,
334
  if nome_refeicao in a.get("refeicao", [])
335
  and a["macros"]["gordura"] >= 5
336
  and a["nome"] not in usados
 
337
  ]
338
 
339
  logging.debug(f"Gordura opções disponíveis antes do filtro: {len(gordura_opcoes)} itens")
 
308
  and a["macros"]["carboidrato"] >= 10
309
  and a["nome"] not in usados
310
  and a["categoria"] != "oleaginosa"
311
+ and not any(allerg in restricoes for allerg in a.get("alergias", []))
312
  ]
313
  if carb_opcoes:
314
  carb_opcoes = sorted(carb_opcoes, key=lambda x: x["macros"]["calorias"])
 
335
  if nome_refeicao in a.get("refeicao", [])
336
  and a["macros"]["gordura"] >= 5
337
  and a["nome"] not in usados
338
+ and not any(allerg in restricoes for allerg in a.get("alergias", []))
339
  ]
340
 
341
  logging.debug(f"Gordura opções disponíveis antes do filtro: {len(gordura_opcoes)} itens")