Update main.py

main.py

@@ -405,12 +405,24 @@ async def on_chat_start():
 
     wd_principaux_interlocuteurs = " ".join(i for i in df1.principaux_interlocuteurs)
     exclure_mots = ['d', 'du', 'de', 'la', 'des', 'le', 'et', 'est', 'elle', 'une', 'en', 'que', 'aux', 'qui', 'ces', 'les', 'dans', 'sur', 'l', 'un', 'pour', 'par', 'il', 'ou', 'à', 'ce', 'a', 'sont', 'cas', 'plus', 'leur', 'se', 's', 'vous', 'au', 'c', 'aussi', 'toutes', 'autre', 'comme']
+    elements_wordcloud_principaux_interlocuteurs = []
     wordcloud_principaux_interlocuteurs = WordCloud(background_color='white',
                       stopwords=exclure_mots, max_words=80).generate(wd_principaux_interlocuteurs)
     fig_wordcloud_principaux_interlocuteurs = plt.figure(1,figsize=(12,15))
     plt.imshow(wordcloud_principaux_interlocuteurs,interpolation="bicubic")
     plt.axis('off')
-    elements_wordcloud_principaux_interlocuteurs = [cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs", figure=fig_wordcloud_principaux_interlocuteurs, size="large", display="inline"),]
+    elements_wordcloud_principaux_interlocuteurs.append(cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs", figure=fig_wordcloud_principaux_interlocuteurs, size="large", display="inline"))
+    
+    options_entreprise1 = ['10 à 50 employés'] 
+    df1_entreprise1 = df1.loc[df1['taille_entreprise'].isin(options_entreprise1)] 
+    wd_principaux_interlocuteurs_entreprise1 = " ".join(i for i in df1_entreprise1.principaux_interlocuteurs)
+    wordcloud_principaux_interlocuteurs_entreprise1 = WordCloud(background_color='white',
+                      stopwords=exclure_mots, max_words=80).generate(wd_principaux_interlocuteurs_entreprise1)
+    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"))
+    
     await cl.Message( content="Nuage de mots des principaux interlocuteurs",elements=elements_wordcloud_principaux_interlocuteurs,).send()
     
 @cl.on_message