Update login.py

login.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import json
 import os
-import subprocess
 
 USUARIOS_FILE = "usuarios.json"
+SESION_FILE = "sesion.json"
 
-# ---------------- FUNCIONES BÁSICAS ---------------- #
+# ---------------- FUNCIONES ---------------- #
 
 def cargar_usuarios():
     if os.path.exists(USUARIOS_FILE):
@@ -17,13 +17,24 @@ def guardar_usuarios(data):
     with open(USUARIOS_FILE, "w") as f:
         json.dump(data, f, indent=4)
 
+def cargar_sesion():
+    if os.path.exists(SESION_FILE):
+        with open(SESION_FILE, "r") as f:
+            return json.load(f)
+    return {}
+
+def guardar_sesion(usuario):
+    with open(SESION_FILE, "w") as f:
+        json.dump({"usuario": usuario}, f)
+
 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!"
+    guardar_sesion(email)  # inicia sesión automáticamente al registrarse
+    return "✅ Registro exitoso. ¡Ahora estás logueado!"
 
 def iniciar_sesion(email, password):
     usuarios = cargar_usuarios()
@@ -31,14 +42,13 @@ def iniciar_sesion(email, password):
         return "❌ Usuario no encontrado."
     if usuarios[email]["password"] != password:
         return "🔐 Contraseña incorrecta."
-    with open("sesion.json", "w") as f:
-        json.dump({"usuario": email}, f)
-    return f"✅ Bienvenido de nuevo, {usuarios[email]['nombre']}! Redirigiendo a NutriBot..."
+    guardar_sesion(email)
+    return f"✅ Bienvenido de nuevo, {usuarios[email]['nombre']}!"
 
-def lanzar_app():
-    # Ejecuta la app principal (app.py)
-    subprocess.Popen(["python", "app.py"])
-    return "🚀 NutriBot se está iniciando..."
+def cerrar_sesion():
+    if os.path.exists(SESION_FILE):
+        os.remove(SESION_FILE)
+    return "🔒 Has cerrado sesión."
 
 # ---------------- INTERFAZ GRADIO ---------------- #
 
@@ -50,24 +60,28 @@ with gr.Blocks(title="NutriBot - Inicio de Sesión") as login_app:
     ---
     """)
 
-    with gr.Tab("🔑 Iniciar sesión"):
-        email_login = gr.Textbox(label="Correo electrónico")
-        pass_login = gr.Textbox(label="Contraseña", type="password")
-        btn_login = gr.Button("Entrar")
-        salida_login = gr.Textbox(label="Mensaje")
-        btn_login.click(iniciar_sesion, inputs=[email_login, pass_login], outputs=salida_login)
-
-        # Al iniciar sesión, se lanza NutriBot
-        btn_continuar = gr.Button("Abrir NutriBot")
-        salida_continuar = gr.Textbox(label="Estado")
-        btn_continuar.click(lanzar_app, outputs=salida_continuar)
+    sesion_actual = cargar_sesion()
+    if sesion_actual:
+        gr.Markdown(f"👋 Hola, **{sesion_actual['usuario']}**. Ya estás logueado.")
+        btn_logout = gr.Button("Cerrar sesión")
+        salida_logout = gr.Textbox(label="Estado")
+        btn_logout.click(cerrar_sesion, outputs=salida_logout)
+        # Aquí podrías redirigir directamente a app.py
+        # import app
+    else:
+        with gr.Tab("🔑 Iniciar sesión"):
+            email_login = gr.Textbox(label="Correo electrónico")
+            pass_login = gr.Textbox(label="Contraseña", type="password")
+            btn_login = gr.Button("Entrar")
+            salida_login = gr.Textbox(label="Mensaje")
+            btn_login.click(iniciar_sesion, inputs=[email_login, pass_login], outputs=salida_login)
 
-    with gr.Tab("📝 Registrarse"):
-        nombre_reg = gr.Textbox(label="Nombre completo")
-        email_reg = gr.Textbox(label="Correo electrónico")
-        pass_reg = gr.Textbox(label="Contraseña", type="password")
-        btn_reg = gr.Button("Crear cuenta")
-        salida_reg = gr.Textbox(label="Resultado del registro")
-        btn_reg.click(registrar_usuario, inputs=[nombre_reg, email_reg, pass_reg], outputs=salida_reg)
+        with gr.Tab("📝 Registrarse"):
+            nombre_reg = gr.Textbox(label="Nombre completo")
+            email_reg = gr.Textbox(label="Correo electrónico")
+            pass_reg = gr.Textbox(label="Contraseña", type="password")
+            btn_reg = gr.Button("Crear cuenta")
+            salida_reg = gr.Textbox(label="Resultado del registro")
+            btn_reg.click(registrar_usuario, inputs=[nombre_reg, email_reg, pass_reg], outputs=salida_reg)
 
 login_app.launch()