Update main.py

main.py

@@ -1,6 +1,4 @@
 from langchain_community.llms import HuggingFaceEndpoint
-from langchain_community.utilities import SQLDatabase
-from sqlalchemy import create_engine
 from langchain_core.prompts import PromptTemplate
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain.memory import ChatMessageHistory, ConversationBufferMemory
@@ -8,16 +6,14 @@ 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_community.agent_toolkits import create_sql_agent
 
 import os
 import pandas as pd
 import numpy as np
-import random
 
 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_csv_agent
 
 from deep_translator import GoogleTranslator
 
@@ -48,25 +44,12 @@ async def on_chat_start():
     llm = HuggingFaceEndpoint(
         repo_id=repo_id, temperature=1.0
     )
-    idDb = random.randrange(0, 1001, 3)
-    if cl.user_session.get("createdb") == None:
-        df = pd.read_csv("https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/enquete-DClic-juin2022-dataviz-ord1.csv")
-        engine = create_engine("sqlite:///epnum" + str(idDb) + ".db")
-        df.to_sql("epnum" + str(idDb), engine, index=False)
-        db = SQLDatabase(engine=engine)
-        cl.user_session.set("createdb", "OK")
-        cl.user_session.set("db", db)
-    else:
-        db = cl.user_session.get("db")
-    
-    agent = create_sql_agent(llm, db=db, agent_type="zero-shot-react-description", verbose=True)
-    
-    #agent = create_csv_agent(
-    #    llm,
-    #    "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/enquete-DClic-juin2022-dataviz-ord1.csv",
-    #    verbose=True,
-    #    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
-    #)
+    agent = create_csv_agent(
+        llm,
+        "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/enquete-DClic-juin2022-dataviz-ord1.csv",
+        verbose=True,
+        agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
+    )
     cl.user_session.set("runnable", agent)
 
 @cl.on_message