Update modules/auth/auth.py

modules/auth/auth.py

@@ -56,7 +56,12 @@ def create_auth_interface():
 
         def handle_login(user, pwd):
             success, role = authenticate_user(user, pwd)
-            return f"Bienvenido, {user} ({role})" if success else "Credenciales incorrectas."
+            if success:
+                st.session_state["username"] = user
+                st.session_state["role"] = role
+                st.session_state["logged_in"] = True
+                return f"Bienvenido, {user} ({role})"
+            return "Credenciales incorrectas."
 
         login_btn.click(fn=handle_login, inputs=[username, password], outputs=message)
     return auth_interface