import sqlite3

def init_db():
    conn = sqlite3.connect("data/mindmentor.db")
    c = conn.cursor()

    c.execute('''CREATE TABLE IF NOT EXISTS user_sessions (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT,
        session_date TEXT
    )''')

    c.execute('''CREATE TABLE IF NOT EXISTS cognitive_logs (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        session_id INTEGER,
        task_name TEXT,
        score INTEGER,
        reaction_time REAL
    )''')

    c.execute('''CREATE TABLE IF NOT EXISTS ei_logs (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        session_id INTEGER,
        drill_type TEXT,
        entry TEXT,
        timestamp TEXT
    )''')

    c.execute('''CREATE TABLE IF NOT EXISTS bigfive_scores (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        user_id INTEGER,
        trait TEXT,
        score INTEGER
    )''')

    conn.commit()
    conn.close()

if __name__ == "__main__":
    init_db()