Update main.py

main.py

@@ -13,11 +13,7 @@ import numpy as np
 
 from langchain.agents import AgentExecutor
 from langchain.agents.agent_types import AgentType
-#from langchain_experimental.agents.agent_toolkits import create_csv_agent
-
-from langchain_community.agent_toolkits import create_sql_agent
-from langchain_community.utilities import SQLDatabase
-from sqlalchemy import create_engine
+from langchain_experimental.agents.agent_toolkits import create_csv_agent
 
 from deep_translator import GoogleTranslator
 
@@ -71,22 +67,13 @@ async def on_chat_start():
         llm = HuggingFaceEndpoint(
             repo_id=repo_id, temperature=1.0
         )
-        
-        
-        df = pd.read_csv("https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/EmploisRome/" + selectRomes[1:] + ".csv")
-        engine = create_engine("sqlite:///sphinx" + str(cl.user_session.get("id")) + ".db")
-        df.to_sql("sphinx" + str(cl.user_session.get("id")), engine, index=False)
-        db = SQLDatabase(engine=engine)
-        agent = create_sql_agent(llm, db=db, verbose=True)
-        
-        
-        #agent = create_csv_agent(
-        #    llm,
-        #    "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/EmploisRome/" + selectRomes[1:] + ".csv",
-        #    verbose=False,
-        #    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-        #    max_iterations=5,
-        #)
+        agent = create_csv_agent(
+            llm,
+            "https://cipen.univ-gustave-eiffel.fr/typo3conf/ext/cipen_package/Resources/Public/Dataviz/datalab/crossfilterings/EmploisRome/" + selectRomes[1:] + ".csv",
+            verbose=False,
+            agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+            max_iterations=5,
+        )
         cl.user_session.set("runnable", agent)
 
 @cl.on_message