Update main.py

main.py

@@ -309,6 +309,17 @@ def datavisualisation_chiffres_cles_emplois(url):
 def listToString(list):
     return str(list)
 
+def arrayOfSecteur(array):
+    df_naf = pd.read_csv("./public/secteur_naf_v01.csv")
+    df_naf = df_naf[['libelle_secteur_naf', 'code_rome']].copy()
+    df_naf = df_naf.groupby(df_naf['code_rome'], as_index=False).agg({'libelle_secteur_naf': '; '.join})
+    
+    df_rome = pd.DataFrame({'code_rome': array})
+    
+    df_romeNaf = pd.merge(df_rome, df_naf, on="code_rome")
+    df_secteur = df_romeNaf[['libelle_secteur_naf']].copy()
+    return df_secteur.values.tolist()
+    
 def arrayToString(array):
     arrayList = []
     
@@ -467,6 +478,8 @@ async def construction_NCS(competenceList, chatProfile):
     emploisST = context.to_string(index = False)
     if chatProfile == 'ROMESKILLS':
         romeListArray = cl.user_session.get("codeRomeArray")
+        stringSecteur = await creation_liste_entreprises(arrayOfSecteur(romeListArray))
+        print(stringSecteur)
         ficheClesMetier = await document_chiffres_cles_emplois("https://dataemploi.francetravail.fr/metier/chiffres-cles/NAT/FR/", romeListArray)
         contentChatBot = str(emploisST).replace('[','').replace(']','').replace('{','').replace('}','') + ficheClesMetier
         finals_df = context[['intitule','typeContratLibelle','experienceLibelle','competences','description','qualitesProfessionnelles','salaire','lieuTravail','formations']].copy()
@@ -730,6 +743,14 @@ async def API_France_Travail(romeListArray):
     results_df = pd.DataFrame(results)
     return results_df
     
+@cl.step(type="tool")
+async def creation_liste_entreprises(arrayOfsecteur):
+    docsearch = await connexion_vector_database()
+    for i in range(0, len(arraySecteur)):
+        retrieve_comp = docsearch.similarity_search(arraySecteur[i][0], k=1, filter={"categorie": {"$eq": "inpiSecteur"}})
+        codeSecteur = retrieve_comp[0].metadata['codefinal']
+    return codeSecteur
+        
 @cl.step(type="tool")
 async def creation_liste_code_Rome(competence, categorie):
     docsearch = await connexion_vector_database()