Update main.py

main.py

@@ -6,6 +6,8 @@ import numpy as np
 import plotly.express as px
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
+import matplotlib.pyplot as plt
+from wordcloud import WordCloud
 from typing import List
 from pathlib import Path
 from langchain_openai import ChatOpenAI
@@ -216,6 +218,7 @@ async def on_chat_start():
     df_interlocuteur = df1.groupby('principaux_interlocuteurs').size().reset_index(name='obs')
     fig_interlocuteur = px.bar(df_interlocuteur, x='obs', y='principaux_interlocuteurs', orientation='h', color='obs', title="Les principaux interlocuteurs du CAA", labels={'obs':'nombre'}, color_discrete_sequence=px.colors.qualitative.Safe, text_auto=True).update_layout(font=dict(size=9,color="RebeccaPurple"))
     elements.append(cl.Plotly(name="chart_interlocuteur", figure=fig_interlocuteur, display="inline", size="small"))
+    
     #await cl.sleep(2)
     df_interlocuteur_entreprise = df1.groupby(['principaux_interlocuteurs', 'taille_entreprise']).size().reset_index(name='obs')
     fig_interlocuteur_entreprise = px.bar(df_interlocuteur_entreprise, x='obs', y='principaux_interlocuteurs', orientation='h', color='taille_entreprise', title="Les principaux interlocuteurs du CAA par taille d'entreprise", labels={'obs':'nombre'}, color_discrete_sequence=px.colors.qualitative.Safe, text_auto=True).update_layout(font=dict(size=9,color="RebeccaPurple"))
@@ -399,6 +402,15 @@ async def on_chat_start():
     await cl.Message(content="📥 Télécharger l'ensemble des tableaux de données ayant servi à la datavisualisation", actions=actions).send()
 
     await cl.Message(content="Datavisualisation de l'enquête des recruteurs des chargé.e.s d'affaires en agencement", elements=elements).send()
+
+    wd_principaux_interlocuteurs = " ".join(i for i in df1.principaux_interlocuteurs)
+    wordcloud_principaux_interlocuteurs = WordCloud(background_color='white',
+                      max_words=40).generate(wd_principaux_interlocuteurs)
+    fig_wordcloud_principaux_interlocuteurs = plt.figure(1,figsize=(12,15))
+    plt.imshow(wordcloud_principaux_interlocuteurs,interpolation="bilinear")
+    plt.axis('off')
+    elements_wordcloud_principaux_interlocuteurs = [cl.Pyplot(name="plot_wordcloud_principaux_interlocuteurs", figure=fig_wordcloud_principaux_interlocuteurs, display="inline"),]
+    await cl.Message( content="Nuage de mots des principaux interlocuteurs",elements=elements_wordcloud_principaux_interlocuteurs,).send()
     
 @cl.on_message
 async def on_message(message: cl.Message):