Spaces:
Runtime error
Runtime error
Update main.py
Browse files
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)
|