Update chatbot/database.py

chatbot/database.py

@@ -1,5 +1,5 @@
 import sqlite3
-from config import DB_PATH
+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,6 @@ 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,
@@ -18,33 +17,24 @@ def create_table():
     conn.commit()
     conn.close()
 
-def get_chat_sessions(user_id):
-    """Fetches all chat sessions 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 DISTINCT session_id FROM chat_history WHERE user_id=? ORDER BY timestamp DESC", (user_id,))
-    sessions = cursor.fetchall()
-    conn.close()
-    return [session[0] for session in sessions]
-
-def get_chat_history(user_id, session_id):
-    """Fetches chat history for a given user and session."""
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute("SELECT user_message, bot_response FROM chat_history WHERE user_id=? AND session_id=? ORDER BY timestamp", 
-                   (user_id, session_id))
+    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, session_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()
-    cursor.execute("INSERT INTO chat_history (user_id, session_id, user_message, bot_response) VALUES (?, ?, ?, ?)", 
-                   (user_id, session_id, user_message, 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()