Update main.py

main.py

@@ -3,6 +3,7 @@ import json
 import bcrypt
 import pandas as pd
 import numpy as np
+import plotly.express as px
 from typing import List
 from pathlib import Path
 from langchain_openai import ChatOpenAI
@@ -141,15 +142,34 @@ async def chat_profile():
         cl.ChatProfile(name="Traitement des données d'enquête : «Expé CFA : questionnaire auprès des professionnels de la branche de l'agencement»",markdown_description="Vidéo exploratoire autour de l'événement",icon="/public/logo-ofipe.png",),
     ]
     
-@cl.set_starters
-async def set_starters():
-    return [
-        cl.Starter(
-            label="Répartition du nombre de CAA dans les entreprises",
-            message="Quel est le nombre de chargé.e d'affaires en agencement dans chaque type d'entreprises?",
-            icon="/public/request-theme.svg",
-            )
+#@cl.set_starters
+#async def set_starters():
+#    return [
+#        cl.Starter(
+#            label="Répartition du nombre de CAA dans les entreprises",
+#            message="Quel est le nombre de chargé.e d'affaires en agencement dans chaque type d'entreprises?",
+#            icon="/public/request-theme.svg",
+#            )
+#    ]
+
+@cl.on_chat_start
+async def on_chat_start():
+    await cl.Message(f"> SURVEYIA").send()
+    figures = ""
+    df = pd.read_csv('survey.csv')
+    df_taille = df.groupby('taille_entreprise').size().reset_index(name='obs')
+    figures = figures + "<div class='gridvid'>" + px.bar(df_temps, 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).show() + "</div>"
+    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'}, color_discrete_sequence=px.colors.qualitative.Light24, text_auto=True).update_layout(showlegend=False).show() + "</div>"
+    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'}, color_discrete_sequence=px.colors.qualitative.Light24, text_auto=True).show() + "</div>"
+    sources = [
+        cl.Text(name="Datavisualisation", content=figures, display="inline")
     ]
+    await cl.Message(
+        content="Représentations : ",
+        elements=sources,
+    ).send()
     
 @cl.on_message
 async def on_message(message: cl.Message):