deploy: switch to chatterbox requirements @ d36fc80

tools_api/scene_writer.py

@@ -1,18 +1,25 @@
 """
-Hook → DramaBox scene drafting via AWS Bedrock (Amazon Nova).
+Hook → DramaBox scene drafting via an LLM.
 
 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.
+
+Provider chain: AWS Bedrock (primary) → Pollinations (fallback). If both fail,
+the endpoint errors and the frontend falls back to its offline template engine.
 """
 from __future__ import annotations
 
 import os
 
-from steps.lang._shared import bedrock_converse
+from steps.lang._shared import MODEL, bedrock_converse, build_client, log_llm_call
+
+# Bedrock model for scene drafting. Defaults to qwen.qwen3-next-80b-a3b — the
+# same model the backend's translation step uses, so it's known-good on this
+# account (works with Converse, no inference-profile requirement). Override via
+# env SCENE_BEDROCK_MODEL.
+SCENE_MODEL = os.getenv("SCENE_BEDROCK_MODEL", "qwen.qwen3-next-80b-a3b")
 
-# 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")
+_TEMPERATURE = 0.8
 
 _SYSTEM_PROMPT = """You are a scriptwriter for DramaBox, a text-to-speech engine that performs short, emotionally directed monologues.
 
@@ -31,33 +38,97 @@ CONTENT:
 - Output ONLY the scene text. No preamble, no markdown, no surrounding quotes."""
 
 
-def write_scene(
-    *,
+def _build_user_prompt(
     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 = (
+    return (
         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."
     )
+
+
+def _clean(scene: str) -> str:
+    """Strip whitespace and a markdown fence if the model wrapped its output."""
+    s = (scene or "").strip()
+    if s.startswith("```"):
+        s = s.split("\n", 1)[-1].rsplit("```", 1)[0].strip()
+    return s
+
+
+def _bedrock_scene(user_prompt: str) -> str:
+    """Primary provider — AWS Bedrock."""
     raw = bedrock_converse(
         _SYSTEM_PROMPT,
         user_prompt,
-        temperature=0.8,
+        temperature=_TEMPERATURE,
         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()
+    return _clean(raw)
+
+
+def _pollinations_scene(user_prompt: str) -> str:
+    """Fallback provider — Pollinations (OpenAI-compatible, see _shared.py)."""
+    client = build_client()
+    response = client.chat.completions.create(
+        model=MODEL,
+        messages=[
+            {"role": "system", "content": _SYSTEM_PROMPT},
+            {"role": "user", "content": user_prompt},
+        ],
+        temperature=_TEMPERATURE,
+    )
+    raw = (response.choices[0].message.content or "").strip()
+    log_llm_call(
+        step="dramabox_scene",
+        provider="pollinations",
+        model=MODEL,
+        system_prompt=_SYSTEM_PROMPT,
+        user_prompt=user_prompt,
+        response=raw,
+        temperature=_TEMPERATURE,
+    )
+    return _clean(raw)
+
+
+def write_scene(
+    *,
+    hook_pattern: str,
+    hook_title: str,
+    register: str,
+    register_label: str,
+    niche: str,
+) -> str:
+    """Draft a directed DramaBox scene — Bedrock primary, Pollinations fallback.
+
+    Raises if both providers fail; the caller surfaces a 500 and the frontend
+    falls back to its offline template engine.
+    """
+    user_prompt = _build_user_prompt(
+        hook_pattern, hook_title, register, register_label, niche
+    )
+
+    # Primary — Bedrock.
+    try:
+        scene = _bedrock_scene(user_prompt)
+        if scene:
+            return scene
+        print("[scene] Bedrock returned an empty scene — trying Pollinations.")
+    except Exception as e:  # noqa: BLE001
+        print(f"[scene] Bedrock failed ({e}) — trying Pollinations.")
+
+    # Fallback — Pollinations.
+    try:
+        scene = _pollinations_scene(user_prompt)
+    except Exception as e:  # noqa: BLE001
+        raise RuntimeError(f"Bedrock and Pollinations both failed: {e}")
     if not scene:
-        raise RuntimeError("Scene model returned an empty response.")
+        raise RuntimeError("Bedrock and Pollinations both returned empty scenes.")
     return scene