Update main.py

main.py

@@ -155,21 +155,18 @@ async def chat_profile():
 @cl.on_chat_start
 async def on_chat_start():
     await cl.Message(f"> SURVEYIA").send()
-    figures = " "
+    elements = []
     df = pd.read_csv('./public/survey.csv')
     df_taille = df.groupby('taille_entreprise').size().reset_index(name='obs')
-    figures = figures + "<div class='gridvid'>" + px.bar(df_taille, x='taille_entreprise', y='obs', color='obs', title="La taille des entreprises ayant répondu", labels={'obs':'nombre'}, text_auto=True).update_layout(showlegend=False).show() + "</div>"
+    fig_taille = px.bar(df_taille, x='taille_entreprise', y='obs', color='obs', title="La taille des entreprises ayant répondu", labels={'obs':'nombre'}, color_discrete_sequence=px.colors.qualitative.Light24, text_auto=True).update_layout(showlegend=False)
+    elements.append(cl.Plotly(name="chart_taille", figure=fig_taille, display="inline"))
     df_temps = df.groupby('temps_active_domaine_agencement').size().reset_index(name='obs')
-    figures = figures + "<div class='gridvid'>" + px.bar(df_temps, x='temps_active_domaine_agencement', y='obs', color='obs', title="L’engagement dans le domaine de l’agencement", labels={'obs':'nombre'}, text_auto=True).update_layout(showlegend=False).show() + "</div>"
+    fig_temps = px.bar(df_temps, x='temps_active_domaine_agencement', y='obs', color='obs', title="L’engagement dans le domaine de l’agencement", labels={'obs':'nombre'}, color_discrete_sequence=px.colors.qualitative.Light24, text_auto=True).update_layout(showlegend=False)
+    elements.append(cl.Plotly(name="chart_temps", figure=fig_temps, display="inline"))
     df_temps_entreprise = df.groupby(['temps_active_domaine_agencement', 'taille_entreprise']).size().reset_index(name='obs')
-    figures = figures + "<div class='gridvid'>" + px.bar(df_temps_entreprise, x='temps_active_domaine_agencement', y='obs', color='taille_entreprise', title="L’engagement dans le domaine de l’agencement par taille d'entreprise", labels={'obs':'nombre'}, text_auto=True).show() + "</div>"
-    sources = [
-        cl.Text(name="Datavisualisation", content=figures, display="inline")
-    ]
-    await cl.Message(
-        content="Représentations : ",
-        elements=sources,
-    ).send()
+    fig_temps_entreprise = px.bar(df_temps_entreprise, x='temps_active_domaine_agencement', y='obs', color='taille_entreprise', title="L’engagement dans le domaine de l’agencement par taille d'entreprise", labels={'obs':'nombre'}, color_discrete_sequence=px.colors.qualitative.Light24, text_auto=True)
+    elements.append(cl.Plotly(name="chart_temps_entreprise", figure=fig_temps_entreprise, display="inline"))
+    await cl.Message(content="Datavisualisation", elements=elements).send()
     
 @cl.on_message
 async def on_message(message: cl.Message):