Update main.py

main.py

@@ -431,6 +431,7 @@ async def chat_profile():
     return [
         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.ChatProfile(name="Référentiel National des Entreprises - RNE",markdown_description="Les contacts entreprises",icon="./public/favicon.png",),
     ]
 @cl.on_chat_start
 async def start():
@@ -444,9 +445,11 @@ async def start():
     if chatProfile[1] == 'ROMESKILLS':
         contextChat = await homeRome()
         categorie = cl.user_session.set("categorie", os.environ['PINECONE_API_KEYROME'])
-    else:
+    elif chatProfile[1] == 'ESCOSKILLS':
         contextChat = await homeEsco()
         categorie = cl.user_session.set("categorie", os.environ['PINECONE_API_KEYESCO'])
+    else:
+        contextChat = await homeRne()
     
     os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
     repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
@@ -969,7 +972,10 @@ async def setup_agent(settings):
     chat_profile = cl.user_session.get("chat_profile")
     chatProfile = chat_profile.split(' - ')
     
-    await construction_NCS(competenceList, chatProfile[1])
+    if chatProfile[1] == "ROMESKILLS" or chatProfile[1] == "ESCOSKILLS":
+        await construction_NCS(competenceList, chatProfile[1])
+    else:
+        await construction_NCS(competenceList, chatProfile[1])
     
     contextChat = cl.user_session.get("contextChatBot")
     if not contextChat: