Update app.py

app.py

@@ -2,64 +2,61 @@ import gradio as gr
 import datetime
 from transformers import pipeline
 from registro import guardar_registro, cargar_registro
+from chatbot_nuti import buscar_respuesta
 import random
 
-# Cargamos estado personal
+# Cargar estado local
 estado = cargar_registro()
-
-# Aseguramos claves básicas
 for k, v in {
     "meta_diaria": 0, "restantes": 0, "quemas": 0, "puntos": 0,
     "registro": [], "badges": [], "emocion": "", "nivel": 1
 }.items():
     estado.setdefault(k, v)
 
-# Clasificación de comida
+# Modelo de comida
 modelo = pipeline("image-classification", model="nateraw/food")
 
-# Calorías promedio (ejemplo)
+# Calorías promedio
 calorias_promedio = {
     "pizza": 285, "hamburger": 354, "apple": 95, "banana": 105,
     "salad": 150, "pasta": 350, "rice": 200, "ice cream": 210,
     "donut": 195, "fish": 206, "chicken": 239, "broccoli": 55
 }
 
-# Colores por grupo
-colores_alimentos = {
-    "verde": ["broccoli", "lettuce", "spinach", "avocado"],
-    "rojo": ["apple", "strawberry", "meat", "tomato"],
-    "amarillo": ["banana", "corn", "rice", "pasta"],
-    "blanco": ["chicken", "fish", "yogurt"]
-}
-
-# Actividades y gasto calórico por hora (aprox)
+# Actividades
 actividades = {
-    "Correr": 700,
-    "Caminar": 250,
-    "Fútbol": 600,
-    "Natación": 550,
-    "Básquetbol": 500,
-    "Voleibol": 400,
-    "Yoga": 200,
+    "Correr": 700, "Caminar": 250, "Fútbol": 600,
+    "Natación": 550, "Básquetbol": 500, "Voleibol": 400, "Yoga": 200,
 }
 
-# Función: Calcular meta diaria
+# Base de nutriólogos por país/estado/ciudad
+nutriologos = [
+    {"nombre": "Dra. Ana López", "pais": "México", "estado": "Sonora", "ciudad": "Hermosillo",
+     "especialidad": "Nutrición deportiva", "direccion": "Av. Universidad 200", "contacto": "Instagram: @dra.analpz"},
+    {"nombre": "Lic. Jorge Ramírez", "pais": "México", "estado": "CDMX", "ciudad": "Coyoacán",
+     "especialidad": "Control de peso", "direccion": "Centro NutriVida, Av. Pacífico 300", "contacto": "Tel: +52 55 1234 5678"},
+    {"nombre": "Dra. Camila Torres", "pais": "Colombia", "estado": "Antioquia", "ciudad": "Medellín",
+     "especialidad": "Nutrición infantil", "direccion": "Cra 40 #23", "contacto": "Email: camilatorresnutri@gmail.com"},
+    {"nombre": "Dr. Luis Fernández", "pais": "España", "estado": "Madrid", "ciudad": "Madrid",
+     "especialidad": "Nutrición clínica", "direccion": "Calle Mayor 45", "contacto": "Web: www.drfernandez.es"},
+    {"nombre": "Lic. Sofía Rivera", "pais": "México", "estado": "Jalisco", "ciudad": "Guadalajara",
+     "especialidad": "Nutrición general y recetas personalizadas", "direccion": "Av. Chapultepec 100", "contacto": "IG: @sofiariveranutri"},
+]
+
+# ------------------------------- FUNCIONES -------------------------------
+
 def calcular_calorias(edad, peso, altura, sexo, actividad):
     if sexo.lower() == "hombre":
         bmr = 88.36 + (13.4 * peso) + (4.8 * altura) - (5.7 * edad)
     else:
         bmr = 447.6 + (9.2 * peso) + (3.1 * altura) - (4.3 * edad)
-    niveles = {
-        "Sedentario": 1.2, "Ligero": 1.375, "Moderado": 1.55,
-        "Activo": 1.725, "Muy activo": 1.9
-    }
+    niveles = {"Sedentario": 1.2, "Ligero": 1.375, "Moderado": 1.55, "Activo": 1.725, "Muy activo": 1.9}
     tdee = bmr * niveles.get(actividad, 1.2)
     estado["meta_diaria"] = round(tdee)
     estado["restantes"] = round(tdee)
     guardar_registro(estado)
     return f"🔥 Tu meta calórica diaria es de {round(tdee)} cal."
 
-# Analizar foto
 def analizar_foto(img):
     if estado["meta_diaria"] == 0:
         return "⚠️ Primero calcula tu meta diaria."
@@ -67,24 +64,28 @@ def analizar_foto(img):
     nombre = resultado[0]['label'].lower()
     calorias = calorias_promedio.get(nombre, 250)
     estado["restantes"] = max(0, estado["restantes"] - calorias)
-    estado["registro"].append({"fecha": str(datetime.datetime.now()), "tipo": "comida", "nombre": nombre, "calorias": calorias})
+    estado["registro"].append({
+        "fecha": str(datetime.datetime.now()), "tipo": "comida",
+        "nombre": nombre, "calorias": calorias
+    })
     estado["puntos"] += 10
     guardar_registro(estado)
     return f"🍽️ {nombre.title()} ≈ {calorias} cal. Restan {estado['restantes']} / {estado['meta_diaria']} cal."
 
-# Registrar ejercicio
 def registrar_actividad(tipo, duracion):
     if tipo not in actividades:
         return "Selecciona una actividad válida."
     quemadas = int(actividades[tipo] * (duracion / 60))
     estado["restantes"] += quemadas
     estado["quemas"] += quemadas
-    estado["registro"].append({"fecha": str(datetime.datetime.now()), "tipo": "actividad", "nombre": tipo, "calorias": quemadas})
+    estado["registro"].append({
+        "fecha": str(datetime.datetime.now()), "tipo": "actividad",
+        "nombre": tipo, "calorias": quemadas
+    })
     estado["puntos"] += 15
     guardar_registro(estado)
     return f"🏃‍♂️ {tipo} durante {duracion} min ≈ {quemadas} cal quemadas. Restantes: {estado['restantes']}."
 
-# Coach emocional
 def coach_emocional(emocion):
     estado["emocion"] = emocion
     sugerencias = {
@@ -96,7 +97,6 @@ def coach_emocional(emocion):
     guardar_registro(estado)
     return sugerencias.get(emocion.lower(), "💬 Cuéntame más sobre cómo te sientes.")
 
-# Misiones diarias
 misiones = [
     "🥤 Bebe 8 vasos de agua.",
     "🍎 Come 3 frutas distintas.",
@@ -111,24 +111,28 @@ def generar_mision():
     guardar_registro(estado)
     return f"🎯 Tu misión de hoy: {mision}"
 
-# Chatbot Nuti
 def chatbot_nuti(pregunta):
-    respuestas = {
-        "hambre": "Si tienes hambre, prueba una manzana o zanahorias. Sacian sin muchas calorías.",
-        "cena": "Una cena ligera con proteína y vegetales es ideal.",
-        "subir de peso": "Agrega frutos secos, aguacate y avena a tus comidas.",
-        "bajar de peso": "Evita bebidas azucaradas y prioriza proteína magra y fibra.",
-        "agua": "Bebe al menos 2 litros al día 💧."
-    }
-    for clave, resp in respuestas.items():
-        if clave in pregunta.lower():
-            return f"🤖 Nuti dice: {resp}"
-    return "🤖 Nuti: No estoy seguro, pero intenta comer balanceado y escuchar a tu cuerpo 💚."
+    respuesta = buscar_respuesta(pregunta)
+    return f"🤖 Nuti dice: {respuesta}"
+
+# NUEVA FUNCIÓN: Buscar nutriólogos
+def buscar_nutriologo(pais, estado, ciudad):
+    resultados = [n for n in nutriologos if
+                  n["pais"].lower() == pais.lower() and
+                  n["estado"].lower() == estado.lower() and
+                  n["ciudad"].lower() == ciudad.lower()]
+    if not resultados:
+        return "😔 No se encontraron nutriólogos en esa zona aún."
+    salida = "👩‍⚕️ Nutriólogos disponibles:\n"
+    for n in resultados:
+        salida += f"\n• {n['nombre']} — {n['especialidad']}\n  📍 {n['direccion']}\n  📞 {n['contacto']}\n"
+    return salida
+
+# ------------------------------- INTERFAZ -------------------------------
 
-# Interfaz principal
 with gr.Blocks() as demo:
     gr.Markdown("# 🥦 NutriBot Ultra — Tu Asistente de Bienestar Integral 🌈")
-    gr.Markdown("IA que une nutrición, emociones, y diversión 💪🍎")
+    gr.Markdown("IA que une nutrición, emociones y diversión 💪🍎")
 
     with gr.Tab("🧍 Tu perfil"):
         edad = gr.Number(label="Edad (años)")
@@ -164,6 +168,14 @@ with gr.Blocks() as demo:
         salida_mision = gr.Textbox(label="Tu reto de hoy")
         btn_mision.click(generar_mision, outputs=salida_mision)
 
+    with gr.Tab("👩‍⚕️ Nutriólogos recomendados"):
+        pais = gr.Textbox(label="País")
+        estado_ = gr.Textbox(label="Estado o provincia")
+        ciudad = gr.Textbox(label="Ciudad")
+        btn_buscar = gr.Button("Buscar nutriólogos")
+        salida_nutri = gr.Textbox(label="Resultados")
+        btn_buscar.click(buscar_nutriologo, inputs=[pais, estado_, ciudad], outputs=salida_nutri)
+
     with gr.Tab("🤖 Nuti - Tu asistente"):
         pregunta = gr.Textbox(label="Hazle una pregunta a Nuti 🍓")
         btn_chat = gr.Button("Preguntar")