deploy: switch to chatterbox requirements @ c5db0d8

steps/lang/_shared.py

@@ -95,12 +95,15 @@ def parse_json_array(raw: str) -> list:
     raise ValueError(f"Could not parse JSON array from LLM response:\n{raw[:200]}")
 
 
-def bedrock_converse(system_prompt: str, user_text: str, temperature: float = 0.1, step: str = "bedrock") -> str:
-    """Make a single Bedrock converse call and return the raw response text."""
+def bedrock_converse(system_prompt: str, user_text: str, temperature: float = 0.1, step: str = "bedrock", model_id=None) -> str:
+    """Make a single Bedrock converse call and return the raw response text.
+
+    model_id: optional override; defaults to the BEDROCK_MODEL env var.
+    """
     import boto3
 
     region = os.getenv("AWS_REGION", "us-east-1")
-    model_id = os.getenv("BEDROCK_MODEL", "qwen.qwen3-next-80b-a3b")
+    model_id = model_id or os.getenv("BEDROCK_MODEL", "qwen.qwen3-next-80b-a3b")
 
     client = boto3.client("bedrock-runtime", region_name=region)
     response = client.converse(

tools_api/router.py

@@ -13,10 +13,11 @@ from typing import Optional
 
 from fastapi import APIRouter, File, Form, HTTPException, Request, UploadFile
 from fastapi.responses import FileResponse, JSONResponse, PlainTextResponse
+from pydantic import BaseModel
 
 from server import limiter, _download_url, _is_allowed_video_host
 
-from . import audio_cleanup, dramabox, subtitles, voice_clone
+from . import audio_cleanup, dramabox, scene_writer, subtitles, voice_clone
 from .storage import (
     file_url,
     new_run_dir,
@@ -285,6 +286,44 @@ async def audio_cleanup_endpoint(
     })
 
 
+# ── Scene writer ─────────────────────────────────────────────────────
+
+class SceneRequest(BaseModel):
+    hook_pattern: str
+    hook_title: str = ""
+    register: str = ""
+    register_label: str = ""
+    niche: str
+
+
+@router.post("/scene")
+@limiter.limit("30/hour")
+async def scene_endpoint(request: Request, body: SceneRequest):
+    """Draft a directed DramaBox scene from a viral hook + the creator's niche."""
+    niche = body.niche.strip()
+    hook_pattern = body.hook_pattern.strip()
+    if not niche:
+        raise HTTPException(400, "niche is required")
+    if not hook_pattern:
+        raise HTTPException(400, "hook_pattern is required")
+    if len(niche) > 400:
+        raise HTTPException(400, "niche exceeds 400 char limit")
+
+    try:
+        scene = await asyncio.to_thread(
+            scene_writer.write_scene,
+            hook_pattern=hook_pattern,
+            hook_title=body.hook_title.strip(),
+            register=body.register.strip(),
+            register_label=body.register_label.strip(),
+            niche=niche,
+        )
+    except Exception as e:  # noqa: BLE001
+        raise HTTPException(500, f"Scene generation failed: {e}")
+
+    return JSONResponse({"scene": scene})
+
+
 # ── File download ────────────────────────────────────────────────────
 
 @router.get("/file/{run_id}/{filename}")

tools_api/scene_writer.py

@@ -0,0 +1,63 @@
+"""
+Hook → DramaBox scene drafting via AWS Bedrock (Amazon Nova).
+
+The frontend's "Adapt to my niche" step POSTs to /api/tools/scene; this turns a
+viral-hook structure plus the creator's niche into a directed DramaBox scene.
+"""
+from __future__ import annotations
+
+import os
+
+from steps.lang._shared import bedrock_converse
+
+# Amazon Nova Micro — the quickest/cheapest Nova text model. Override via env;
+# if Bedrock asks for an inference profile, set "us.amazon.nova-micro-v1:0".
+SCENE_MODEL = os.getenv("SCENE_BEDROCK_MODEL", "amazon.nova-micro-v1:0")
+
+_SYSTEM_PROMPT = """You are a scriptwriter for DramaBox, a text-to-speech engine that performs short, emotionally directed monologues.
+
+Write ONE scene from the given viral hook and the creator's niche.
+
+FORMAT — follow exactly:
+- Structure: <speaker description>, "<dialogue>" <stage action> "<more dialogue>"
+- Text inside double quotes is SPOKEN aloud, verbatim.
+- Text outside quotes is a STAGE DIRECTION, never spoken (e.g. She sighs deeply. A long pause.).
+- Phonetic sounds may go inside quotes: "Hahaha", "Mmmmm", "Ugh".
+- NEVER put Ahem, Pfft, Sigh, Gasp or Cough inside quotes — use a stage direction.
+
+CONTENT:
+- Follow the hook's structure, rewritten specifically for the creator's niche — do not just slot the topic in.
+- About 3 short sentences — one performable take, roughly 10-20 seconds spoken.
+- Output ONLY the scene text. No preamble, no markdown, no surrounding quotes."""
+
+
+def write_scene(
+    *,
+    hook_pattern: str,
+    hook_title: str,
+    register: str,
+    register_label: str,
+    niche: str,
+) -> str:
+    """Draft a directed DramaBox scene with Bedrock Nova. Raises on empty output."""
+    user_prompt = (
+        f'Viral hook: "{hook_title}"\n'
+        f"Hook structure: {hook_pattern}\n"
+        f"Emotional register: {register_label} ({register})\n"
+        f"Creator's niche / topic: {niche}\n\n"
+        "Write the DramaBox scene now."
+    )
+    raw = bedrock_converse(
+        _SYSTEM_PROMPT,
+        user_prompt,
+        temperature=0.8,
+        step="dramabox_scene",
+        model_id=SCENE_MODEL,
+    )
+    scene = (raw or "").strip()
+    # Strip a markdown fence if the model wrapped its output.
+    if scene.startswith("```"):
+        scene = scene.split("\n", 1)[-1].rsplit("```", 1)[0].strip()
+    if not scene:
+        raise RuntimeError("Scene model returned an empty response.")
+    return scene