Update chatbot_nuti.py

chatbot_nuti.py

@@ -1,12 +1,10 @@
-import json
 import random
+import json
 import os
 from datetime import datetime
 
 MEMORIA_FILE = "memoria_nuti.json"
 
-# ------------------ FUNCIONES DE MEMORIA ------------------ #
-
 def cargar_memoria():
     if os.path.exists(MEMORIA_FILE):
         with open(MEMORIA_FILE, "r", encoding="utf-8") as f:
@@ -15,91 +13,89 @@ def cargar_memoria():
 
 def guardar_memoria(memoria):
     with open(MEMORIA_FILE, "w", encoding="utf-8") as f:
-        json.dump(memoria, f, indent=4, ensure_ascii=False)
-
-def recordar_usuario(email):
-    memoria = cargar_memoria()
-    return memoria.get(email, {"historial": [], "estado_emocional": "neutral"})
-
-def actualizar_memoria(email, entrada, respuesta, estado=None):
-    memoria = cargar_memoria()
-    if email not in memoria:
-        memoria[email] = {"historial": [], "estado_emocional": "neutral"}
-    memoria[email]["historial"].append({"pregunta": entrada, "respuesta": respuesta, "fecha": str(datetime.now())})
-    if estado:
-        memoria[email]["estado_emocional"] = estado
-    guardar_memoria(memoria)
-
-# ------------------ BASE DE CONOCIMIENTO ------------------ #
-
-TEMAS_NUTRICION = {
+        json.dump(memoria, f, ensure_ascii=False, indent=2)
+
+# -------------------- RESPUESTAS BASE --------------------
+TEMAS = {
+    "agua": [
+        "El agua es clave para todo tu cuerpo 💧, regula la digestión, la energía y hasta el ánimo.",
+        "Si te cuesta beber agua, puedes agregarle pepino, limón o menta para darle sabor. 🍋",
+        "Una buena meta: un vaso de agua antes de cada comida."
+    ],
     "desayuno": [
-        "Un desayuno equilibrado con proteína y fruta mejora tu energía y concentración. 🍳🍓",
-        "Evita los cereales ultraprocesados, opta por avena o yogurt natural con fruta.",
+        "El desayuno es el motor del día 🌞. Asegúrate de incluir proteína, fibra y fruta.",
+        "La avena con leche vegetal y frutas es una gran opción equilibrada.",
+        "Evita solo tomar café: tu cuerpo necesita energía real, no solo cafeína."
     ],
     "almuerzo": [
-        "El almuerzo debe aportar energía sin pesadez. Incluye verduras, proteína y granos integrales. 🥗",
-        "Recuerda no saltarte comidas, eso puede afectar tu metabolismo y tu ánimo.",
+        "A la hora de comer, piensa en colores: verde, naranja, rojo y blanco. 🌈",
+        "Agrega proteína magra y vegetales, y evita bebidas azucaradas.",
+        "Comer sin pantallas ayuda a digerir mejor y disfrutar los sabores."
     ],
-    "emociones": [
-        "A veces comemos para calmar emociones. Tómate un momento para respirar y sentir. 🌿",
-        "Tu cuerpo refleja tus emociones. Cuida tu mente tanto como tu alimentación. 💚",
+    "cena": [
+        "La cena debe ser ligera para descansar bien 🌙. Sopas, ensaladas o huevo son ideales.",
+        "Evita comer justo antes de dormir, deja pasar al menos 2 horas.",
+        "Si tienes hambre de noche, elige algo liviano como yogurt natural o fruta."
     ],
-    "motivacion": [
-        "No se trata de ser perfecto, sino constante. Cada día cuenta. 🌞",
-        "Tu cuerpo es un proyecto a largo plazo, no una tarea rápida.",
+    "ejercicio": [
+        "Después de entrenar, come proteína y algo de carbohidrato para recuperarte. 💪",
+        "Recuerda hidratarte antes, durante y después del ejercicio.",
+        "Un pequeño paseo después de comer mejora la digestión y el ánimo."
     ],
+    "peso": [
+        "El peso no define tu salud. Lo importante son tus hábitos y cómo te sientes. ❤️",
+        "Subir o bajar de peso lleva tiempo. Sé constante y paciente.",
+        "Evita compararte; cada cuerpo tiene su ritmo y belleza natural."
+    ],
+    "emociones": [
+        "Tus emociones influyen mucho en tu forma de comer. 🌿",
+        "Si comes por ansiedad, respira o da un paseo antes de decidir.",
+        "Comer con calma también alimenta tu mente."
+    ]
 }
 
-PALABRAS_CLAVE = {
-    "ansiedad": "Cuando sientas ansiedad, respira profundo. A veces no es hambre, sino emoción no atendida.",
-    "estrés": "Evita comer con prisa. El estrés afecta tu digestión y tu descanso.",
-    "agua": "Beber suficiente agua regula tu cuerpo y mente. Un vaso cada hora es un buen hábito. 💧",
-    "peso": "El peso no define tu valor. Enfócate en cómo te sientes y en tu bienestar integral.",
-    "deporte": "Moverte todos los días, aunque sea poco, mejora tu humor y tu digestión. 💪",
-}
-
-RESPUESTAS_LIBRES = [
-    "Cuéntame más sobre cómo te sientes con tu alimentación. 💬",
-    "Eso suena interesante. ¿Qué cambios te gustaría hacer en tu rutina?",
-    "Me gusta cómo piensas, tu enfoque hacia el bienestar es muy maduro. 🌿",
-    "Recuerda: equilibrio, no perfección. Un paso a la vez es progreso. ✨"
+MENSAJES_GENERICOS = [
+    "Tu cuerpo te habla todos los días: escúchalo sin juzgarlo. 💚",
+    "Pequeños cambios diarios crean grandes resultados.",
+    "Recuerda moverte, hidratarte y descansar. Son pilares del bienestar.",
+    "No busques perfección, busca equilibrio."
 ]
 
-# ------------------ FUNCIÓN PRINCIPAL ------------------ #
-
-def buscar_respuesta(pregunta, email="usuario_desconocido"):
-    texto = pregunta.lower()
-    memoria_usuario = recordar_usuario(email)
-    estado = memoria_usuario.get("estado_emocional", "neutral")
-
-    # Detectar tono emocional del usuario
-    if any(p in texto for p in ["triste", "mal", "solo", "ansioso", "estresado", "deprimido"]):
-        estado = "triste"
-    elif any(p in texto for p in ["feliz", "contento", "motivado", "bien", "tranquilo"]):
-        estado = "positivo"
-
-    # Buscar respuestas por tema
-    for clave, respuestas in TEMAS_NUTRICION.items():
-        if clave in texto:
-            respuesta = random.choice(respuestas)
-            actualizar_memoria(email, pregunta, respuesta, estado)
-            return respuesta
-
-    # Buscar por palabra clave específica
-    for palabra, respuesta in PALABRAS_CLAVE.items():
-        if palabra in texto:
-            actualizar_memoria(email, pregunta, respuesta, estado)
-            return respuesta
-
-    # Si hay conversación previa, intenta mantener contexto
-    if memoria_usuario["historial"]:
-        ultima_respuesta = memoria_usuario["historial"][-1]["respuesta"]
-        respuesta = f"Me alegra que sigamos hablando de esto. {random.choice(RESPUESTAS_LIBRES)} (Antes me dijiste: '{ultima_respuesta[:60]}...')"
-        actualizar_memoria(email, pregunta, respuesta, estado)
-        return respuesta
+# -------------------- FUNCIÓN PRINCIPAL --------------------
+def buscar_respuesta(usuario, pregunta):
+    pregunta = pregunta.lower().strip()
+    memoria = cargar_memoria()
+    if usuario not in memoria:
+        memoria[usuario] = {"historial": [], "temas": []}
+
+    historial = memoria[usuario]["historial"]
+    temas_previos = memoria[usuario]["temas"]
+
+    # Detectar tema nuevo
+    tema_detectado = None
+    for tema in TEMAS:
+        if tema in pregunta:
+            tema_detectado = tema
+            break
+
+    # Si se detecta tema, dar respuesta personalizada
+    if tema_detectado:
+        respuesta = random.choice(TEMAS[tema_detectado])
+        memoria[usuario]["temas"].append(tema_detectado)
+    else:
+        # Si no hay tema nuevo, responder en base al historial
+        if historial:
+            ultima = historial[-1]["pregunta"]
+            tema_ref = temas_previos[-1] if temas_previos else "nutrición"
+            respuesta = f"Como mencionaste antes sobre {tema_ref}, te recomiendo también que {random.choice(TEMAS.get(tema_ref, MENSAJES_GENERICOS))}"
+        else:
+            respuesta = random.choice(MENSAJES_GENERICOS)
+
+    historial.append({
+        "fecha": str(datetime.now()),
+        "pregunta": pregunta,
+        "respuesta": respuesta
+    })
 
-    # Respuesta por defecto
-    respuesta = random.choice(RESPUESTAS_LIBRES)
-    actualizar_memoria(email, pregunta, respuesta, estado)
+    guardar_memoria(memoria)
     return respuesta