Update chatbot_nuti.py

chatbot_nuti.py

@@ -4,9 +4,9 @@ import os
 from datetime import datetime
 
 MEMORIA_FILE = "memoria_nuti.json"
-PERFIL_FILE = "perfil_usuario.json"  # Nuevo archivo para leer los datos de salud
+PERFIL_FILE = "perfil_usuario.json"  # Archivo donde se guarda info del usuario
 
-# --- Cargar y guardar memoria de conversación --- #
+# --- Funciones de carga y guardado --- #
 def cargar_memoria():
     if os.path.exists(MEMORIA_FILE):
         with open(MEMORIA_FILE, "r", encoding="utf-8") as f:
@@ -17,7 +17,18 @@ def guardar_memoria(memoria):
     with open(MEMORIA_FILE, "w", encoding="utf-8") as f:
         json.dump(memoria, f, ensure_ascii=False, indent=2)
 
-# --- Temas clave y mensajes base --- #
+def obtener_perfil_usuario():
+    """Lee condición alimenticia y alergias desde el perfil."""
+    if os.path.exists(PERFIL_FILE):
+        with open(PERFIL_FILE, "r", encoding="utf-8") as f:
+            perfil = json.load(f)
+            return {
+                "condicion": perfil.get("condicion", "").lower(),
+                "alergias": perfil.get("alergias", "").lower()
+            }
+    return {"condicion": "", "alergias": ""}
+
+# --- Temas clave --- #
 TEMAS = {
     "agua": [
         "El agua regula todo tu cuerpo 💧: digestión, energía y concentración. Trata de tomar un vaso cada hora.",
@@ -53,20 +64,13 @@ MENSAJES_BASE = [
     "Pequeños hábitos constantes crean grandes resultados. 🌟"
 ]
 
-# --- Detectar condición del usuario (enfermedad o restricción alimenticia) --- #
-def obtener_condicion_usuario():
-    if os.path.exists(PERFIL_FILE):
-        with open(PERFIL_FILE, "r", encoding="utf-8") as f:
-            perfil = json.load(f)
-            return perfil.get("condicion", "").lower()
-    return ""
-
-
-# --- Función principal: Respuesta inteligente y contextual --- #
+# --- Respuesta inteligente y contextual --- #
 def buscar_respuesta(pregunta, historial):
     pregunta = pregunta.lower().strip()
     memoria = cargar_memoria()
-    condicion = obtener_condicion_usuario()
+    perfil = obtener_perfil_usuario()
+    condicion = perfil["condicion"]
+    alergias = perfil["alergias"]
 
     usuario = "default_user"
     if usuario not in memoria:
@@ -75,14 +79,14 @@ def buscar_respuesta(pregunta, historial):
     historial_usuario = memoria[usuario]["historial"]
     temas_previos = memoria[usuario]["temas_previos"]
 
-    # Detectar tema de la pregunta
+    # Detectar tema principal
     tema_detectado = None
     for tema in TEMAS:
         if tema in pregunta:
             tema_detectado = tema
             break
 
-    # Crear respuesta base dinámica
+    # Generar respuesta base
     if tema_detectado:
         respuesta_base = random.choice(TEMAS[tema_detectado])
         respuesta = f"{respuesta_base} Además, recuerda que cada pequeño paso que das cuenta. 💪"
@@ -97,30 +101,36 @@ def buscar_respuesta(pregunta, historial):
         else:
             respuesta = random.choice(MENSAJES_BASE)
 
-    # --- Ajustar respuesta según condición del usuario --- #
+    # --- Ajustes personalizados según condición --- #
     if condicion:
         if "diabet" in condicion:
-            respuesta += "\n\n💡 Como tienes diabetes, recuerda cuidar tus carbohidratos y preferir frutas con bajo índice glucémico (como manzana o fresa)."
+            respuesta += "\n\n💡 Como tienes diabetes, controla los azúcares simples y elige frutas de bajo índice glucémico."
         elif "hipertens" in condicion:
-            respuesta += "\n\n💡 Consejo para tu hipertensión: reduce el consumo de sal, embutidos y comida procesada. Prefiere alimentos frescos."
+            respuesta += "\n\n💡 Consejo para tu hipertensión: modera la sal y los alimentos procesados."
         elif "lactosa" in condicion:
-            respuesta += "\n\n🥛 Ojo con los lácteos: elige versiones sin lactosa o bebidas vegetales (avena, almendra o soya)."
+            respuesta += "\n\n🥛 Ojo con los lácteos: elige sin lactosa o bebidas vegetales."
         elif "celia" in condicion or "gluten" in condicion:
-            respuesta += "\n\n🌾 Recuerda evitar harinas con gluten y optar por avena sin gluten, arroz o quinoa."
+            respuesta += "\n\n🌾 Evita el gluten. Prefiere avena sin gluten, arroz o quinoa."
         elif "anemia" in condicion:
-            respuesta += "\n\n🥬 Consejo para tu anemia: incluye alimentos ricos en hierro como lentejas, espinacas y carne magra."
+            respuesta += "\n\n🥬 Agrega alimentos ricos en hierro como lentejas, espinacas y carne magra."
         elif "colesterol" in condicion:
-            respuesta += "\n\n❤️ Para controlar el colesterol, evita frituras y opta por aguacate, nueces y aceite de oliva."
+            respuesta += "\n\n❤️ Reduce frituras y elige aguacate, nueces y aceite de oliva."
         else:
-            respuesta += f"\n\n🤖 Estoy considerando tu condición alimenticia: {condicion}. Ajustaré mis recomendaciones a eso."
+            respuesta += f"\n\n🤖 Estoy tomando en cuenta tu condición: {condicion}."
 
-    # Guardar historial actualizado
+    # --- Ajustes adicionales según alergias --- #
+    if alergias:
+        alergias_lista = [a.strip() for a in alergias.split(",")]
+        respuesta += "\n\n⚠️ También recordaré tus alergias para evitar riesgos: "
+        respuesta += ", ".join(alergias_lista) + "."
+        respuesta += "\nTen cuidado con etiquetas de productos y avísame si notas alguna reacción."
+
+    # Guardar historial
     historial_usuario.append({
         "fecha": str(datetime.now()),
         "pregunta": pregunta,
         "respuesta": respuesta
     })
-
     memoria[usuario]["historial"] = historial_usuario
     memoria[usuario]["temas_previos"] = temas_previos
     guardar_memoria(memoria)