Update modules/auth.py

modules/auth.py

@@ -1,14 +1,14 @@
 # /modules/auth.py
 import streamlit as st
 import pandas as pd
+import os
 
-# Función para cargar usuarios (en una aplicación real, esto sería una base de datos)
-@st.cache_data
+# Función para cargar usuarios
 def load_users():
-    try:
-        return pd.read_csv("users.csv")
-    except FileNotFoundError:
-        return pd.DataFrame(columns=["username", "password", "role"])
+    if not os.path.exists("users.csv"):
+        df = pd.DataFrame(columns=["username", "password", "role"])
+        df.to_csv("users.csv", index=False)
+    return pd.read_csv("users.csv")
 
 # Función para guardar usuarios
 def save_users(df):
@@ -28,4 +28,14 @@ def register_user(username, password, role):
 def authenticate_user(username, password):
     users = load_users()
     user = users[(users["username"] == username) & (users["password"] == password)]
-    return not user.empty
+    if not user.empty:
+        return True
+    return False
+
+# Función para obtener el rol del usuario
+def get_user_role(username):
+    users = load_users()
+    user = users[users["username"] == username]
+    if not user.empty:
+        return user.iloc[0]["role"]
+    return None