Update chatbot/database.py

chatbot/database.py

@@ -1,5 +1,5 @@
 import sqlite3
-from config import DB_PATH  # Import database path from config.py
+from config import DB_PATH # Import database path from config.py
 
 def create_table():
     """Creates the chat history table if it doesn't exist."""
@@ -8,7 +8,7 @@ def create_table():
     cursor.execute("""
         CREATE TABLE IF NOT EXISTS chat_history (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
-            user_id TEXT,
+            session_id TEXT,
             user_message TEXT,
             bot_response TEXT,
             timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
@@ -17,27 +17,24 @@ def create_table():
     conn.commit()
     conn.close()
 
-def get_chat_history(user_id):
-    """Fetches chat history for a given user."""
+def get_chat_history(session_id):
+    """Fetches chat history for a given session."""
     conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
-    cursor.execute("SELECT user_message, bot_response FROM chat_history WHERE user_id=? ORDER BY timestamp", 
-                   (str(user_id),))  # Ensure user_id is a string
+    cursor.execute("SELECT user_message, bot_response FROM chat_history WHERE session_id=? ORDER BY timestamp", 
+                   (session_id,))
     history = cursor.fetchall()
     conn.close()
     return history
 
-def save_chat(user_id, user_message, bot_response):
+def save_chat(session_id, user_message, bot_response):
     """Stores a chat message in the database."""
     conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
-    
-    # Ensure all values are stored as strings
-    cursor.execute("INSERT INTO chat_history (user_id, user_message, bot_response) VALUES (?, ?, ?)", 
-                   (str(user_id), str(user_message), str(bot_response)))
-    
+    cursor.execute("INSERT INTO chat_history (session_id, user_message, bot_response) VALUES (?, ?, ?)", 
+                   (session_id, user_message, bot_response))
     conn.commit()
     conn.close()
 
 # Initialize the database
-create_table()
+create_table()