Update app.py

app.py

@@ -1,110 +1,230 @@
 import gradio as gr
-import json
-import os
 import datetime
 import random
+import json
+import os
+from login import iniciar_sesion, registrar_usuario
 from registro import guardar_registro, cargar_registro
 from chatbot_nuti import buscar_respuesta
 from optimizacion import resize_imagen, cargar_modelo_ligero, cache_resultado
 
-# ---- Archivos ----
-USUARIOS_FILE = "usuarios.json"
+# ---------------- SESIÓN ----------------
 SESION_FILE = "sesion.json"
 
-def cargar_usuarios():
-    if os.path.exists(USUARIOS_FILE):
-        with open(USUARIOS_FILE,"r") as f:
-            return json.load(f)
-    return {}
-
-def guardar_usuarios(data):
-    with open(USUARIOS_FILE,"w") as f:
-        json.dump(data,f,indent=4)
-
-def registrar_usuario(nombre,email,password):
-    usuarios = cargar_usuarios()
-    if email in usuarios:
-        return "⚠️ Ya existe una cuenta con ese correo."
-    usuarios[email] = {"nombre": nombre,"password": password}
-    guardar_usuarios(usuarios)
-    return "✅ Registro exitoso. ¡Ahora puedes iniciar sesión!"
-
-def iniciar_sesion(email,password):
-    usuarios = cargar_usuarios()
-    if email not in usuarios:
-        return "❌ Usuario no encontrado.", False, ""
-    if usuarios[email]["password"] != password:
-        return "🔐 Contraseña incorrecta.", False, ""
-    # Guardar sesión
-    with open(SESION_FILE,"w") as f:
-        json.dump({"usuario": email,"nombre": usuarios[email]["nombre"]}, f)
-    return f"✅ Bienvenido de nuevo, {usuarios[email]['nombre']}!", True, usuarios[email]['nombre']
-
-# ---- Estado ----
+def obtener_usuario_actual():
+    if os.path.exists(SESION_FILE):
+        with open(SESION_FILE, "r") as f:
+            data = json.load(f)
+            return data.get("usuario", None)
+    return None
+
+# ---------------- ESTADO ----------------
 estado = cargar_registro()
-for k,v in {"meta_diaria":0,"restantes":0,"quemas":0,"puntos":0,"registro":[],"badges":[],"emocion":"","nivel":1}.items():
-    estado.setdefault(k,v)
+for k, v in {
+    "meta_diaria": 0, "restantes": 0, "quemas": 0, "puntos": 0,
+    "registro": [], "badges": [], "emocion": "", "nivel": 1
+}.items():
+    estado.setdefault(k, v)
 
+# ---------------- MODELO ----------------
 modelo = cargar_modelo_ligero()
 
-# ---- Funciones ejemplo (tu código original aquí) ----
-# Solo pongo la de calcular_calorias para acortar
+# ---------------- DATOS ----------------
+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
+}
+
+actividades = {
+    "Correr": 700, "Caminar": 250, "Fútbol": 600,
+    "Natación": 550, "Básquetbol": 500, "Voleibol": 400, "Yoga": 200,
+}
+
+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"},
+]
+
+misiones = [
+    "🥤 Bebe 8 vasos de agua.",
+    "🍎 Come 3 frutas distintas.",
+    "🚶 Camina 20 minutos.",
+    "🌈 Come 4 colores de comida diferentes.",
+    "🧘 Haz 10 minutos de respiración o yoga."
+]
+
+# ---------------- 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)
+    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}
-    tdee = bmr * niveles.get(actividad,1.2)
+        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}
+    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."
 
-# ---- Interfaz ----
+def analizar_foto(img):
+    if estado["meta_diaria"] == 0:
+        return "⚠️ Primero calcula tu meta diaria."
+    img_opt = resize_imagen(img)
+    resultado = cache_resultado(str(img_opt.tobytes()), modelo, img_opt)
+    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["puntos"] += 10
+    guardar_registro(estado)
+    return f"🍽️ {nombre.title()} ≈ {calorias} cal. Restan {estado['restantes']} / {estado['meta_diaria']} cal."
+
+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["puntos"] += 15
+    guardar_registro(estado)
+    return f"🏃‍♂️ {tipo} durante {duracion} min ≈ {quemadas} cal quemadas. Restantes: {estado['restantes']}."
+
+def coach_emocional(emocion):
+    estado["emocion"] = emocion
+    sugerencias = {
+        "ansiedad": "🍌 Come un plátano o un puñado de nueces. Haz 3 respiraciones profundas.",
+        "tristeza": "🍫 Un cuadrito de chocolate oscuro puede ayudar. Escucha tu música favorita.",
+        "estrés": "🍵 Toma té verde y haz una pausa de 5 minutos.",
+        "feliz": "😄 ¡Genial! Mantén esa energía con frutas frescas o un smoothie."
+    }
+    guardar_registro(estado)
+    return sugerencias.get(emocion.lower(), "💬 Cuéntame más sobre cómo te sientes.")
+
+def generar_mision():
+    mision = random.choice(misiones)
+    estado["mision_actual"] = mision
+    guardar_registro(estado)
+    return f"🎯 Tu misión de hoy: {mision}"
+
+def chatbot_nuti(pregunta):
+    respuesta = buscar_respuesta(pregunta)
+    return f"🤖 Nuti dice: {respuesta}"
+
+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 ----------------
+usuario_actual = obtener_usuario_actual()
+
 with gr.Blocks() as demo:
-    user_name = gr.State("")
-    
-    # --- Login / Registro ---
-    with gr.Column():
-        gr.Markdown("### Bienvenido a NutriBot Ultra 🥦\nPrimero inicia sesión o crea una cuenta.")
-        
-        gr.Markdown("**Registro:**")
-        nombre_reg = gr.Textbox(label="Nombre completo")
-        email_reg = gr.Textbox(label="Correo electrónico")
-        pass_reg = gr.Textbox(label="Contraseña", type="password")
-        salida_reg = gr.Textbox(label="Resultado")
-        btn_reg = gr.Button("Crear cuenta")
-        btn_reg.click(registrar_usuario, inputs=[nombre_reg,email_reg,pass_reg], outputs=salida_reg)
-        
-        gr.Markdown("**Iniciar sesión:**")
+    if usuario_actual is None:
+        # ---------- LOGIN / REGISTRO ----------
+        gr.Markdown("# 🥗 Bienvenido a NutriBot Ultra 🌈")
+        gr.Markdown("Por favor inicia sesión o crea una cuenta para continuar")
+
         email_login = gr.Textbox(label="Correo electrónico")
         pass_login = gr.Textbox(label="Contraseña", type="password")
         salida_login = gr.Textbox(label="Mensaje")
+
+        nombre_reg = gr.Textbox(label="Nombre completo")
+        email_reg = gr.Textbox(label="Correo electrónico")
+        pass_reg = gr.Textbox(label="Contraseña", type="password")
+        salida_reg = gr.Textbox(label="Resultado del registro")
+
         btn_login = gr.Button("Entrar")
-    
-    # --- Tabs de NutriBot, ocultos hasta login ---
-    tabs_nuti = gr.Tabs(visible=False)
-    with tabs_nuti:
-        saludo = gr.Markdown("## 👋 Hola")
+        btn_reg = gr.Button("Registrarse")
+
+        def login_click(email, password):
+            msg = iniciar_sesion(email, password)
+            if "✅" in msg:
+                # Si login exitoso, recarga la página para mostrar app principal
+                usuario = email
+                with open(SESION_FILE, "w") as f:
+                    json.dump({"usuario": usuario}, f)
+                return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+            return msg
+
+        def reg_click(nombre, email, password):
+            return registrar_usuario(nombre, email, password)
+
+        btn_login.click(login_click, inputs=[email_login, pass_login], outputs=salida_login)
+        btn_reg.click(reg_click, inputs=[nombre_reg, email_reg, pass_reg], outputs=salida_reg)
+
+    else:
+        # ---------- APP PRINCIPAL ----------
+        gr.Markdown(f"👋 Hola, **{usuario_actual}**. Bienvenido a NutriBot Ultra!")
+
         with gr.Tab("🧍 Tu perfil"):
             edad = gr.Number(label="Edad (años)")
             peso = gr.Number(label="Peso (kg)")
             altura = gr.Number(label="Altura (cm)")
-            sexo = gr.Radio(["Hombre","Mujer"], label="Sexo")
-            actividad = gr.Dropdown(["Sedentario","Ligero","Moderado","Activo","Muy activo"], label="Nivel de actividad")
+            sexo = gr.Radio(["Hombre", "Mujer"], label="Sexo")
+            actividad = gr.Dropdown(["Sedentario", "Ligero", "Moderado", "Activo", "Muy activo"], label="Nivel de actividad")
             btn_meta = gr.Button("Calcular meta calórica")
             salida_meta = gr.Textbox(label="Resultado")
-            btn_meta.click(calcular_calorias, inputs=[edad,peso,altura,sexo,actividad], outputs=salida_meta)
-    
-    # --- Función del login ---
-    def login_click(email,password):
-        mensaje, exito, nombre = iniciar_sesion(email,password)
-        if exito:
-            user_name.value = nombre
-            saludo.update(f"## 👋 Hola, **{nombre}**")
-            tabs_nuti.update(visible=True)
-        return mensaje
-    btn_login.click(login_click, inputs=[email_login, pass_login], outputs=salida_login)
+            btn_meta.click(calcular_calorias, inputs=[edad, peso, altura, sexo, actividad], outputs=salida_meta)
+
+        with gr.Tab("📸 Tu comida"):
+            img = gr.Image(type="pil", label="Sube una foto de tu comida")
+            btn_foto = gr.Button("Analizar comida")
+            salida_foto = gr.Textbox(label="Resultado")
+            btn_foto.click(analizar_foto, inputs=img, outputs=salida_foto)
+
+        with gr.Tab("🏃 Actividad física"):
+            tipo = gr.Dropdown(list(actividades.keys()), label="Tipo de actividad")
+            duracion = gr.Number(label="Duración (minutos)")
+            btn_act = gr.Button("Registrar actividad")
+            salida_act = gr.Textbox(label="Resultado")
+            btn_act.click(registrar_actividad, inputs=[tipo, duracion], outputs=salida_act)
+
+        with gr.Tab("🧠 Coach emocional"):
+            emocion = gr.Textbox(label="¿Cómo te sientes hoy?")
+            btn_emocion = gr.Button("Enviar emoción")
+            salida_emocion = gr.Textbox(label="Consejo")
+            btn_emocion.click(coach_emocional, inputs=emocion, outputs=salida_emocion)
+
+        with gr.Tab("🎯 Misión saludable"):
+            btn_mision = gr.Button("Generar misión diaria")
+            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")
+            salida_chat = gr.Textbox(label="Respuesta de Nuti")
+            btn_chat.click(chatbot_nuti, inputs=pregunta, outputs=salida_chat)
 
 demo.launch()