Update chatbot_nuti.py

chatbot_nuti.py

@@ -1,56 +1,105 @@
-import openai
-import os
 import json
+import random
+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:
+            return json.load(f)
+    return {}
+
+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 = {
+    "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.",
+    ],
+    "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.",
+    ],
+    "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. 💚",
+    ],
+    "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.",
+    ],
+}
+
+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. ✨"
+]
+
+# ------------------ 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
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-# Archivo para guardar historial de conversaciones
-HISTORIAL_FILE = "historial_nuti.json"
-
-def cargar_historial(usuario):
-    if os.path.exists(HISTORIAL_FILE):
-        with open(HISTORIAL_FILE, "r", encoding="utf-8") as f:
-            data = json.load(f)
-        return data.get(usuario, [])
-    return []
-
-def guardar_historial(usuario, historial):
-    if os.path.exists(HISTORIAL_FILE):
-        with open(HISTORIAL_FILE, "r", encoding="utf-8") as f:
-            data = json.load(f)
-    else:
-        data = {}
-    data[usuario] = historial
-    with open(HISTORIAL_FILE, "w", encoding="utf-8") as f:
-        json.dump(data, f, indent=2, ensure_ascii=False)
-
-def buscar_respuesta(usuario, mensaje):
-    """
-    Genera una respuesta de Nuti inteligente y con memoria.
-    """
-    historial = cargar_historial(usuario)
-    
-    prompt = f"""
-Eres Nuti, un asistente nutricional amigable, motivador y educativo.
-Usuario: {usuario}
-Historial de conversación: {historial}
-Nuevo mensaje del usuario: "{mensaje}"
-
-Responde de forma clara, completa, personalizada y práctica.
-Incluye consejos de nutrición y motivación según la conversación previa.
-Mantén coherencia con lo que ya se ha dicho y no repitas consejos idénticos.
-"""
-    try:
-        respuesta = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": prompt}],
-            max_tokens=250,
-            temperature=0.9
-        )
-        texto_respuesta = respuesta['choices'][0]['message']['content'].strip()
-        historial.append(f"Usuario: {mensaje}")
-        historial.append(f"Nuti: {texto_respuesta}")
-        guardar_historial(usuario, historial)
-        return texto_respuesta
-    except Exception as e:
-        return f"🤖 Nuti dice: Lo siento, hubo un error generando la respuesta. ({e})"
+    # Respuesta por defecto
+    respuesta = random.choice(RESPUESTAS_LIBRES)
+    actualizar_memoria(email, pregunta, respuesta, estado)
+    return respuesta