Update main.py

main.py

@@ -276,6 +276,19 @@ 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("datavizChiffresClesMetiers")
+async def on_action(action):
+    romeListArray = action.value
+    for j in range(0, len(romeListArray)):
+        table = await datavisualisation_chiffres_cles_emplois("https://dataemploi.pole-emploi.fr/metier/chiffres-cles/NAT/FR/" + romeListArray[j])
+        plot_demandeur = plotDemandeur(htmlToDataframe(table[0]), romeListArray[j])
+        if len(table[1]) > 0:
+            plot_salaire = plotSalaire(htmlToDataframe(table[1]))
+        plot_difficulte = plotDifficulte(htmlToDataframe(table[2]))
+        plot_repartitionContrat = plotRepartition(htmlToDataframe(table[3]), "Répartition des embauches du métier : type de contrat")
+        plot_repartitionEntreprise = plotRepartition(htmlToDataframe(table[4]), "Répartition des embauches du métier : type entreprise")
+    
+    
 @cl.action_callback("download")
 async def on_action(action):
     content = []
@@ -404,7 +417,7 @@ async def construction_NCS(competenceList):
     listEmplois = finals_df.values.tolist()
     stringEmplois = ''
     for i in range(0,len(listEmplois)):
-        stringEmplois += "\n✔️ Emploi : " + str(listEmplois[i][0]) + ";\n◉ Contrat : " + str(listEmplois[i][1]) + ";\n◉Compétences professionnelles : " + str(listEmplois[i][3]) + ";\n" + "◉ Salaire : " + str(listEmplois[i][6]) + ";\n◉ Qualification : " + str(listEmplois[i][5]) + ";\n◉ Localisation : " + str(listEmplois[i][7]) + ";\n◉ Expérience : " + str(listEmplois[i][2]) + ";\n◉ Niveau de qualification : " + str(listEmplois[i][8]) + ";\n◉ Description de l'emploi : " + str(listEmplois[i][4]) + "\n"
+        stringEmplois += "\n✔️ Emploi : " + str(listEmplois[i][0]) + ";\n◉ Contrat : " + str(listEmplois[i][1]) + ";\n◉ Compétences professionnelles : " + str(listEmplois[i][3]) + ";\n" + "◉ Salaire : " + str(listEmplois[i][6]) + ";\n◉ Qualification : " + str(listEmplois[i][5]).replace("'libelle'","\n• 'libelle") + ";\n◉ Localisation : " + str(listEmplois[i][7]) + ";\n◉ Expérience : " + str(listEmplois[i][2]) + ";\n◉ Niveau de qualification : " + str(listEmplois[i][8]) + ";\n◉ Description de l'emploi : " + str(listEmplois[i][4]) + "\n"
     await cl.sleep(1)
     listEmplois_name = f"Liste des emplois"
     text_elements = []
@@ -420,6 +433,11 @@ async def construction_NCS(competenceList):
     )
     await cl.Message(author="Datapcc : 🌐🌐🌐",content="📈 Source France Travail : " + listClesMetier_name, elements=text_ClesMetier).send()
     await cl.sleep(1)
+    datavizChiffresClesMetiers = [
+        cl.Action(name="datavizChiffresClesMetiers", value=romeListArray, description="Afficher la datavisualisation des chiffres clés des métiers")
+    ]
+    await cl.Message(author="Datapcc : 🌐🌐🌐",content="📊 Afficher la datavisualisation des chiffres clés des métiers", actions=datavizChiffresClesMetiers).send()
+    await cl.sleep(1)
     codeArray = romeListArray
     ficheMetiers = []
     for i in range(0,len(codeArray)):
@@ -503,7 +521,7 @@ async def document_chiffres_cles_emplois(url, codes):
             allentreprise = removeTags(allEntreprise[0])
         else:
             allentreprise = ""
-        all = all + "\n\nChiffres-clés Métier " + allTitre + ":\nDemandeurs d'emploi et Offres d'emploi : " + allEmbauches + ". Salaires proposés dans les offres : " + allSalaires + ". Salaires médians constatés : " + allSalairesMedian + ". Difficultés de recrutement pour les entreprises : " + alldiff + ". Origine des difficultés : " + alldiffOrigin + ". Répartition des embauches par type de contrat : " + allTypeContrat + ". Répartition des embauches par taille d'entreprise : " + allentreprise + "."
+        all = all + "\n\nChiffres-clés Métier : \n**" + allTitre + "**:\n◉ Demandeurs d'emploi et Offres d'emploi : " + allEmbauches.replace("Plus de données sur les Demandeurs d'emploi","").replace("Plus de données","") + ".\n◉ Salaires proposés dans les offres : " + allSalaires + ".\n◉ Salaires médians constatés : " + allSalairesMedian + ".\n◉ Difficultés de recrutement pour les entreprises : " + alldiff + ".\n◉ Origine des difficultés : " + alldiffOrigin + ".\n◉ Répartition des embauches par type de contrat : " + allTypeContrat + ".\n◉ Répartition des embauches par taille d'entreprise : " + allentreprise + "."
     return all
 
 @cl.step(type="tool")