import sqlite3

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

# Create personality_logs table (already existing)
c.execute("""
CREATE TABLE IF NOT EXISTS personality_logs (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    session_id INTEGER,
    trait TEXT,
    score REAL,
    FOREIGN KEY(session_id) REFERENCES user_sessions(id)
)
""")

# ✅ Add this new EI logs table
c.execute("""
CREATE TABLE IF NOT EXISTS ei_logs (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    session_id INTEGER,
    drill_type TEXT,
    response TEXT,
    timestamp TEXT,
    FOREIGN KEY(session_id) REFERENCES user_sessions(id)
)
""")

conn.commit()
conn.close()
print("✅ Tables created: personality_logs & ei_logs.")