Update app.py

app.py

@@ -32,17 +32,48 @@ from agent_collaboratif_avid import (
     SIMILARITY_TOP_K,
     MAX_VALIDATION_LOOPS
 )
+import bcrypt
 from chainlit.data.sql_alchemy import SQLAlchemyDataLayer
-from supabase import Client
+from supabase import create_client, Client
+from supabase.lib.client_options import ClientOptions
 
 SUPABASE_URL = os.environ.get("SUPABASE_URL")
 SUPABASE_ANON_KEY = os.environ.get("SUPABASE_ANON_KEY")
 CONNINFO = os.environ.get("CONNINFO")
 
-storage_client = Client(SUPABASE_URL, SUPABASE_ANON_KEY)
+url: str = SUPABASE_URL
+key: str = SUPABASE_ANON_KEY
+supabase: Client = create_client(url, key, options=ClientOptions(auto_refresh_token=False,persist_session=True))
+
+# =============================================================================
+# CONFIGURATION DE L'AUTHENTIFICATION (Optionnel)
+# =============================================================================
+
+@cl.password_auth_callback
+def auth_callback(username: str, password: str) -> Optional[cl.User]:
+    """
+    Callback d'authentification (optionnel).
+    À configurer selon vos besoins.
+    """
+    # Exemple simple (à remplacer par votre logique)
+    auth = json.loads(os.environ.get("CHAINLIT_AUTH_LOGIN"))
+    ident = auth[0]['ident']
+    pwd = auth[0]['pwd']
+    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 = auth[0]['role']
+    
+    if resultLogAdmin == True and resultPwdAdmin == True and resultRole == "adminavid":
+        return cl.User(
+            identifier=auth[0]['ident'],
+            metadata={"role": "adminavid", "provider": "credentials"}
+        )
+    else:
+        return None
+
 @cl.data_layer
 def get_data_layer():
-    return SQLAlchemyDataLayer(conninfo=CONNINFO, storage_provider=storage_client)
+    return SQLAlchemyDataLayer(conninfo=CONNINFO, storage_provider=supabase)
 
 # =============================================================================
 # CONFIGURATION LANGSMITH
@@ -425,24 +456,6 @@ Merci d'avoir utilisé l'agent collaboratif !
     
     await cl.Message(content=end_msg).send()
 
-# =============================================================================
-# CONFIGURATION DE L'AUTHENTIFICATION (Optionnel)
-# =============================================================================
-
-@cl.password_auth_callback
-def auth_callback(username: str, password: str) -> Optional[cl.User]:
-    """
-    Callback d'authentification (optionnel).
-    À configurer selon vos besoins.
-    """
-    # Exemple simple (à remplacer par votre logique)
-    if username == "admin" and password == "password":
-        return cl.User(
-            identifier=username,
-            metadata={"role": "admin", "provider": "credentials"}
-        )
-    return None
-
 # =============================================================================
 # CONFIGURATION DU DATA LAYER (Supabase/PostgreSQL)
 # =============================================================================