Update main.py

main.py

@@ -24,6 +24,8 @@ from chainlit.input_widget import TextInput, Select, Switch, Slider
 from deep_translator import GoogleTranslator
 from IPython.display import display
 
+from surveycaa import surveyCaa, surveyRh
+
 from literalai import LiteralClient
 literal_client = LiteralClient(api_key=os.getenv("LITERAL_API_KEY"))
 
@@ -185,11 +187,9 @@ async def on_chat_start():
     await cl.Message(f"> SURVEYIA").send()
     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()
+        fileSurvey = await surveyRh()
     cl.user_session.set("fileSurvey", fileSurvey)
 
 @cl.on_message