Update main.py

main.py

@@ -30,12 +30,19 @@ import bcrypt
         
 @cl.password_auth_callback
 def auth_callback(username: str, password: str):
-    resultLogAdmin = bcrypt.checkpw(username.encode('utf-8'), bcrypt.hashpw(os.environ['CHAINLIT_AUTH_LOGIN'].encode('utf-8'), bcrypt.gensalt())) 
-    resultPwdAdmin = bcrypt.checkpw(password.encode('utf-8'), bcrypt.hashpw(os.environ['CHAINLIT_AUTH_PASSWD'].encode('utf-8'), bcrypt.gensalt())) 
-    if resultLogAdmin and resultPwdAdmin:
+    #resultLogAdmin = bcrypt.checkpw(username.encode('utf-8'), bcrypt.hashpw(os.environ['CHAINLIT_AUTH_LOGIN'].encode('utf-8'), bcrypt.gensalt())) 
+    #resultPwdAdmin = bcrypt.checkpw(password.encode('utf-8'), bcrypt.hashpw(os.environ['CHAINLIT_AUTH_PASSWD'].encode('utf-8'), bcrypt.gensalt())) 
+    resultLogAdmin = bcrypt.checkpw(username.encode('utf-8'), bcrypt.hashpw(next(d['ident'] for d in os.environ['CHAINLIT_AUTH_LOGIN'] if d['ident'] == username).encode('utf-8'), bcrypt.gensalt())) 
+    resultPwdAdmin = bcrypt.checkpw(password.encode('utf-8'), bcrypt.hashpw(next(d['pwd'] for d in os.environ['CHAINLIT_AUTH_LOGIN'] if d['ident'] == username).encode('utf-8'), bcrypt.gensalt())) 
+    resultRole = next(d['role'] for d in os.environ['CHAINLIT_AUTH_LOGIN'] if d['ident'] == username)
+    if resultLogAdmin and resultPwdAdmin and resultRole == "admindatapcc":
         return cl.User(
             identifier="🧑‍💼 Admin Datapcc", metadata={"role": "admin", "provider": "credentials"}
         )
+    elif resultLogAdmin and resultPwdAdmin and resultRole == "userdatapcc":
+        return cl.User(
+            identifier="🧑‍🎓 User Datapcc", metadata={"role": "user", "provider": "credentials"}
+        )
         
 def modele(document):
     if document == "Note de composante sectorielle":