Update app.py

app.py

@@ -15,17 +15,22 @@ from langchain_core.messages import HumanMessage, AIMessage
 
 # --- Database Setup ---
 
-conn = sqlite3.connect("user.db")
-c = conn.cursor()
-c.execute('''CREATE TABLE IF NOT EXISTS users
-             (id INTEGER PRIMARY KEY AUTOINCREMENT,
-              username TEXT UNIQUE NOT NULL,
-              password TEXT NOT NULL,
-              previous_chat_history TEXT,
-              previous_products_bought TEXT)''')
-
-conn.commit()
-conn.close()
+# Database initialization with caching
+@st.cache_resource
+def init_db():
+    conn = sqlite3.connect('users.db', check_same_thread=False)
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS users
+                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
+                  username TEXT UNIQUE NOT NULL,
+                  password TEXT NOT NULL,
+                  previous_chat_history TEXT,
+                  previous_products_bought TEXT)''')
+    conn.commit()
+    return conn
+
+conn = init_db()
+
 
 class User:
     def __init__(self, id, username, password, chat_history = None, products_bought = None):