import os
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))
os.chdir(ROOT)

from imaginaryarts import (
    Project,
    ScenesBuilder, VideoSceneBuilder, VideoSceneDirection,
    AssetsBuilder, VideoAssetBuilder,
    ImageModel, VideoModel, ImageDirection, ImageDirectionStyle,
    Voices, create_voiceover,
    VideoBuilder, AudioFile, CameraMotion, Color,
    DynamicText, DynamicTextFlow, TextStyle, Fonts, Shadow, Position, Effects,
    Grade, FilmLook,
)

project = Project(id="opium_wars", workdir=ROOT / "opium_wars")

scenes = ScenesBuilder(
    project=project,
    type="storyline",
    # Direction applying to the whole video (across all scenes)
    direction=VideoSceneDirection(
        subject=(
            "WIDE 16:9 cinematic compositions — this is a landscape YouTube video, "
            "so compose horizontally: sweeping panoramas, long rooms in "
            "perspective, subjects offset to a third with atmosphere filling the "
            "rest. The first five scenes are intimate second-person POV inside an "
            "opium den (the viewer IS the customer — show their hands, their tray, "
            "their view; never their face). The last five pull back to grand "
            "scale: harbours, fleets, an empire. Opium smoke belongs ONLY to the "
            "scenes inside the den that call for it — the other scenes are clear "
            "of smoke."
        ),
        environment=(
            "Guangzhou (Canton), China, 1839, at the end of the Qing dynasty. "
            "Period-accurate everything: a lantern-lit opium den with low wooden "
            "beds and silk cushions, changshan robes and queue braids, porcelain "
            "teapots and long bamboo opium pipes, the Thirteen Factories "
            "waterfront, junks and sampans crowding the Pearl River, British East "
            "India Company clipper ships, wooden opium chests. And for the final "
            "beat: young Queen Victoria, British imperial regalia, circa 1839. No "
            "anachronisms."
        ),
        details=(
            "This is an excerpt of a documentary-style explainer about the Opium "
            "Wars. It opens immersively — the viewer is a Chinese customs officer "
            "relaxing in an opium den — then reveals that they are complicit in "
            "British opium smuggling, and ends on the hook question of how the "
            "British Empire became a drug cartel. Mood arc: warm, hazy and "
            "seductive inside the den; then increasingly vast, cold and ominous "
            "as the true scale of the trade is revealed."
        ),
        motion=(
            "RESTRAINED 'living illustration' animation: slow, limited motion, not "
            "depicting full life-like events or actions. Keep the art style, palette "
            "and characters of the first frame."
        ),
    ),
    scenes=[
        VideoSceneBuilder(
            name="doors",
            voiceover="A sigh escapes you as you open the doors, and a smell like honey and flowers creeps up your nostrils.",
            direction=VideoSceneDirection(
                subject=(
                    "Second-person POV: your own hands pushing open heavy carved "
                    "wooden doors from a dark blue evening street into a glowing "
                    "golden-red opium den, warm lantern light spilling out toward "
                    "you across the threshold."
                ),
                motion=(
                    "The doors keep swinging slowly open, the wedge of warm light "
                    "widens across the frame, and the camera eases forward through "
                    "the doorway."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="den",
            voiceover="On low beds, men sprawl amidst cushions, smiling serenely through the haze.",
            direction=VideoSceneDirection(
                subject=(
                    "The den interior in wide view: a long lantern-lit room in "
                    "perspective, men in silk robes reclining on low wooden beds among "
                    "cushions, eyes half-closed and serene. One or two of them are "
                    "holding long bamboo opium pipes, resting them on their chests or "
                    "loosely in one hand. Layers of opium smoke drift in flat graceful "
                    "strata across the whole room."
                ),
                motion=(
                    "The lantern flames flicker, the layered smoke drifts slowly "
                    "sideways across the room, and one man's chest rises and falls. "
                    "The camera drifts slowly to the right."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="tray",
            voiceover="A young woman hands you a tray with dim sum, tea, and a pipe. Your usual.",
            direction=VideoSceneDirection(
                subject=(
                    "POV again: a young woman in an elegant robe offering a lacquered "
                    "tray directly toward the viewer — bamboo steamers of dim sum, a "
                    "small porcelain teapot and cup, and a long opium pipe laid "
                    "across it — her face gentle and knowing in the lamplight."
                ),
                motion=(
                    "She lowers the tray toward the viewer and holds it there, steady "
                    "and offered, the pipe clearly visible on it; steam curls from the "
                    "teapot. The camera holds still."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="inhale",
            # This scene's image prompt is authored but never generated:
            # continuity="extends" animates shot 3's final frame directly, so only
            # the video prompt is used and the motion line carries the whole beat.
            voiceover="With the first hit of opium, the edges of your world soften, and your guilt evaporates in a warm, weightless cloud.",
            direction=VideoSceneDirection(
                subject=(
                    "POV: your own hand has taken the long opium pipe from the tray "
                    "and holds it up toward the centre of the frame, close to the "
                    "camera, where your mouth would be; the woman and the lamplit den "
                    "still behind it."
                ),
                motion=(
                    "Your hand lifts the pipe from the tray and brings it toward the "
                    "camera, to the centre of the frame where your mouth would be, and "
                    "holds it there as you draw on it; the bowl glows and a thin thread "
                    "of smoke rises from it."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="dissolve",
            # SILENT — no voiceover at all. The high lands in a four-second pause
            # in the narration, which is why the timeline below is laid out by
            # hand instead of by tile(). Also extends: it continues out of the
            # inhale's last frame, so the two read as one take.
            direction=VideoSceneDirection(
                subject=(
                    "POV from the same position, now dissolving: the lamplit den "
                    "coming apart into soft drifting pink and rose smoke, the lanterns "
                    "blooming into weightless floating orbs, the walls and figures "
                    "unravelling into haze."
                ),
                motion=(
                    "Soft pink and rose smoke blooms out toward the camera and fills "
                    "the frame, and the room dissolves into it — walls, lanterns and "
                    "figures coming apart into drifting weightless haze — while the "
                    "camera tilts slowly backwards and down, as if you are lying back "
                    "on the bed."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="bribe",
            voiceover="Just today, foreign merchants bribed you to smuggle in the very stuff now curling through your lungs.",
            direction=VideoSceneDirection(
                subject=(
                    "A flashback beat, colder colours and clean dawn air: on a foggy dock at "
                    "dawn, a European merchant discreetly pressing a heavy pouch of "
                    "silver into the hands of a Chinese customs officer in uniform, "
                    "wooden chests being carried from a rowboat behind them, cold sea "
                    "mist lying low around their feet."
                ),
                motion=(
                    "The pouch passes from hand to hand and the officer's fingers close "
                    "around it; behind them a chest is carried along the dock; cold "
                    "mist drifts across the boards. The camera drifts slowly left."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="china",
            # The original beat six, split at its sentence break: one shot cannot
            # hold ~9.5 s of narration when Veo's ceiling is 8 s — and the
            # pull-back reads better as its own beat anyway.
            voiceover="We're in China, around 1839.",
            direction=VideoSceneDirection(
                subject=(
                    "The great pull-back begins: a high, wide establishing view of "
                    "the Pearl River at Guangzhou in hazy golden-indigo light — junks "
                    "and sampans scattered across the water, the roofs of the trading "
                    "city along the shore, clean air and thin morning river mist over "
                    "the far distance."
                ),
                motion=(
                    "The junks and sampans rock gently on the water, banners flutter, "
                    "and the camera pulls slowly back to reveal more of the harbour."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="harbour",
            voiceover="And this year, forty thousand chests of opium will flow through the port of Guangzhou, making the dealers fabulously rich.",
            direction=VideoSceneDirection(
                subject=(
                    "The same waterfront, now down at quay level: a sweeping wide "
                    "panorama of the Guangzhou docks — the Pearl River crowded with "
                    "junks, sampans and tall Western clipper ships, the trading houses "
                    "along the shore, endless stacks of wooden opium chests on the "
                    "quay, all under a clear golden-indigo sky."
                ),
                motion=(
                    "Dock workers carry opium chests along the quay toward the stacks, "
                    "the tall ships rock gently at anchor, and the camera pans slowly "
                    "to the right across the waterfront."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="fleet",
            voiceover="And you are a willing accomplice to the world's biggest narco kingpin: the British Empire.",
            direction=VideoSceneDirection(
                subject=(
                    "Ominous and iconic: a fleet of British East India Company ships "
                    "under sail at sea in formation, seen low from the waterline, their "
                    "dark hulls and tall pale sails towering over the viewer, heavy "
                    "storm clouds massing above them. Cold indigo sea, gold light "
                    "breaking behind the fleet. Clean, cold sea air."
                ),
                motion=(
                    "The fleet is under way: the ships move steadily forward through "
                    "the water, bow waves breaking white at their hulls and their sails "
                    "full, rising and falling on the swell as they come on. The camera "
                    "drifts slowly forward, low over the water."
                ),
            ),
        ),
        VideoSceneBuilder(
            name="queen",
            voiceover="But how did Queen Victoria become the head of a drug cartel?",
            direction=VideoSceneDirection(
                subject=(
                    "The closing hook image: young Queen Victoria enthroned in "
                    "imperial regalia, composed like a majestic official portrait — "
                    "and red opium poppies bloom among the royal ornaments and gold "
                    "carving at her feet, climbing the base of the throne. Regal, "
                    "beautiful, quietly damning. The air in the throne room is clear."
                ),
                motion=(
                    "Almost still: the poppies sway very faintly, the gold ornaments "
                    "glint, and the light shifts slowly across the throne. The camera "
                    "is locked off."
                ),
            ),
        ),
    ],
).build()

script = scenes.script
voiceover = create_voiceover(script, voice=Voices.AmericanExplainer, project=project)

# The shot table. One row per scene, read twice: `duration`/`continuity` at the
# asset stage, `fade_in`/`fade_out`/`motion` at the timeline stage.
#
# `duration` is an editing decision, not a default:
#   LONGER than the narration window  -> the clip is cut while still moving
#                                        (ambient shots: mist, water, crowds)
#   SHORTER than the narration window -> the clip plays out and freezes on a
#                                        composed frame (shots that hold a pose)
# Every clip is placed with fit="freeze", so a narrator who reads faster or
# slower than estimated only changes how long a settled frame holds — nothing
# ever loops or restarts.
SHOTS = [
    # 1 — the doors. trim: the first 1.5 s of the clip (the doors barely moving)
    # is cut off the FRONT, so the shot opens already in motion — a render-time
    # view of the same file, nothing regenerates. The clip is then 6.5 s in a
    # ~7.8 s window and freezes on its last frame, which the push covers and the
    # long handover into shot 2 eats half of.
    dict(duration=8000, continuity="none", trim=[1500, None],
         fade_in=300, out=Effects.FadeOut(700),
         motion=CameraMotion.zoom(start=1.0, end=1.06, easing="ease-in-out")),

    # 2 — the den. follows: the room glimpsed through the doorway at the end of
    # shot 1 has to be THIS room — same lanterns, same haze, same palette. The
    # clip outruns its 4.3-6.3 s window on purpose so the smoke never stops.
    # 700 ms in (street -> room is a big jump) and 800 ms out.
    dict(duration=6000, continuity="follows", trim=None,
         fade_in=700, out=Effects.FadeOut(800),
         motion=None),

    # 3 — the tray. First shot of the one continuous POV take. Short (4 s) on
    # purpose: it is the source frame for shot 4's "extends", so it MUST reach
    # its own last frame before the cut — a clip cut short here makes the hand
    # jump. It plays out and holds the offered tray, with a slow push that the
    # next two shots continue. No out transition: see shot 4.
    dict(duration=4000, continuity="none", trim=None,
         fade_in=800, out=None,
         motion=CameraMotion.zoom(start=1.00, end=1.05, easing="linear")),

    # 4 — the inhale. extends: no keyframe at all; Veo resumes from shot 3's
    # final frame, so this is the same take continuing. Hence no transition on
    # either side (a crossfade would dissolve a shot into itself), and the camera
    # picks the zoom up at exactly 1.05 and keeps pushing — a jump to a different
    # zoom value here is what makes a seamless continuation look jagged.
    # 6 s, not 8: like shot 3 it is an "extends" source and must play out.
    dict(duration=6000, continuity="extends", trim=None,
         fade_in=0, out=None,
         motion=CameraMotion.zoom(start=1.05, end=1.10, easing="linear")),

    # 5 — the dissolve. SILENT, exactly PAUSE_MS long, and extends again: the
    # smoke blooms and the room comes apart only after the narration has stopped.
    # The zoom picks up at 1.10 and eases back out to 1.00 as the camera tilts
    # back — the lean-back is in the video prompt, the widening frame here.
    # Hard out into the flashback.
    dict(duration=4000, continuity="extends", trim=None,
         fade_in=0, out=None,
         motion=CameraMotion.zoom(start=1.10, end=1.00, easing="ease-in-out")),

    # 6 — the bribe. Hard cut IN from the dream. Resolves into a held frame: the
    # fist closed around the silver, which the slow push keeps alive for up to
    # ~2.3 s of freeze on a slow read.
    dict(duration=6000, continuity="none", trim=None,
         fade_in=0, out=Effects.FadeOut(300),
         motion=CameraMotion.zoom(start=1.0, end=1.05, easing="linear")),

    # 7 — the pull-back. A short establishing beat; the clip outlives its
    # 1.9-2.7 s window in every case, so the river is still moving at the cut.
    # 900 ms out: the long dissolve into shot 8.
    dict(duration=4000, continuity="none", trim=None,
         fade_in=300, out=Effects.FadeOut(900),
         motion=None),

    # 8 — the harbour. follows: the same water, weather and light as shot 7,
    # seen from the quay instead of above — which is exactly why a 900 ms
    # cross-dissolve works here: the two frames share a horizon and blend rather
    # than fight. The longest beat in the piece, so on a slow read it can freeze
    # ~1.9 s; a slow push (not a pan — Veo is already panning right inside the
    # clip) carries the hold without fighting it.
    dict(duration=8000, continuity="follows", trim=None,
         fade_in=900, out=Effects.FadeOut(300),
         motion=CameraMotion.zoom(start=1.0, end=1.05, easing="linear")),

    # 9 — the fleet. The ships are making way, not just swaying; a matching slow
    # push continues that forward drift through any frozen tail. Its OUT effect
    # is the one non-fade transition in the piece: the Queen zooms in over the
    # fleet. 1200 ms because ffmpeg's zoomin holds the outgoing frame for its
    # first half and does all the movement in the second — and since the
    # transition straddles the cut, that dead half sits before the cut, where
    # the fleet is simply still playing.
    dict(duration=6000, continuity="none", trim=None,
         fade_in=300, out=Effects.ZoomIn(1200),
         motion=CameraMotion.zoom(start=1.0, end=1.05, easing="linear")),

    # 10 — the Queen. Nearly a still by design; the slow push does the work, and
    # a longer fade to black lets the closing question hang. fade_in=0: the
    # zoomin above owns this boundary.
    dict(duration=8000, continuity="none", trim=None,
         fade_in=0, out=Effects.FadeOut(600),
         motion=CameraMotion.zoom(start=1.0, end=1.06, easing="ease-in-out")),
]

# 3) Keyframes at 16:9, clips at 720p and silent — the narration and the music
#    carry the whole soundtrack, and audio=False roughly halves the clip price.
assets = AssetsBuilder(
    project=project,
    model=ImageModel(name="nanobanana2", size=(1920, 1080)),
    video_model=VideoModel(name="veo31", resolution="720p"),
    direction=ImageDirection(art_style=ImageDirectionStyle.OrientalLacquerInk()),
    assets=[
        VideoAssetBuilder.from_scene(scene, duration=shot["duration"],
                                     audio=False, continuity=shot["continuity"])
        for scene, shot in zip(scenes, SHOTS)
    ],
).build()

# 4) Placement, by hand. tile() refuses silent scenes — they have no voice onset
#    to time an image by — so this is tile()'s exact geometry written out: every
#    scene keeps its natural position inside the ONE continuous take, shifted by
#    a running offset that grows by SCENE_GAP at each cut (and by PAUSE_MS at the
#    silent one). Because the offset only ever grows, the render-time assembler
#    inserts real silence at the take's natural gap midpoints and the read never
#    sounds spliced.
LEAD_MS, SCENE_GAP, PAUSE_MS, TAIL_MS = 1000, 450, 4000, 3000

onsets, ends, off = [], [], None
for seg in voiceover.scenes:
    if seg is None:                       # the silent beat: hold the timeline open
        off += PAUSE_MS
        onsets.append(None)
        ends.append(None)
        continue
    off = (LEAD_MS - seg.start) if off is None else off + SCENE_GAP
    onsets.append(seg.start + off)        # absolute ms of this scene's first word
    ends.append(seg.end + off)

# Visual windows: every cut lands on a voice onset (shot 1 starts at 0), the
# silent shot owns exactly the PAUSE_MS before the next onset, and the last shot
# holds TAIL_MS past the last word.
starts = [0 if i == 0 else on for i, on in enumerate(onsets)]
for i, s in enumerate(starts):
    if s is None:                         # the silent shot, sized from its successor
        starts[i] = onsets[i + 1] - PAUSE_MS
bounds = [(starts[i], starts[i + 1] if i + 1 < len(starts) else ends[-1] + TAIL_MS)
          for i in range(len(starts))]

# The 16:9 canvas at Veo's own 24 fps, with the stills version's grade and film
# look kept byte-for-byte so the remake matches the original's colour.
video = VideoBuilder(
    project=project,
    background=Color(system="black"),
    size=(1920, 1080),
    grade=Grade(temperature=0.08, fade=0.12, contrast=0.05),
    film_look=FilmLook.LoFi().with_(soften_sigma=0.8, grain_amount=10),
    config={"fps": 24},
)

for i, (start, end) in enumerate(bounds):
    shot = SHOTS[i]

    # A trim is a render-time VIEW of the built clip — the asset is untouched.
    source = assets[i] if shot["trim"] is None else assets[i].trim(shot["trim"])

    clip = video.add_video(
        source,
        interval=[start, end],
        fit="freeze",          # never loop: a restart mid-drift reads as a glitch
        z_index=1,
    )
    # A boundary takes whichever side asks for the longer handover, and that
    # side's kind with it — so an out effect names the transition INTO the next
    # shot, and a fade_in only has to match it when it should win.
    if shot["fade_in"]:
        clip.add_effect(Effects.FadeIn(shot["fade_in"]))
    if shot["out"] is not None:
        clip.add_effect(shot["out"])

    # Camera motion only where it earns its keep. Veo already moves the camera
    # inside the clip with real parallax, so a zoompan on top of every shot just
    # looks busy — but a zoompan is the one thing that keeps a FROZEN tail alive,
    # because it goes on moving over the held frame. So: every shot that can
    # freeze gets a slow push, and the shots that are always cut get none.
    if shot["motion"] is not None:
        video.add_motion(shot["motion"], interval=[start, end])

    if onsets[i] is None:                 # the silent shot: no voice, no captions
        continue

    placed = video.add_voiceover(voiceover.scenes[i], at=onsets[i])

    # Landscape captions: smaller relative to the wide frame, low in the frame
    # like documentary subtitles rather than centre-screen phone captions.
    video.add_dynamic_text(DynamicText(
        words=placed.words,
        text_style=TextStyle(
            font=Fonts.Barlow,
            size=54,
            color=Color(system="white"),
            bold=True,
            shadow=Shadow.md,
        ),
        flow=DynamicTextFlow.WordByWord(max_words=30, max_lines=1),
        position=Position.BOTTOM,
    ))

# 5) An ethereal, ducked music bed — the only other sound in the piece, and the
#    only thing playing under the four-second hold.
video.add_audio(AudioFile("../songs/Ethereal.mp3").trim([0, None]), volume=0.13)

out_path = video.render()
print(f"\nDone → {out_path}")
Gallery
