1from .domain.model import CaptionTrack, Clip, CropBox, FitSpec, OverlaySpec, Project, TransformSpec
2 
3 
4def project(
5    *,
6    width: int = 1080,
7    height: int = 1920,
8    fps: int = 30,
9    background: str = "black",
10    output: str = "final/output.mp4",
11    work_dir: str = "final/timeline-render",
12    open_after: bool = False,
13    overlay_styles: dict | None = None,
14    caption_styles: dict | None = None,
15) -> Project:
16    return Project(
17        width=width,
18        height=height,
19        fps=fps,
20        background=background,
21        output=output,
22        work_dir=work_dir,
23        open_after=open_after,
24        overlay_styles=overlay_styles or {},
25        caption_styles=caption_styles or {},
26    )
27 
28 
29def clip(
30    *,
31    id: str,
32    source: str,
33    label: str | None = None,
34    take: tuple[float | None, float | None] | None = None,
35    src_in: float | None = None,
36    src_out: float | None = None,
37    trim: tuple[float, float] | None = None,
38    trim_start: float = 0.0,
39    trim_end: float = 0.0,
40    speed: float = 1.0,
41    fit: FitSpec | None = None,
42    crop: CropBox | None = None,
43    transform: TransformSpec | None = None,
44    overlays: list[OverlaySpec] | None = None,
45    captions: CaptionTrack | None = None,
46    muted: bool = False,
47    volume: float = 1.0,
48) -> Clip:
49    if take is not None:
50        src_in, src_out = take
51    if trim is not None:
52        trim_start, trim_end = trim
53    return Clip(
54        id=id,
55        source=source,
56        label=label or id,
57        src_in=src_in,
58        src_out=src_out,
59        trim_start=trim_start,
60        trim_end=trim_end,
61        speed=speed,
62        fit=fit or contain(),
63        crop=crop,
64        transform=transform or TransformSpec(),
65        overlays=tuple(overlays or []),
66        captions=captions,
67        muted=muted,
68        volume=volume,
69    )
70 
71 
72def contain(*, background: str = "black") -> FitSpec:
73    return FitSpec(mode="contain", background=background)
74 
75 
76def cover(*, anchor: str = "center", background: str | None = None) -> FitSpec:
77    return FitSpec(mode="cover", anchor=anchor, background=background)
78 
79 
80def crop_box(x: float, y: float, width: float, height: float) -> CropBox:
81    return CropBox(x=x, y=y, width=width, height=height)
82 
83 
84def transform(
85    *,
86    hflip: bool = False,
87    vflip: bool = False,
88    zoom: float = 1.0,
89    video_pts: float = 1.0,
90    audio_tempo: float = 1.0,
91    eq: dict | None = None,
92) -> TransformSpec:
93    return TransformSpec(
94        hflip=hflip,
95        vflip=vflip,
96        zoom=zoom,
97        video_pts=video_pts,
98        audio_tempo=audio_tempo,
99        eq=eq or {},
100    )
101 
102 
103def text_overlay(
104    text: str,
105    *,
106    style: str = "default",
107    start: float | None = None,
108    end: float | None = None,
109    options: dict | None = None,
110    **overrides,
111) -> OverlaySpec:
112    merged = dict(options or {})
113    merged.update(overrides)
114    return OverlaySpec(
115        text=text,
116        style=style,
117        start=start,
118        end=end,
119        options=merged,
120    )
121 
122 
123def captions(
124    transcript: str,
125    *,
126    style: str = "karaoke",
127    mode: str = "karaoke",
128    trim: str = "word",
129    max_words_per_cue: int = 4,
130    source_offset: float = 0.0,
131    replacements: dict[str, str] | None = None,
132    skip_tokens: list[str] | tuple[str, ...] | None = None,
133) -> CaptionTrack:
134    return CaptionTrack(
135        transcript=transcript,
136        style=style,
137        mode=mode,
138        trim=trim,
139        max_words_per_cue=max_words_per_cue,
140        source_offset=source_offset,
141        replacements=replacements or {},
142        skip_tokens=tuple(skip_tokens or ()),
143    )
144 
145 
146__all__ = [
147    "project",
148    "clip",
149    "contain",
150    "cover",
151    "crop_box",
152    "transform",
153    "text_overlay",
154    "captions",
155    "CaptionTrack",
156    "Clip",
157    "CropBox",
158    "FitSpec",
159    "OverlaySpec",
160    "Project",
161    "TransformSpec",
162]
163