Initial commit: Nat20 Notes — TTRPG session transcription & summarization

2026-07-06 23:45:54 +12:00
commit 0f9e9d4c5c
49 changed files with 2926 additions and 0 deletions
--- a/backend/app/database.py
+++ b/backend/app/database.py
@@ -0,0 +1,136 @@
+import json
+import sqlite3
+import threading
+import time
+import uuid
+from contextlib import contextmanager
+
+from . import config
+
+_local = threading.local()
+
+
+def get_conn():
+    if not hasattr(_local, "conn"):
+        _local.conn = sqlite3.connect(config.DB_PATH, check_same_thread=False)
+        _local.conn.row_factory = sqlite3.Row
+        _local.conn.execute("PRAGMA foreign_keys = ON")
+    return _local.conn
+
+
+@contextmanager
+def tx():
+    conn = get_conn()
+    try:
+        yield conn
+        conn.commit()
+    except Exception:
+        conn.rollback()
+        raise
+
+
+def init_db():
+    with tx() as conn:
+        conn.execute("""
+            CREATE TABLE IF NOT EXISTS settings (
+                key TEXT PRIMARY KEY,
+                value TEXT NOT NULL
+            )
+        """)
+        conn.execute("""
+            CREATE TABLE IF NOT EXISTS sessions (
+                id TEXT PRIMARY KEY,
+                name TEXT NOT NULL,
+                original_filename TEXT NOT NULL,
+                video_path TEXT,
+                audio_path TEXT,
+                transcript_path TEXT,
+                status TEXT NOT NULL DEFAULT 'uploaded',
+                created_at REAL NOT NULL
+            )
+        """)
+        conn.execute("""
+            CREATE TABLE IF NOT EXISTS speakers (
+                id TEXT PRIMARY KEY,
+                session_id TEXT NOT NULL REFERENCES sessions(id) ON DELETE CASCADE,
+                raw_label TEXT NOT NULL,
+                display_name TEXT,
+                UNIQUE(session_id, raw_label)
+            )
+        """)
+        conn.execute("""
+            CREATE TABLE IF NOT EXISTS notes (
+                session_id TEXT PRIMARY KEY REFERENCES sessions(id) ON DELETE CASCADE,
+                dm_notes TEXT,
+                player_recap TEXT,
+                generated_at REAL
+            )
+        """)
+        conn.execute("""
+            CREATE TABLE IF NOT EXISTS jobs (
+                id TEXT PRIMARY KEY,
+                session_id TEXT NOT NULL REFERENCES sessions(id) ON DELETE CASCADE,
+                job_type TEXT NOT NULL,        -- 'transcribe' | 'summarize'
+                status TEXT NOT NULL DEFAULT 'queued',  -- queued|running|done|error
+                progress TEXT,                 -- free-text progress message
+                error TEXT,
+                error_stage TEXT,               -- which pipeline stage failed, e.g. 'transcription'
+                error_detail TEXT,               -- full traceback, for logs/advanced view only
+                created_at REAL NOT NULL,
+                updated_at REAL NOT NULL
+            )
+        """)
+        # seed defaults if empty
+        existing = {r["key"] for r in conn.execute("SELECT key FROM settings")}
+        for k, v in config.DEFAULT_SETTINGS.items():
+            if k not in existing:
+                conn.execute("INSERT INTO settings (key, value) VALUES (?, ?)", (k, v))
+
+
+def get_settings() -> dict:
+    conn = get_conn()
+    rows = conn.execute("SELECT key, value FROM settings").fetchall()
+    return {r["key"]: r["value"] for r in rows}
+
+
+def update_settings(patch: dict):
+    with tx() as conn:
+        for k, v in patch.items():
+            conn.execute(
+                "INSERT INTO settings (key, value) VALUES (?, ?) "
+                "ON CONFLICT(key) DO UPDATE SET value=excluded.value",
+                (k, str(v)),
+            )
+
+
+def new_id() -> str:
+    return uuid.uuid4().hex[:12]
+
+
+def now() -> float:
+    return time.time()
+
+
+def create_job(session_id: str, job_type: str) -> str:
+    job_id = new_id()
+    with tx() as conn:
+        conn.execute(
+            "INSERT INTO jobs (id, session_id, job_type, status, created_at, updated_at) "
+            "VALUES (?, ?, ?, 'queued', ?, ?)",
+            (job_id, session_id, job_type, now(), now()),
+        )
+    return job_id
+
+
+def update_job(job_id: str, **fields):
+    if not fields:
+        return
+    fields["updated_at"] = now()
+    cols = ", ".join(f"{k} = ?" for k in fields)
+    with tx() as conn:
+        conn.execute(f"UPDATE jobs SET {cols} WHERE id = ?", (*fields.values(), job_id))
+
+
+def get_job(job_id: str) -> dict | None:
+    row = get_conn().execute("SELECT * FROM jobs WHERE id = ?", (job_id,)).fetchone()
+    return dict(row) if row else None