Update main.py

main.py

@@ -1,4 +1,5 @@
 import os
+import json
 from typing import List
 from pathlib import Path
 from langchain_huggingface import HuggingFaceEmbeddings
@@ -20,6 +21,23 @@ from langchain.callbacks.base import BaseCallbackHandler
 
 import chainlit as cl
 from literalai import LiteralClient
+@cl.password_auth_callback
+def auth_callback(username: str, password: str):
+    auth = json.loads(os.environ['CHAINLIT_AUTH_LOGIN'])
+    ident = next(d['ident'] for d in auth if d['ident'] == username)
+    pwd = next(d['pwd'] for d in auth if d['ident'] == username)
+    resultLogAdmin = bcrypt.checkpw(username.encode('utf-8'), bcrypt.hashpw(ident.encode('utf-8'), bcrypt.gensalt())) 
+    resultPwdAdmin = bcrypt.checkpw(password.encode('utf-8'), bcrypt.hashpw(pwd.encode('utf-8'), bcrypt.gensalt())) 
+    resultRole = next(d['role'] for d in auth if d['ident'] == username)
+    if resultLogAdmin and resultPwdAdmin and resultRole == "admindatapcc":
+        return cl.User(
+            identifier=ident + " : 🧑‍💼 Admin Datapcc", metadata={"role": "admin", "provider": "credentials"}
+        )
+    elif resultLogAdmin and resultPwdAdmin and resultRole == "userdatapcc":
+        return cl.User(
+            identifier=ident + " : 🧑‍🎓 User Datapcc", metadata={"role": "user", "provider": "credentials"}
+        )
+        
 literal_client = LiteralClient(api_key=os.getenv("LITERAL_API_KEY"))
 
 chunk_size = 1024