Update helper.py

helper.py

@@ -24,12 +24,16 @@ ph = PasswordHasher()
 cipher_key = Fernet.generate_key()
 cipher = Fernet(cipher_key)
 
+# Database file path
+DB_FILE = 'grimvault.db'
+
 def get_db_connection():
-    conn = sqlite3.connect('database/grimvault.db')
+    conn = sqlite3.connect(DB_FILE)
     conn.row_factory = sqlite3.Row
     return conn
 
-def create_tables(conn):
+def create_tables():
+    conn = get_db_connection()
     c = conn.cursor()
     c.execute('''CREATE TABLE IF NOT EXISTS users
                  (username TEXT PRIMARY KEY, password_hash TEXT, embedding_hash TEXT, 
@@ -38,6 +42,7 @@ def create_tables(conn):
                  (id INTEGER PRIMARY KEY, username TEXT, filename TEXT, 
                  content BLOB, size INTEGER, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP)''')
     conn.commit()
+    conn.close()
 
 @spaces.GPU
 def get_embedding(text):
@@ -239,4 +244,7 @@ def record_login_attempt(username, success):
         del lockout_dict[username]
     else:
         lockout_dict[username][0] += 1
-        lockout_dict[username][1] = time.time()
+        lockout_dict[username][1] = time.time()
+
+# Initialize database
+create_tables()