Update main.py

main.py

@@ -26,6 +26,7 @@ import numpy as np
 import chainlit as cl
 from chainlit.input_widget import Select, TextInput
 from chainlit import user_session
+from homeskills import homeRome, homeEsco
 from offres_emploi import Api
 from offres_emploi.utils import dt_to_str_iso
 import datetime
@@ -409,7 +410,8 @@ def to_cache(file):
 @cl.set_chat_profiles
 async def chat_profile():
     return [
-        cl.ChatProfile(name="Catalogue ROME - ROMESKILLS",markdown_description="Note composante sectorielle",icon="./public/favicon.png",),
+        cl.ChatProfile(name="Catalogue ROME - ROMESKILLS",markdown_description="Les compétences du catalogue ROME",icon="./public/favicon.png",),
+        cl.ChatProfile(name="Classification ESCO - ESCOSKILLS",markdown_description="Les compétences de la classification ESCO",icon="./public/favicon.png",),
     ]
 @cl.on_chat_start
 async def start():
@@ -421,42 +423,9 @@ async def start():
     chatProfile = chat_profile.split(' - ')
     
     if chatProfile[1] == 'ROMESKILLS':
-        app_user = cl.user_session.get("user")
-        welcomeUser = app_user.identifier
-        welcomeUserArray = welcomeUser.split('@')
-        welcomeUserStr = welcomeUserArray[0].replace('.',' ')
-        await cl.Message(f"> Bonjour {welcomeUserStr}").send()
-        
-        df_allcompetences = pd.read_csv('./public/referentiel_competence.csv')
-        df_competences = df_allcompetences[['libelle_competence']].copy()
-        df_competences = df_competences.sort_values(by=['libelle_competence'])
-        competences_list = df_competences['libelle_competence'].tolist()
-        competences_list.sort()
-        competences_list.insert(0, "")
-        cl.user_session.set("arraySettingsComp", competences_list)
-        settings = await cl.ChatSettings(
-            [
-                Select(
-                    id="competence",
-                    label="Compétences",
-                    values=competences_list,
-                    initial_index=0,
-                ),
-                TextInput(id="competenceInput", label="ou saisir une compétence voire des objectifs pédagogiques", placeholder="ou saisir une compétence voire des objectifs pédagogiques", tooltip="saisir une compétence voire des objectifs pédagogiques"),
-            ]
-        ).send()
-        value = settings["competence"]
-    
-        if len(value) < 2:
-            warning = [
-                        cl.Image(name="Warning", size="small", display="inline", path="./public/warning.png")
-            ]
-            await cl.Message(author="Datapcc : 🌐🌐🌐",content="1️⃣ Cliquez sur le bouton dont l'image suit, dans le prompt, pour commencer à élaborer une note sectorielle de la chaîne documentaire APCC!").send()
-            await cl.Message(author="Datapcc : 🌐🌐🌐",content="", elements=warning).send()
-            await cl.Message(author="Datapcc : 🌐🌐🌐",content="2️⃣ Puis sélectionnez ou saisissez une compétence ou des objectifs pédagogiques. Et vous êtes prêt!\n\n🔗 Plateforme de feedback et de fil d'activité : https://cloud.getliteral.ai/").send()
-    contextChat = cl.user_session.get("contextChatBot")
-    if not contextChat:
-        contextChat = df_competences.to_string(index = False)
+        contextChat = await homeRome()
+    else:
+        contextChat = await homeEsco()
     os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
     repo_id = "mistralai/Mistral-7B-Instruct-v0.3"