login

code_interface/badgeage.py

@@ -2,7 +2,7 @@ import streamlit as st
 import sqlite3
 import datetime
 # Connect to the database
-conn = sqlite3.connect('C:\Users\neiro\Documents\GitHub\sgbd-usine\donnes\company.db')
+conn = sqlite3.connect(r'C:\Users\neiro\Documents\GitHub\sgbd-usine\donnes\company.db')
 cursor = conn.cursor()
 # Set the title of the Streamlit app
 st.title("Machine de Badgeage")

streamlit_app.py

@@ -21,7 +21,7 @@ if not st.session_state.logged_in:
     if login_button:
         cursor.execute("SELECT status FROM employees WHERE employee_code = ? AND mot_de_passe = ?", (employee_code, mot_de_passe))
         result = cursor.fetchone()
-        if result and result[0] == 'RH':
+        if result and (result[0] == 'RH' or result[0] == 'admin'):
             st.session_state.logged_in = True
             st.session_state.status = result[0]
         else: