datacipen commited on
Commit
131858c
·
verified ·
1 Parent(s): 3fc6b8e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -2
main.py CHANGED
@@ -13,6 +13,7 @@ from langchain_community.agent_toolkits import create_sql_agent
13
  import os
14
  import pandas as pd
15
  import numpy as np
 
16
 
17
  from langchain.agents import AgentExecutor
18
  from langchain.agents.agent_types import AgentType
@@ -47,10 +48,11 @@ async def on_chat_start():
47
  llm = HuggingFaceEndpoint(
48
  repo_id=repo_id, temperature=1.0
49
  )
 
50
  if cl.user_session.get("createdb") == None:
51
  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")
52
- engine = create_engine("sqlite:///epnum.db")
53
- df.to_sql("epnum", engine, index=False)
54
  db = SQLDatabase(engine=engine)
55
  cl.user_session.set("createdb", "OK")
56
  cl.user_session.set("db", db)
 
13
  import os
14
  import pandas as pd
15
  import numpy as np
16
+ import random
17
 
18
  from langchain.agents import AgentExecutor
19
  from langchain.agents.agent_types import AgentType
 
48
  llm = HuggingFaceEndpoint(
49
  repo_id=repo_id, temperature=1.0
50
  )
51
+ idDb = randrange(0, 101, 2)
52
  if cl.user_session.get("createdb") == None:
53
  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")
54
+ engine = create_engine("sqlite:///epnum" + idDb + ".db")
55
+ df.to_sql("epnum" + idDb, engine, index=False)
56
  db = SQLDatabase(engine=engine)
57
  cl.user_session.set("createdb", "OK")
58
  cl.user_session.set("db", db)