Update main.py

main.py

@@ -19,22 +19,22 @@ from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableLam
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.prompts import PromptTemplate
 
-#@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"}
-#        )
+@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"}
+        )
         
 @cl.author_rename
 def rename(orig_author: str):