Update main.py

main.py

@@ -421,7 +421,37 @@ async def on_chat_start():
     fig_wordcloud_principaux_interlocuteurs_entreprise1 = plt.figure(1,figsize=(12,15))
     plt.imshow(wordcloud_principaux_interlocuteurs_entreprise1,interpolation="bicubic")
     plt.axis('off')
-    elements_wordcloud_principaux_interlocuteurs.append(cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs", figure=fig_wordcloud_principaux_interlocuteurs_entreprise1, size="large", display="inline"))
+    elements_wordcloud_principaux_interlocuteurs.append(cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs_entreprise_10_50", figure=fig_wordcloud_principaux_interlocuteurs_entreprise1, size="large", display="inline"))
+
+    options_entreprise2 = ['51 à 100 employés'] 
+    df1_entreprise2 = df1.loc[df1['taille_entreprise'].isin(options_entreprise2)] 
+    wd_principaux_interlocuteurs_entreprise2 = " ".join(i for i in df1_entreprise2.principaux_interlocuteurs)
+    wordcloud_principaux_interlocuteurs_entreprise2 = WordCloud(background_color='white',
+                      stopwords=exclure_mots, max_words=80).generate(wd_principaux_interlocuteurs_entreprise2)
+    fig_wordcloud_principaux_interlocuteurs_entreprise2 = plt.figure(1,figsize=(12,15))
+    plt.imshow(wordcloud_principaux_interlocuteurs_entreprise2,interpolation="bicubic")
+    plt.axis('off')
+    elements_wordcloud_principaux_interlocuteurs.append(cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs_entreprise_51_100", figure=fig_wordcloud_principaux_interlocuteurs_entreprise2, size="large", display="inline"))
+
+    options_entreprise3 = ['Moins de 10 employés'] 
+    df1_entreprise3 = df1.loc[df1['taille_entreprise'].isin(options_entreprise3)] 
+    wd_principaux_interlocuteurs_entreprise3 = " ".join(i for i in df1_entreprise3.principaux_interlocuteurs)
+    wordcloud_principaux_interlocuteurs_entreprise3 = WordCloud(background_color='white',
+                      stopwords=exclure_mots, max_words=80).generate(wd_principaux_interlocuteurs_entreprise3)
+    fig_wordcloud_principaux_interlocuteurs_entreprise3 = plt.figure(1,figsize=(12,15))
+    plt.imshow(wordcloud_principaux_interlocuteurs_entreprise3,interpolation="bicubic")
+    plt.axis('off')
+    elements_wordcloud_principaux_interlocuteurs.append(cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs_entreprise_moins_10", figure=fig_wordcloud_principaux_interlocuteurs_entreprise3, size="large", display="inline"))
+
+    options_entreprise4 = ['Plus de 100 employés'] 
+    df1_entreprise4 = df1.loc[df1['taille_entreprise'].isin(options_entreprise4)] 
+    wd_principaux_interlocuteurs_entreprise4 = " ".join(i for i in df1_entreprise4.principaux_interlocuteurs)
+    wordcloud_principaux_interlocuteurs_entreprise4 = WordCloud(background_color='white',
+                      stopwords=exclure_mots, max_words=80).generate(wd_principaux_interlocuteurs_entreprise4)
+    fig_wordcloud_principaux_interlocuteurs_entreprise4 = plt.figure(1,figsize=(12,15))
+    plt.imshow(wordcloud_principaux_interlocuteurs_entreprise4,interpolation="bicubic")
+    plt.axis('off')
+    elements_wordcloud_principaux_interlocuteurs.append(cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs_entreprise_plus_100", figure=fig_wordcloud_principaux_interlocuteurs_entreprise4, size="large", display="inline"))
     
     await cl.Message( content="Nuage de mots des principaux interlocuteurs",elements=elements_wordcloud_principaux_interlocuteurs,).send()