Update main.py

main.py

@@ -10,7 +10,7 @@ from langchain.schema import StrOutputParser
 
 from langchain.agents import AgentExecutor
 from langchain.agents.agent_types import AgentType
-from langchain_experimental.agents.agent_toolkits import create_csv_agent
+from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent, create_csv_agent
 
 
 import chainlit as cl
@@ -46,16 +46,21 @@ async def set_starters():
 @cl.on_message
 async def on_message(message: cl.Message):
     await cl.Message(f"> SURVEYIA").send()
+    df = pd.read_csv(
+        "https://raw.githubusercontent.com/pandas-dev/pandas/main/doc/data/titanic.csv"
+    )
     model = await LLMistral()
     
-    agent = create_csv_agent(
-        model,
-        "./public/ExpeCFA_LP_CAA.csv",
-        verbose=False,
-        allow_dangerous_code=True,
-        handle_parsing_errors=True,
-        agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
-    )
+    agent = create_pandas_dataframe_agent(model, df, verbose=True)
+    
+    #agent = create_csv_agent(
+    #    model,
+    #    "./public/ExpeCFA_LP_CAA.csv",
+    #    verbose=False,
+    #    allow_dangerous_code=True,
+    #    handle_parsing_errors=True,
+    #    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
+    #)
     
     msg = cl.Message(content="")