Update homeskills.py

homeskills.py

@@ -1,8 +1,28 @@
+import os
 import pandas as pd
 import chainlit as cl
+from pinecone import Pinecone
 from chainlit.input_widget import Select, TextInput
 from chainlit import user_session
 
+def connexion_vector_database_secteurs():
+    os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
+    os.environ['PINECONE_INDEX_NAME'] = os.environ['PINECONE_INDEX_NAME']
+    pc = Pinecone(api_key=os.environ["PINECONE_API_KEY"])
+    index = pc.Index(index_name=os.environ['PINECONE_INDEX_NAME'])
+    allsecteurs = index.query(
+        top_k=515,
+        vector= [0] * 768, # embedding dimension
+        namespace='',
+        include_metadata=True,
+        include_values=True,
+        filter={'categorie': {'$eq': 'inpiSecteur'}}
+    )
+    secteurs = []
+    for i in range(0,len(allsecteurs['matches'])):
+        secteurs.append(str(allsecteurs['matches'][i]['metadata']['activiteprincipale']) + " " + str(allsecteurs['matches'][i]['metadata']['activitesecondaire']))
+    return secteurs
+    
 async def homeRome():
     app_user = cl.user_session.get("user")
     welcomeUser = app_user.identifier
@@ -97,18 +117,9 @@ async def homeRne():
     welcomeUserArray = welcomeUser.split('@')
     welcomeUserStr = welcomeUserArray[0].replace('.',' ')
     await cl.Message(f"> Bonjour {welcomeUserStr}").send()
-    fileSecteurs = "./public/secteurs_codes.csv"
-    df_allsecteurs = pd.read_csv(fileSecteurs, sep=";")
-    df_allsecteurs['libelle_secteur'] = df_allsecteurs[df_allsecteurs.columns[0:2]].apply(
-        lambda x: '; '.join(x.dropna().astype(str)),
-        axis=1
-    )
-    df_secteurs = df_allsecteurs[['libelle_secteur']].copy()
-    df_secteurs = df_secteurs.sort_values(by=['libelle_secteur'])
-    secteurs_list = df_secteurs['libelle_secteur'].tolist()
-    secteurs_list.sort()
-    secteurs_list.insert(0, "")
-    cl.user_session.set("arraySettingsRne", secteurs_list)
+    secteurs_list = connexion_vector_database_secteurs()
+    df_secteurs = pd.DataFrame(secteurs_list)
+    cl.user_session.set("arraySettingsComp", secteurs_list)
     settings = await cl.ChatSettings(
         [
             Select(