Update main.py

main.py

@@ -137,6 +137,26 @@ def write_response(response_dict: dict):
     # Check if the response is an answer.
     return response_dict["answer"]
 
+@cl.action_callback("download")
+async def on_action(action):
+    content = []
+    content.append(action.value)
+    arrayContent = np.array(content)
+    df = pd.DataFrame(arrayContent)
+    with open('./public/' + action.description + '.txt', 'wb') as csv_file:
+        df.to_csv(path_or_buf=csv_file, index=False,header=False, encoding='utf-8')
+    elements = [
+        cl.File(
+            name= action.description + ".txt",
+            path="./public/" + action.description + ".txt",
+            display="inline",
+        ),
+    ]
+    await cl.Message(
+        author="COPILOT", content="[Lien] 🔗", elements=elements
+    ).send()
+    await action.remove()
+    
 @cl.set_chat_profiles
 async def chat_profile():
     return [
@@ -221,6 +241,9 @@ async def on_chat_start():
     content_all="<p>Tableaux des données de La \"taille des entreprises ayant répondu\"</p>" + df_taille.to_html() + "<p>Tableaux des données de \"L’engagement dans le domaine de l’agencement\"</p>" + df_temps.to_html() + "<p>Tableaux des données de \"L’engagement dans le domaine de l’agencement\" par taille d'entreprise</p>" + df_temps_entreprise.to_html() + "<p>Tableaux des données de \"Le nombre de chargé.e d’affaires en agencement\"</p>" + df_nb_charge.to_html() + "<p>Tableaux des données de \"Le nombre de chargé.e d’affaires en agencement par taille d'entreprise\"</p>" + df_nb_charge_entreprise.to_html() + "<p>Tableaux des données de \"Le nombre de chargé.e d’affaires en agencement par année d'engagement\"</p>" + df_nb_charge_engagement.to_html() + "<p>Tableaux des données de \"Le profil des répondants\"</p>" + df_statut.to_html() + "<p>Tableaux des données de \"Les principaux interlocuteurs du CAA\"</p>" + df_interlocuteur.to_html() + "<p>Tableaux des données de \"Les principaux interlocuteurs du CAA\" par taille d'entreprise</p>" + df_interlocuteur_entreprise.to_html() + "<p>Tableaux des données de \"Les principaux interlocuteurs du CAA\" par nombre chargé.e d'affaires</p>" + df_interlocuteur_nb_charge.to_html() + "<p>Tableaux des données de \"Les principales compétences attendues\"</p>" + df_competences.to_html() + "<p>Tableaux des données de \"Les principales compétences attendues\" par taille d'entreprise</p>" + df_competences_entreprise.to_html() + "<p>Tableaux des données de \"Les principales compétences attendues\" par nombre chargé.e d'affaires</p>" + df_competences_nb_charge.to_html()
     tableau_all = [cl.Text(name="Tableaux", content=content_all, display="side")]
     await cl.Message(content="📚 Tableaux de toutes les données", elements=tableau_all,).send()
+    actions = [cl.Action(name="download", value=content_all, description="download_tableau")]
+    await cl.Message(author="COPILOT",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()
     
 @cl.on_message