Update main.py

main.py

@@ -478,8 +478,7 @@ 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)
+        stringLsitOfEntreprise = await creation_liste_entreprises(arrayOfSecteur(romeListArray))
         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()
@@ -495,6 +494,13 @@ async def construction_NCS(competenceList, chatProfile):
         )
         await cl.Message(author="Datapcc : 🌐🌐🌐",content="👨‍💼 Source France Travail : " + listEmplois_name, elements=text_elements).send()
         await cl.sleep(1)
+        listEntreprise_name = f"Liste des entreprises"
+        entreprise_elements = []
+        entreprise_elements.append(
+            cl.Text(content="Question : " + competenceList + "\n\nRéponse :\n" + stringLsitOfEntreprise, name=listEntreprise_name)
+        )
+        await cl.Message(author="Datapcc : 🌐🌐🌐",content="🏭 Source Registre National des Entreprises : " + listEntreprise_name, elements=entreprise_elements).send()
+        await cl.sleep(1)
         listClesMetier_name = f"Chiffres clés des emplois"
         text_ClesMetier = []
         text_ClesMetier.append(
@@ -746,11 +752,45 @@ async def API_France_Travail(romeListArray):
 @cl.step(type="tool")
 async def creation_liste_entreprises(arrayOfsecteur):
     docsearch = await connexion_vector_database()
-    for i in range(0, len(arrayOfsecteur)):
-        retrieve_comp = docsearch.similarity_search(arrayOfsecteur[i][0], k=1, filter={"categorie": {"$eq": "inpiSecteur"}})
+    for j in range(0, len(arrayOfsecteur)):
+        retrieve_comp = docsearch.similarity_search(arrayOfsecteur[j][0], k=1, filter={"categorie": {"$eq": "inpiSecteur"}})
         codeSecteur = int(retrieve_comp[0].metadata['codefinal'])
-        codeSecteurSTR = "0" + str(codeSecteur)
-    return int(codeSecteurSTR)
+        if len(codeSecteur) == 7:
+            codeSecteurSTR = "0" + str(codeSecteur)
+        else:
+            codeSecteurSTR = str(codeSecteur)
+        token = await connexion_registre_national_entreprises()
+        url = f"https://registre-national-entreprises.inpi.fr/api/companies?page=1&pageSize=500&codeCategory={codeSecteurSTR}"
+        print(url)
+        headers = {"Authorization": f"Bearer {token}"}
+        response = requests.get(url, headers=headers)
+        if response.status_code == 200:
+            print('OK')
+            documents = response.json()
+            df = pd.DataFrame(documents)
+            ficheEntreprise = ''
+            for i in range (0, len(df)):
+              if str(df['formality'][i]['content']).find('cessation') == -1 and str(df['formality'][i]['content']).find('personneMorale') != -1:
+                if str(df['formality'][i]['content']['personneMorale']).find('adresseEntreprise') != -1:
+                  ficheEntreprise += "\n🏭 Dénomination : " + str(df['formality'][i]['content']['personneMorale']['identite']['entreprise']['denomination']) + "; Code SIREN : " + str(df['formality'][i]['content']['personneMorale']['identite']['entreprise']['siren']) + "\n\tAdresse : "
+                  if str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']).find("'numVoie'") != -1:
+                     ficheEntreprise += "n° " + str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']['numVoie'])
+                  if str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']).find("'typeVoie'") != -1:
+                     ficheEntreprise += " " + str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']['typeVoie'])
+                  if str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']).find("'voie'") != -1:
+                     ficheEntreprise += " " + str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']['voie'])
+                  if str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']).find("'complementLocalisation'") != -1:
+                     ficheEntreprise += ", " + str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']['complementLocalisation'])+ ","
+                  if str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']).find("'codePostal'") != -1:
+                     ficheEntreprise += " " + str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']['codePostal'])
+                  if str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']).find("'commune'") != -1:
+                     ficheEntreprise += " " + str(df['formality'][i]['content']['personneMorale']['adresseEntreprise']['adresse']['commune'])
+                if str(df['formality'][i]['content']['personneMorale']).find("autresEtablissements") != -1:
+                     ficheEntreprise += "\n\tActivités : " + str(df['formality'][101]['content']['personneMorale']['autresEtablissements'][0]['activites'][0]['descriptionDetaillee']) + "\n\tCode APE : " + str(df['formality'][101]['content']['personneMorale']['autresEtablissements'][0]['activites'][0]['codeApe'])
+        else:
+            ficheEntreprise += "Il n'y a pas d'entreprises pour le secteur sélectionné : " + arrayOfsecteur[i][0]
+    return ficheEntreprise
+            
         
 @cl.step(type="tool")
 async def creation_liste_code_Rome(competence, categorie):
@@ -863,6 +903,17 @@ async def creation_liste_skills_Esco(competence, categorie):
                 await cl.Message(author="Datapcc : 🌐🌐🌐",content="Votre saisie est erronée. Nous continuons l'action avec les métiers sélectionnés automatiquement pour vous : " + codeRomeString).send()
             cl.user_session.set("codeRomeArray", arrayCodeRome)
     
+@cl.step(type="tool")
+async def connexion_registre_national_entreprises():
+    url = "https://registre-national-entreprises.inpi.fr/api/sso/login"
+    headers = {"Content-Type": "application/json"}
+    data = {"username": os.environ['RNE_CLIENT_ID'], "password": os.environ['RNE_CLIENT_SECRET']}
+    response = requests.post(url, json=data, headers=headers)
+    if response.status_code == 200:
+        return response.json()["token"]
+    else:
+        raise Exception(f"Échec de l'authentification. Code d'erreur : {response.status_code}")
+
 @cl.step(type="tool")
 async def connexion_France_Travail():
     client = Api(client_id=os.environ['POLE_EMPLOI_CLIENT_ID'],