"""Visual effects modules for ComfyUI-TRELLIS2_Motion."""

from .onion_skin import apply_onion_skinning, OnionSkinConfig
from .wireframe import render_wireframe, wireframe_overlay
from .cel_shading import apply_cel_shading, CelShadingConfig
from .lighting import (
    apply_lighting_preset,
    create_studio_lighting,
    create_outdoor_lighting,
    create_dramatic_lighting,
    create_flat_lighting,
)
from .post_processing import (
    apply_bloom,
    apply_depth_of_field,
    apply_motion_blur,
    apply_outline,
)

__all__ = [
    # Onion skinning
    "apply_onion_skinning",
    "OnionSkinConfig",
    # Wireframe
    "render_wireframe",
    "wireframe_overlay",
    # Cel shading
    "apply_cel_shading",
    "CelShadingConfig",
    # Lighting
    "apply_lighting_preset",
    "create_studio_lighting",
    "create_outdoor_lighting",
    "create_dramatic_lighting",
    "create_flat_lighting",
    # Post-processing
    "apply_bloom",
    "apply_depth_of_field",
    "apply_motion_blur",
    "apply_outline",
]