database.py

import sqlite3

def init_conversation_db():
    conn = sqlite3.connect('conversation_history.db')
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS conversations (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            session_id TEXT,
            user_query TEXT,
            ai_response TEXT,
            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    conn.commit()
    conn.close()

init_conversation_db()














# import sqlite3

# def create_messages_db():
#     conn = sqlite3.connect('MASTER.db')
#     cursor = conn.cursor()
#     cursor.execute("""
#         CREATE TABLE IF NOT EXISTS Messages (
#             message_id INTEGER PRIMARY KEY AUTOINCREMENT,
#             thread_id TEXT,
#             role TEXT,
#             message TEXT,
#             user_id INTEGER,
#             FOREIGN KEY(user_id) REFERENCES Users(user_id)
#         )
#     """)
#     conn.commit()
#     conn.close()


# def write_to_messages_db(thread_id, role, message):
#     conn = sqlite3.connect('MASTER.db')
#     cursor = conn.cursor()
#     insert_query = """
#         INSERT INTO Messages (thread_id, role, message)
#         VALUES (?, ?, ?)
#     """
#     cursor.execute(insert_query, (thread_id, role, message))
#     conn.commit()
#     conn.close()


# def get_all_thread_messages(thread_id):
#     conn = sqlite3.connect('MASTER.db')
#     cursor = conn.cursor()
#     cursor.execute('''SELECT message 
#                       FROM Messages
#                       WHERE thread_id = ?
#                       ORDER BY message_id''', (thread_id,))
#     rows = cursor.fetchall()
#     conn.close()
#     return [i[0] for i in rows]


# def get_unique_thread_ids():
#     conn = sqlite3.connect('MASTER.db')
#     cursor = conn.cursor()
#     cursor.execute("SELECT DISTINCT thread_id FROM Messages")
#     unique_thread_ids = reversed([row[0] for row in cursor.fetchall()])
#     conn.close()
#     return unique_thread_ids


# def create_documents_db():
#     pass


# def get_uploaded_doc_names():
#     pass