Update app.py

app.py

@@ -21,11 +21,14 @@ llm = ChatAnthropic(
 items = [
     "Contacts Prescripteurs-Ensemble des données.csv",
     "Lieux-Ensemble des données.csv",
-    "Thèmes de séjour-Ensemble des données.csv"
+    "Thèmes de séjour-Ensemble des données.csv",
+    "Participants-Ensemble des données.csv",
+    "Séjours-Séjours 2025 - Réunion équipe.csv"
 ]
 
 # Fonction pour interroger les données CSV
 def query_data(query, csv_file_path):
+
     # Rediriger la sortie standard vers un buffer
     old_stdout = sys.stdout
     new_stdout = io.StringIO()
@@ -44,7 +47,7 @@ def query_data(query, csv_file_path):
         # number_of_head_rows=1
     )
     try:
-        response = agent.invoke(query)
+        response = agent.invoke(f"df.columns : \n{pd.read_csv(csv_file_path).columns}\n\n"+query)
         # Récupérer la sortie de la console
         console_output = new_stdout.getvalue()
 
@@ -54,7 +57,8 @@ def query_data(query, csv_file_path):
         # Retourner la réponse et la sortie de la console
         return {
             "output": response["output"],
-            "console_output": response['intermediate_steps']
+            "console_output":""
+            # "console_output": response['intermediate_steps']
         }
     except Exception as e:
         console_output = new_stdout.getvalue()
@@ -64,7 +68,7 @@ def query_data(query, csv_file_path):
         # Retourner la réponse et la sortie de la console
         return {
             "output": f"**Error :** {e}",
-            "console_output": console_output
+            "console_output": console_output+"\n"
         }
     
 
@@ -72,13 +76,11 @@ def query_data(query, csv_file_path):
 # Fonction de chat
 def chat(question, csv_file_path, chat_history):
     csv_file_path +=".csv"
-    question=f"df.columns : \n{pd.read_csv(csv_file_path).columns}\n\n"+question
-    print(question)
     # Appeler la fonction query_data
     result = query_data(question, csv_file_path)
     
     # Ajouter la question, la réponse finale et la sortie de la console à l'historique du chat
-    chat_history.append((question, f"**Réflexion de l'IA :**\n{result['console_output']}\n\n**Réponse finale**:\n{result['output']}"))
+    chat_history.append((question, f"{result['console_output']}{result['output']}"))
     
     # Retourner l'historique mis à jour
     return chat_history