Update main.py

main.py

@@ -3,7 +3,6 @@ import json
 import bcrypt
 import pandas as pd
 import numpy as np
-import random
 from typing import List
 from pathlib import Path
 
@@ -30,7 +29,6 @@ literal_client = LiteralClient(api_key=os.getenv("LITERAL_API_KEY"))
 
 from surveycaa import surveyCaa
 
-id_sphinx = random.randint(100, 10000)
 @cl.password_auth_callback
 def auth_callback(username: str, password: str):
     auth = json.loads(os.environ['CHAINLIT_AUTH_LOGIN'])
@@ -67,8 +65,8 @@ def create_agent(filename: str):
     # Read the CSV file into a Pandas DataFrame.
     if cl.user_session.get("createdb") == None:
         df = pd.read_csv(filename)
-        engine = create_engine("sqlite:///sphinx" + str(id_sphinx) + ".db")
-        df.to_sql("sphinx" + str(id_sphinx), engine, index=False)
+        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)
         cl.user_session.set("createdb", "OK")
         cl.user_session.set("db", db)