Update main.py

main.py

@@ -362,6 +362,14 @@ def rename(orig_author: str):
     rename_dict = {"ConversationalRetrievalChain": "💬 Assistant conversationnel", "Retriever": "Agent conversationnel", "StuffDocumentsChain": "Chaîne de documents", "LLMChain": "Agent", "HuggingFaceEndpoint": "Mistral AI 🤖"}
     return rename_dict.get(orig_author, orig_author)
 
+@cl.action_callback("listOfEntreprise")
+async def on_action(action):
+    romeListArray = action.value
+    stringLsitOfEntreprise = await creation_liste_entreprises(arrayOfSecteur(romeListArray))
+    await cl.Message(
+        author="Datapcc : 🌐🌐🌐", content=stringLsitOfEntreprise
+    ).send()    
+    
 @cl.action_callback("datavizChiffresClesMetiers")
 async def on_action(action):
     romeListArray = ast.literal_eval(action.value)
@@ -478,7 +486,6 @@ async def construction_NCS(competenceList, chatProfile):
     emploisST = context.to_string(index = False)
     if chatProfile == 'ROMESKILLS':
         romeListArray = cl.user_session.get("codeRomeArray")
-        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()
@@ -494,13 +501,6 @@ 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(
@@ -508,6 +508,11 @@ async def construction_NCS(competenceList, chatProfile):
         )
         await cl.Message(author="Datapcc : 🌐🌐🌐",content="📈 Source France Travail : " + listClesMetier_name, elements=text_ClesMetier).send()
         await cl.sleep(1)
+        listOfEntreprises = [
+            cl.Action(name="listOfEntreprises", value=romeListArray, description="Afficher la liste des entreprises par code secteur d'actvités")
+        ]
+        await cl.Message(author="Datapcc : 🌐🌐🌐",content="🏭 Afficher la liste des entreprises par code secteur d'actvités", actions=listOfEntreprises).send()
+        await cl.sleep(1)
         datavizChiffresClesMetiers = [
             cl.Action(name="datavizChiffresClesMetiers", value=str(romeListArray), description="Afficher la datavisualisation des chiffres clés des métiers")
         ]
@@ -768,7 +773,7 @@ async def creation_liste_entreprises(arrayOfsecteur):
             print('OK')
             documents = response.json()
             df = pd.DataFrame(documents)
-            ficheEntreprise = ''
+            ficheEntreprise = "**Code secteur d'activités : " + codeSecteurSTR + "**"
             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: