app/database.py

import sqlite3

def create_session_db():
    conn = sqlite3.connect(":memory:", check_same_thread=False)
    conn.row_factory = sqlite3.Row
    return conn

def get_session_db(session_id: str):
    from main import sessions
    return sessions.get(session_id, {}).get("conn")

def close_session_db(conn):
    conn.close()