Update app.py

app.py

@@ -1,23 +1,44 @@
 import gradio as gr
-import datetime
-import random
 import json
 import os
+import datetime
+import random
 from registro import guardar_registro, cargar_registro
 from chatbot_nuti import buscar_respuesta
 from optimizacion import resize_imagen, cargar_modelo_ligero, cache_resultado
 
-# -------------------- SESIÓN --------------------
+# -------------------- ARCHIVOS --------------------
+USUARIOS_FILE = "usuarios.json"
 SESION_FILE = "sesion.json"
 
-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
-
-usuario_actual = obtener_usuario_actual()
+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 --------------------
 estado = cargar_registro()
@@ -30,20 +51,18 @@ for k, v in {
 # -------------------- MODELO --------------------
 modelo = cargar_modelo_ligero()
 
-# Calorías promedio
+# -------------------- 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 físicas con MET aproximado
 actividades = {
     "Correr": 700, "Caminar": 250, "Fútbol": 600,
     "Natación": 550, "Básquetbol": 500, "Voleibol": 400, "Yoga": 200,
 }
 
-# Nutriólogos
 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"},
@@ -57,7 +76,9 @@ nutriologos = [
      "especialidad": "Nutrición general y recetas personalizadas", "direccion": "Av. Chapultepec 100", "contacto": "IG: @sofiariveranutri"},
 ]
 
-# -------------------- FUNCIONES --------------------
+# -------------------- FUNCIONES NUTRIBOT --------------------
+# Aquí pegas tus funciones: calcular_calorias, analizar_foto, registrar_actividad, coach_emocional, generar_mision, chatbot_nuti, buscar_nutriologo
+
 def calcular_calorias(edad, peso, altura, sexo, actividad):
     if sexo.lower() == "hombre":
         bmr = 88.36 + (13.4 * peso) + (4.8 * altura) - (5.7 * edad)
@@ -73,15 +94,12 @@ def calcular_calorias(edad, peso, altura, sexo, actividad):
 def analizar_foto(img):
     if estado["meta_diaria"] == 0:
         return "⚠️ Primero calcula tu meta diaria."
-    img_opt = resize_imagen(img, size=(224,224))
+    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["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."
@@ -92,10 +110,7 @@ def registrar_actividad(tipo, duracion):
     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']}."
@@ -111,14 +126,7 @@ def coach_emocional(emocion):
     guardar_registro(estado)
     return sugerencias.get(emocion.lower(), "💬 Cuéntame más sobre cómo te sientes.")
 
-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."
-]
-
+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."]
 def generar_mision():
     mision = random.choice(misiones)
     estado["mision_actual"] = mision
@@ -130,12 +138,8 @@ def chatbot_nuti(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."
+    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"
@@ -143,62 +147,60 @@ def buscar_nutriologo(pais, estado_, ciudad):
 
 # -------------------- INTERFAZ --------------------
 with gr.Blocks() as demo:
-    # Bienvenida
-    if usuario_actual:
-        gr.Markdown(f"👋 Hola, **{usuario_actual}**. Bienvenido a NutriBot Ultra!")
-    else:
-        gr.Markdown("👋 Hola! Bienvenido a NutriBot Ultra!")
-
-    gr.Markdown("# 🥦 NutriBot Ultra — Tu Asistente de Bienestar Integral 🌈")
-    gr.Markdown("IA que une nutrición, emociones y diversión 💪🍎")
-
-    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")
-        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)
-
-    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()
-
+    user_name = gr.State(value="")  # Guarda el nombre del usuario
+
+    # ----- LOGIN/REGISTRO -----
+    with gr.Tab("🔑 Iniciar sesión / Registrarse"):
+        gr.Markdown("### Bienvenido a NutriBot Ultra 🥦\nPrimero inicia sesión o crea una cuenta para continuar.")
+        # 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 del registro")
+        btn_reg = gr.Button("Crear cuenta")
+        btn_reg.click(registrar_usuario, inputs=[nombre_reg,email_reg,pass_reg], outputs=salida_reg)
+        # Login
+        email_login = gr.Textbox(label="Correo electrónico")
+        pass_login = gr.Textbox(label="Contraseña", type="password")
+        salida_login = gr.Textbox(label="Mensaje")
+        btn_login = gr.Button("Entrar")
+        def login_click(email, password, state_user):
+            mensaje, exito, nombre = iniciar_sesion(email,password)
+            if exito: state_user.value = nombre
+            return mensaje, gr.Tabs.update(visible=exito)
+        btn_login.click(login_click, inputs=[email_login, pass_login, user_name], outputs=[salida_login])
+
+    # ----- TABS NUTRIBOT (ocultas hasta login) -----
+    with gr.Tabs(visible=False) as tabs_nuti:
+        gr.Markdown(lambda name: f"## 👋 Hola, **{name}**", user_name)
+        
+        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")
+            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)
+        
+        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)
+        
+