Update main.py

main.py

@@ -5,6 +5,7 @@ from langchain.schema import StrOutputParser
 from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableLambda
 from langchain.schema.runnable.config import RunnableConfig
 from langchain.schema import StrOutputParser
+from langchain.document_loaders.csv_loader import CSVLoader
 import os
 import pandas as pd
 import numpy as np
@@ -37,13 +38,29 @@ async def on_chat_start():
         url="https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/Venus/logo-ofipe.jpg",
     ).send()
     
-    agent = create_csv_agent(
-        ChatAnthropic(temperature=1,model_name="claude-3-sonnet-20240229"),
-        "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/enquete-DClic-juin2022-dataviz-ord1.csv",
-        verbose=False,
-        agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-    )
-    cl.user_session.set("runnable", agent)
+    loader = CSVLoader(file_path="https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/FilesMarche.csv")
+
+    data = loader.load()
+    actions = []
+    for j in range(0,len(data)):
+        actions.append(cl.Action(name="selectRome", value=data[j].Number, label=data[j].Label),)
+
+    selectRome = await cl.AskActionMessage(
+        content="Sélectionnez une formation pour laquelle le COPILOT vous assistera :",
+        actions=actions,
+    ).send()
+
+    if selectRome and selectRome.get("name") == "selectRome":
+        await cl.Message(
+            content=f"Vous pouvez utiliser le COPILOT pour répondre à vos questions sur : \"{selectRome.get('label')}\", Codes ROME : \"{selectRome.get('value')}\"",
+        ).send()
+        agent = create_csv_agent(
+            ChatAnthropic(temperature=1,model_name="claude-3-sonnet-20240229"),
+            "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/EmploisRome/" + selectRome.get('value') + ".csv",
+            verbose=False,
+            agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+        )
+        cl.user_session.set("runnable", agent)
 
 @cl.on_message
 async def on_message(message: cl.Message):