Update app.py

app.py

@@ -7,173 +7,86 @@ from registro import guardar_registro, cargar_registro
 from chatbot_nuti import buscar_respuesta
 from optimizacion import resize_imagen, cargar_modelo_ligero, cache_resultado
 
-# -------------------- ARCHIVOS --------------------
+# ---- Archivos ----
 USUARIOS_FILE = "usuarios.json"
 SESION_FILE = "sesion.json"
 
 def cargar_usuarios():
     if os.path.exists(USUARIOS_FILE):
-        with open(USUARIOS_FILE, "r") as f:
+        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)
+    with open(USUARIOS_FILE,"w") as f:
+        json.dump(data,f,indent=4)
 
-def registrar_usuario(nombre, email, password):
+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}
+    usuarios[email] = {"nombre": nombre,"password": password}
     guardar_usuarios(usuarios)
     return "✅ Registro exitoso. ¡Ahora puedes iniciar sesión!"
 
-def iniciar_sesion(email, password):
+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)
+    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 ----
 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()
 
-# -------------------- 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"},
-]
-
-# -------------------- FUNCIONES NUTRIBOT --------------------
-# Aquí pegas tus funciones: calcular_calorias, analizar_foto, registrar_actividad, coach_emocional, generar_mision, chatbot_nuti, buscar_nutriologo
-
+# ---- Funciones ejemplo (tu código original aquí) ----
+# Solo pongo la de calcular_calorias para acortar
 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."
 
-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.")
-
-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
-    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 --------------------
+# ---- Interfaz ----
 with gr.Blocks() as demo:
-    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
+    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 del registro")
+        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)
-        # Login
+        
+        gr.Markdown("**Iniciar sesión:**")
         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)
-        
+    
+    # --- Tabs de NutriBot, ocultos hasta login ---
+    tabs_nuti = gr.Tabs(visible=False)
+    with tabs_nuti:
+        saludo = gr.Markdown("## 👋 Hola")
         with gr.Tab("🧍 Tu perfil"):
             edad = gr.Number(label="Edad (años)")
             peso = gr.Number(label="Peso (kg)")
@@ -183,24 +96,15 @@ with gr.Blocks() as demo:
             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)
-        
-       
+    
+    # --- 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)
+
+demo.launch()