session.py

import sqlite3
from uuid import uuid4

def init_session_db():
    conn = sqlite3.connect("sessions.db")
    conn.execute("CREATE TABLE IF NOT EXISTS sessions (id TEXT PRIMARY KEY, user TEXT)")
    conn.execute("CREATE TABLE IF NOT EXISTS messages (session_id TEXT, user_input TEXT, ai_output TEXT)")
    conn.commit()
    conn.close()

def create_session(user):
    sid = str(uuid4())
    with sqlite3.connect("sessions.db") as conn:
        conn.execute("INSERT INTO sessions (id, user) VALUES (?, ?)", (sid, user))
    return sid

def log_interaction(session_id, user_input, ai_output):
    with sqlite3.connect("sessions.db") as conn:
        conn.execute("INSERT INTO messages (session_id, user_input, ai_output) VALUES (?, ?, ?)",
                     (session_id, user_input, ai_output))

def get_session_logs(session_id):
    with sqlite3.connect("sessions.db") as conn:
        rows = conn.execute("SELECT user_input, ai_output FROM messages WHERE session_id=?", (session_id,)).fetchall()
    return rows