Update init_db.py

init_db.py

@@ -1,17 +1,38 @@
+# 📄 File: init_db.py
+
 import sqlite3
 
-conn = sqlite3.connect('chat_history.db')
-c = conn.cursor()
+# Name of the database file
+DB_FILE = "chat_history.db"
+
+def init_db():
+    # Connect to the database (creates file if it doesn't exist)
+    conn = sqlite3.connect(DB_FILE)
+    c = conn.cursor()
+
+    # ✅ Create 'chat' table: stores individual messages in a session
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS chat (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            session_id TEXT NOT NULL,
+            timestamp TEXT NOT NULL,
+            user TEXT,
+            bot TEXT
+        )
+    ''')
+
+    # ✅ Create 'sessions' table: stores session metadata
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS sessions (
+            session_id TEXT PRIMARY KEY,
+            created_at TEXT NOT NULL
+        )
+    ''')
 
-# Create table
-c.execute('''
-CREATE TABLE IF NOT EXISTS chat (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    timestamp TEXT,
-    user TEXT,
-    bot TEXT
-)
-''')
+    conn.commit()
+    conn.close()
+    print("✅ Database initialized successfully.")
 
-conn.commit()
-conn.close()
+# Run the function if this script is executed directly
+if __name__ == "__main__":
+    init_db()