Update app.py

app.py

@@ -3,22 +3,12 @@ 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
+from login import registrar_usuario, iniciar_sesion, SESION_FILE
 
-# ---------------- SESIÓN ----------------
-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
-
-# ---------------- ESTADO ----------------
+# -------------------- ESTADO --------------------
 estado = cargar_registro()
 for k, v in {
     "meta_diaria": 0, "restantes": 0, "quemas": 0, "puntos": 0,
@@ -26,21 +16,23 @@ for k, v in {
 }.items():
     estado.setdefault(k, v)
 
-# ---------------- MODELO ----------------
+# -------------------- MODELO --------------------
 modelo = cargar_modelo_ligero()
 
-# ---------------- DATOS ----------------
+# 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
 }
 
+# Actividades físicas
 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"},
@@ -54,15 +46,7 @@ nutriologos = [
      "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 ----------------
+# -------------------- 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)
@@ -116,6 +100,14 @@ 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."
+]
+
 def generar_mision():
     mision = random.choice(misiones)
     estado["mision_actual"] = mision
@@ -138,12 +130,14 @@ def buscar_nutriologo(pais, estado_, ciudad):
         salida += f"\n• {n['nombre']} — {n['especialidad']}\n  📍 {n['direccion']}\n  📞 {n['contacto']}\n"
     return salida
 
-# ---------------- INTERFAZ ----------------
-usuario_actual = obtener_usuario_actual()
-
+# -------------------- INTERFAZ --------------------
 with gr.Blocks() as demo:
-    if usuario_actual is None:
-        # ---------- LOGIN / REGISTRO ----------
+    # Bloques para login/registro y app principal
+    login_block = gr.Column(visible=True)
+    app_block = gr.Column(visible=False)
+
+    # --- LOGIN / REGISTRO ---
+    with login_block:
         gr.Markdown("# 🥗 Bienvenido a NutriBot Ultra 🌈")
         gr.Markdown("Por favor inicia sesión o crea una cuenta para continuar")
 
@@ -162,21 +156,28 @@ with gr.Blocks() as demo:
         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
+                    json.dump({"usuario": email}, f)
+                return msg, gr.update(visible=False), gr.update(visible=True)
+            return msg, gr.update(visible=True), gr.update(visible=False)
 
         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 ----------
+            msg = registrar_usuario(nombre, email, password)
+            if "✅" in msg:
+                with open(SESION_FILE, "w") as f:
+                    json.dump({"usuario": email}, f)
+                return msg, gr.update(visible=False), gr.update(visible=True)
+            return msg, gr.update(visible=True), gr.update(visible=False)
+
+        btn_login.click(login_click, inputs=[email_login, pass_login], outputs=[salida_login, login_block, app_block])
+        btn_reg.click(reg_click, inputs=[nombre_reg, email_reg, pass_reg], outputs=[salida_reg, login_block, app_block])
+
+    # --- APP PRINCIPAL ---
+    with app_block:
+        usuario_actual = None
+        if os.path.exists(SESION_FILE):
+            with open(SESION_FILE, "r") as f:
+                usuario_actual = json.load(f).get("usuario", None)
         gr.Markdown(f"👋 Hola, **{usuario_actual}**. Bienvenido a NutriBot Ultra!")
 
         with gr.Tab("🧍 Tu perfil"):