babc0f58ef
Dockerfile: split into whisperx-base and runtime stages so app-only changes skip the 10min pip install on rebuild. Example compose updated. Player recap: replace hardcoded 'story so far' prompt with selectable styles (story/diary/bullets/custom) via dropdown in Setup and Settings. Settings: add NAT20_ONBOARDING_COMPLETED env var; fix merge logic so env vars properly override defaults (not just empty DB values).
53 lines
2.1 KiB
Python
53 lines
2.1 KiB
Python
from pathlib import Path
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
|
|
from .. import database as db, jobs
|
|
from ..pipeline.turns import load_turns
|
|
from ..pipeline.summarize import summarize_session
|
|
|
|
router = APIRouter(prefix="/api/sessions/{session_id}/notes", tags=["notes"])
|
|
|
|
|
|
@router.post("/generate")
|
|
def generate_notes(session_id: str):
|
|
session = db.get_conn().execute("SELECT * FROM sessions WHERE id = ?", (session_id,)).fetchone()
|
|
if not session:
|
|
raise HTTPException(404, "Session not found")
|
|
if not session["transcript_path"]:
|
|
raise HTTPException(400, "Session hasn't been transcribed yet")
|
|
|
|
settings = db.get_settings()
|
|
speaker_map = {r["raw_label"]: r["display_name"] for r in
|
|
db.get_conn().execute("SELECT raw_label, display_name FROM speakers WHERE session_id = ?", (session_id,))
|
|
if r["display_name"]}
|
|
|
|
job_id = db.create_job(session_id, "summarize")
|
|
|
|
def run(progress_cb):
|
|
turns = load_turns(Path(session["transcript_path"]), speaker_map)
|
|
dm_notes, player_recap = summarize_session(
|
|
turns, settings, progress_cb=progress_cb,
|
|
player_recap_style=settings.get("player_recap_style"),
|
|
player_recap_custom_prompt=settings.get("player_recap_custom_prompt"),
|
|
)
|
|
with db.tx() as conn:
|
|
conn.execute(
|
|
"INSERT INTO notes (session_id, dm_notes, player_recap, generated_at) VALUES (?, ?, ?, ?) "
|
|
"ON CONFLICT(session_id) DO UPDATE SET dm_notes=excluded.dm_notes, "
|
|
"player_recap=excluded.player_recap, generated_at=excluded.generated_at",
|
|
(session_id, dm_notes, player_recap, db.now()),
|
|
)
|
|
conn.execute("UPDATE sessions SET status = 'complete' WHERE id = ?", (session_id,))
|
|
|
|
jobs.submit(job_id, run)
|
|
return {"job_id": job_id}
|
|
|
|
|
|
@router.get("")
|
|
def get_notes(session_id: str):
|
|
row = db.get_conn().execute("SELECT * FROM notes WHERE session_id = ?", (session_id,)).fetchone()
|
|
if not row:
|
|
raise HTTPException(404, "Notes not generated yet")
|
|
return dict(row)
|