Add text overlays to videos

%pip install -qU pixeltable

import pixeltable as pxt

# Create a fresh directory
pxt.drop_dir('overlay_demo', force=True)
pxt.create_dir('overlay_demo')

# Create a video table
videos = pxt.create_table(
    'overlay_demo/videos', {'video': pxt.Video, 'title': pxt.String}
)

# Insert a sample video
videos.insert(
    [
        {
            'video': 's3://multimedia-commons/data/videos/mp4/ffe/ffb/ffeffbef41bbc269810b2a1a888de.mp4',
            'title': 'Sample Video',
        }
    ]
)

# Add a simple watermark in the corner
videos.add_computed_column(
    watermarked=videos.video.overlay_text(
        'My Brand',
        font_size=24,
        color='white',
        opacity=0.7,
        horizontal_align='right',
        horizontal_margin=20,
        vertical_align='top',
        vertical_margin=20,
    )
)

# Add a caption with a semi-transparent background box
videos.add_computed_column(
    captioned=videos.video.overlay_text(
        'This is a sample caption',
        font_size=32,
        color='white',
        box=True,  # Add background box
        box_color='black',
        box_opacity=0.8,
        box_border=[6, 14],  # Padding: [top/bottom, left/right]
        horizontal_align='center',
        vertical_align='bottom',
        vertical_margin=70,  # Distance from bottom
    )
)

# Add video title as an overlay (dynamic per video)
videos.add_computed_column(
    titled=videos.video.overlay_text(
        videos.title,  # Use the title column!
        font_size=48,
        color='yellow',
        opacity=1.0,
        horizontal_align='center',
        vertical_align='top',
        vertical_margin=30,
    )
)

# View all versions
videos.select(
    videos.title,
    videos.video,
    videos.watermarked,
    videos.captioned,
    videos.titled,
).collect()

# Crop using xywh format (default): [x, y, width, height]
videos.add_computed_column(cropped=videos.video.crop([100, 50, 320, 240]))

# Crop using xyxy format (common in object detection pipelines):
# videos.add_computed_column(
#     cropped_xyxy=videos.video.crop([100, 50, 420, 290], bbox_format='xyxy')
# )