| from __future__ import annotations |
|
|
| from copy import deepcopy |
| from dataclasses import dataclass, field |
| from typing import Any |
|
|
|
|
| @dataclass(frozen=True) |
| class CreativeStyle: |
| key: str |
| label: str |
| description: str |
| platform: str |
| template: str |
| scene_duration: float |
| transition_sequence: tuple[str, ...] |
| scene_effect_sequence: tuple[str, ...] |
| render_defaults: dict[str, Any] = field(default_factory=dict) |
| metadata: dict[str, Any] = field(default_factory=dict) |
|
|
| def metadata_payload(self) -> dict[str, Any]: |
| return { |
| "key": self.key, |
| "label": self.label, |
| "description": self.description, |
| "platform": self.platform, |
| "template": self.template, |
| "scene_duration": self.scene_duration, |
| "transition_sequence": list(self.transition_sequence), |
| "scene_effect_sequence": list(self.scene_effect_sequence), |
| "render_defaults": deepcopy(self.render_defaults), |
| "metadata": deepcopy(self.metadata), |
| } |
|
|
|
|
| SCENE_EFFECTS: dict[str, dict[str, str]] = { |
| "none": {"label": "None", "filter": ""}, |
| "sharp_pop": { |
| "label": "Sharp Pop", |
| "filter": "eq=contrast=1.08:saturation=1.20:brightness=0.01,unsharp=5:5:0.8:3:3:0.4", |
| }, |
| "clean_beauty": { |
| "label": "Clean Beauty", |
| "filter": "hqdn3d=1.5:1.5:6:6,eq=saturation=1.08:contrast=1.03", |
| }, |
| "warm_glow": { |
| "label": "Warm Glow", |
| "filter": "eq=contrast=1.04:saturation=1.18:gamma_r=1.04:gamma_b=0.96,gblur=sigma=0.25", |
| }, |
| "cinematic": { |
| "label": "Cinematic", |
| "filter": "eq=contrast=1.14:saturation=0.95:brightness=-0.015,vignette=PI/6", |
| }, |
| "dreamy": { |
| "label": "Dreamy", |
| "filter": "gblur=sigma=0.6,eq=contrast=1.04:saturation=1.18:brightness=0.02", |
| }, |
| "flash_pop": { |
| "label": "Flash Pop", |
| "filter": "eq=contrast=1.16:saturation=1.25:brightness=0.035", |
| }, |
| "grain": { |
| "label": "Fine Grain", |
| "filter": "noise=alls=8:allf=t+u,eq=contrast=1.07:saturation=1.02", |
| }, |
| "motion_blur": { |
| "label": "Motion Blur", |
| "filter": "tmix=frames=3:weights='1 2 1',eq=contrast=1.05:saturation=1.08", |
| }, |
| "noir": { |
| "label": "Noir", |
| "filter": "hue=s=0,eq=contrast=1.18:brightness=-0.02", |
| }, |
| "glitch": { |
| "label": "Glitch", |
| "filter": "rgbashift=rh=4:bh=-4,eq=contrast=1.12:saturation=1.18", |
| }, |
| "rgb_split": { |
| "label": "RGB Split", |
| "filter": "rgbashift=rh=3:gv=1:bh=-3", |
| }, |
| "vhs": { |
| "label": "VHS", |
| "filter": "noise=alls=18:allf=t+u,eq=saturation=0.82:contrast=1.08", |
| }, |
| "crt": { |
| "label": "CRT", |
| "filter": "vignette=PI/4,noise=alls=10:allf=t+u,eq=contrast=1.15:saturation=0.9", |
| }, |
| "bloom": { |
| "label": "Bloom", |
| "filter": "gblur=sigma=0.35,eq=brightness=0.025:saturation=1.14", |
| }, |
| "glow": { |
| "label": "Glow", |
| "filter": "gblur=sigma=0.45,eq=contrast=1.05:brightness=0.03", |
| }, |
| "chromatic_aberration": { |
| "label": "Chromatic Aberration", |
| "filter": "rgbashift=rh=2:rv=1:bh=-2:bv=-1", |
| }, |
| "neon": { |
| "label": "Neon", |
| "filter": "eq=contrast=1.2:saturation=1.55:brightness=0.02", |
| }, |
| "cyberpunk": { |
| "label": "Cyberpunk", |
| "filter": "eq=contrast=1.18:saturation=1.45:gamma_r=1.08:gamma_b=1.18", |
| }, |
| "comic": { |
| "label": "Comic", |
| "filter": "edgedetect=low=0.08:high=0.25,eq=contrast=1.2:saturation=1.35", |
| }, |
| "cartoon": { |
| "label": "Cartoon", |
| "filter": "edgedetect=low=0.05:high=0.2,eq=saturation=1.45:contrast=1.15", |
| }, |
| "anime": { |
| "label": "Anime", |
| "filter": "eq=saturation=1.35:contrast=1.12:brightness=0.02,unsharp=5:5:0.6", |
| }, |
| "sketch": { |
| "label": "Sketch", |
| "filter": "edgedetect=low=0.03:high=0.18,hue=s=0", |
| }, |
| "oil_painting": { |
| "label": "Oil Painting", |
| "filter": "gblur=sigma=0.7,eq=saturation=1.25:contrast=1.1", |
| }, |
| "pixel_art": { |
| "label": "Pixel Art", |
| "filter": "scale=iw/8:ih/8,scale=iw*8:ih*8:flags=neighbor", |
| }, |
| } |
|
|
|
|
| CREATIVE_STYLES: dict[str, CreativeStyle] = { |
| "viral_shorts": CreativeStyle( |
| key="viral_shorts", |
| label="Viral Shorts", |
| description="Fast vertical pacing, punchy captions, bright contrast, and whip-style movement.", |
| platform="tiktok", |
| template="tiktok_zoom", |
| scene_duration=2.4, |
| transition_sequence=("whip", "flash", "zoom", "glitch"), |
| scene_effect_sequence=("sharp_pop", "flash_pop", "clean_beauty"), |
| render_defaults={ |
| "subtitle_format": "ass", |
| "auto_subtitles": True, |
| "audio_normalize": True, |
| "music_volume": 0.24, |
| "music_fade_in": 0.25, |
| "music_fade_out": 0.8, |
| "music_ducking": True, |
| "normalize": True, |
| }, |
| metadata={"energy": "high", "best_for": "hooks, offers, memes, short promos"}, |
| ), |
| "product_demo": CreativeStyle( |
| key="product_demo", |
| label="Product Demo", |
| description="Clean cuts, readable captions, and polished color for launches and tutorials.", |
| platform="instagram_reels", |
| template="modern_minimal", |
| scene_duration=3.2, |
| transition_sequence=("slide", "wipe_left", "push", "dissolve"), |
| scene_effect_sequence=("clean_beauty", "sharp_pop"), |
| render_defaults={ |
| "subtitle_format": "ass", |
| "auto_subtitles": True, |
| "audio_normalize": True, |
| "music_volume": 0.18, |
| "music_fade_in": 0.3, |
| "music_fade_out": 0.7, |
| "music_ducking": True, |
| "normalize": True, |
| }, |
| metadata={"energy": "medium", "best_for": "software demos, ecommerce, tutorials"}, |
| ), |
| "story_vlog": CreativeStyle( |
| key="story_vlog", |
| label="Story Vlog", |
| description="Warm color, softer movement, and natural pacing for personality-led edits.", |
| platform="instagram_reels", |
| template="tiktok_classic", |
| scene_duration=3.8, |
| transition_sequence=("fade", "smooth_right", "dissolve"), |
| scene_effect_sequence=("warm_glow", "dreamy", "clean_beauty"), |
| render_defaults={ |
| "subtitle_format": "ass", |
| "auto_subtitles": True, |
| "audio_normalize": True, |
| "music_volume": 0.22, |
| "music_fade_in": 0.5, |
| "music_fade_out": 1.0, |
| "music_ducking": True, |
| "normalize": True, |
| }, |
| metadata={"energy": "medium", "best_for": "founder updates, day-in-life edits, testimonials"}, |
| ), |
| "podcast_clip": CreativeStyle( |
| key="podcast_clip", |
| label="Podcast Clip", |
| description="Square-safe framing, calmer captions, and narration-first audio treatment.", |
| platform="instagram_feed_square", |
| template="podcast_style", |
| scene_duration=5.0, |
| transition_sequence=("fade", "dissolve"), |
| scene_effect_sequence=("clean_beauty", "sharp_pop"), |
| render_defaults={ |
| "subtitle_format": "ass", |
| "auto_subtitles": True, |
| "audio_normalize": True, |
| "music_volume": 0.12, |
| "music_fade_in": 0.6, |
| "music_fade_out": 1.2, |
| "music_ducking": True, |
| "normalize": True, |
| }, |
| metadata={"energy": "low", "best_for": "interviews, audiograms, education"}, |
| ), |
| "cinematic_story": CreativeStyle( |
| key="cinematic_story", |
| label="Cinematic Story", |
| description="Deeper contrast, film grain, and slower transitions for mini-documentary edits.", |
| platform="youtube_shorts", |
| template="modern_minimal", |
| scene_duration=4.2, |
| transition_sequence=("dissolve", "fadeblack", "smooth_left"), |
| scene_effect_sequence=("cinematic", "grain"), |
| render_defaults={ |
| "subtitle_format": "ass", |
| "auto_subtitles": True, |
| "audio_normalize": True, |
| "music_volume": 0.26, |
| "music_fade_in": 0.8, |
| "music_fade_out": 1.4, |
| "music_ducking": True, |
| "normalize": True, |
| }, |
| metadata={"energy": "low", "best_for": "brand films, travel, documentary shorts"}, |
| ), |
| "news_explainer": CreativeStyle( |
| key="news_explainer", |
| label="News Explainer", |
| description="Readable lower-third style captions with stable landscape or vertical exports.", |
| platform="youtube_1080p", |
| template="news_style", |
| scene_duration=4.0, |
| transition_sequence=("wipe_left", "slide", "fade"), |
| scene_effect_sequence=("sharp_pop", "clean_beauty"), |
| render_defaults={ |
| "subtitle_format": "ass", |
| "auto_subtitles": False, |
| "audio_normalize": True, |
| "music_volume": 0.1, |
| "music_fade_in": 0.5, |
| "music_fade_out": 1.0, |
| "music_ducking": True, |
| "normalize": True, |
| }, |
| metadata={"energy": "medium", "best_for": "explainers, news, thought leadership"}, |
| ), |
| } |
|
|
|
|
| def apply_creative_style(payload: dict[str, Any]) -> dict[str, Any]: |
| style_key = payload.get("creative_style") or payload.get("metadata", {}).get("creative_style") |
| if not style_key: |
| return payload |
|
|
| style = get_creative_style(str(style_key)) |
| output = deepcopy(payload) |
| output["creative_style"] = style.key |
| _set_default(output, "platform", style.platform) |
| _set_default(output, "template", style.template) |
| for key, value in style.render_defaults.items(): |
| _set_default(output, key, deepcopy(value)) |
|
|
| metadata = deepcopy(style.metadata) |
| metadata.update(output.get("metadata", {})) |
| metadata["creative_style"] = style.key |
| metadata["creative_style_label"] = style.label |
| output["metadata"] = metadata |
|
|
| scenes = output.get("scenes") |
| if isinstance(scenes, list): |
| output["scenes"] = [_style_scene(scene, style, index) for index, scene in enumerate(scenes)] |
| return output |
|
|
|
|
| def get_creative_style(key: str | None) -> CreativeStyle: |
| if not key: |
| return CREATIVE_STYLES["viral_shorts"] |
| return CREATIVE_STYLES.get(key, CREATIVE_STYLES["viral_shorts"]) |
|
|
|
|
| def list_creative_styles() -> list[str]: |
| return list(CREATIVE_STYLES.keys()) |
|
|
|
|
| def creative_style_metadata() -> dict[str, dict[str, Any]]: |
| return {key: style.metadata_payload() for key, style in CREATIVE_STYLES.items()} |
|
|
|
|
| def list_scene_effects() -> list[str]: |
| return list(SCENE_EFFECTS.keys()) |
|
|
|
|
| def scene_effect_metadata() -> dict[str, dict[str, str]]: |
| return {key: {"label": value["label"]} for key, value in SCENE_EFFECTS.items()} |
|
|
|
|
| def scene_effect_filter(key: str | None) -> str: |
| if not key: |
| return "" |
| return SCENE_EFFECTS.get(key, SCENE_EFFECTS["none"])["filter"] |
|
|
|
|
| def _set_default(payload: dict[str, Any], key: str, value: Any) -> None: |
| if key not in payload or payload[key] in (None, ""): |
| payload[key] = value |
|
|
|
|
| def _style_scene(scene: Any, style: CreativeStyle, index: int) -> Any: |
| if not isinstance(scene, dict): |
| return scene |
| styled = deepcopy(scene) |
| transition = styled.get("transition") |
| if transition in (None, "", "fade"): |
| styled["transition"] = style.transition_sequence[index % len(style.transition_sequence)] |
| if styled.get("effect") in (None, ""): |
| styled["effect"] = style.scene_effect_sequence[index % len(style.scene_effect_sequence)] |
| _set_default(styled, "background", "blur") |
| _set_default(styled, "layout", "fill") |
| return styled |
|
|
