Update main.py

main.py

@@ -27,8 +27,6 @@ from IPython.display import display
 from literalai import LiteralClient
 literal_client = LiteralClient(api_key=os.getenv("LITERAL_API_KEY"))
 
-from surveycaa import surveyCaa
-
 @cl.password_auth_callback
 def auth_callback(username: str, password: str):
     auth = json.loads(os.environ['CHAINLIT_AUTH_LOGIN'])
@@ -176,8 +174,8 @@ async def on_action(action):
 @cl.set_chat_profiles
 async def chat_profile():
     return [
-        cl.ChatProfile(name="Traitement des données d'enquête : «Expé CFA»",markdown_description="Questionnaire auprès des professionnels de la branche de l'agencement",icon="/public/logo-ofipe.png",),
-        #cl.ChatProfile(name="Articles de recherche sur la Pédagogie Durable",markdown_description="Q&A sur la Pédagogie Durable",icon="/public/logo-ofipe.png",),
+        cl.ChatProfile(name="Survey RH",markdown_description="Questionnaire pour le séminaire RH",icon="/public/logo-ofipe.png",),
+        cl.ChatProfile(name="Survey CAA",markdown_description="Questionnaire auprès des professionnels de la branche de l'agencement",icon="/public/logo-ofipe.png",),
         #cl.ChatProfile(name="Articles de recherche sur les lieux d'apprentissage",markdown_description="Q&A sur les lieux d'apprentissage",icon="/public/logo-ofipe.png",),
         #cl.ChatProfile(name="Articles de recherche sur les espaces d'apprentissage",markdown_description="Q&A sur les espaces d'apprentissage",icon="/public/logo-ofipe.png",),
     ]
@@ -185,7 +183,13 @@ async def chat_profile():
 @cl.on_chat_start
 async def on_chat_start():
     await cl.Message(f"> SURVEYIA").send()
-    fileSurvey = await surveyCaa()
+    chat_profile = cl.user_session.get("chat_profile")
+    if chat_profile == "Survey CAA":
+        from surveycaa import surveyCaa
+        fileSurvey = await surveyCaa()
+    elif chat_profile == "Survey RH":
+        from surveycaa import surveyRh
+        fileSurvey = await surveyCaa()
     cl.user_session.set("fileSurvey", fileSurvey)
 
 @cl.on_message