4. Video Converters

4. Video Converters#

Video converters enable transformations involving video files, particularly adding images to videos.

Overview#

This notebook covers:

Image to Video#

AddImageVideoConverter#

The AddImageVideoConverter adds an image overlay to a video file. To use this converter you’ll need to install opencv which can be done with pip install pyrit[opencv]

import pathlib

from pyrit.prompt_converter import AddImageVideoConverter
from pyrit.setup import IN_MEMORY, initialize_pyrit_async

await initialize_pyrit_async(memory_db_type=IN_MEMORY)  # type: ignore

input_video = str(pathlib.Path(".") / ".." / ".." / ".." / "assets" / "sample_video.mp4")
input_image = str(pathlib.Path(".") / ".." / ".." / ".." / "assets" / "pyrit_architecture.png")

video = AddImageVideoConverter(video_path=input_video)
converted_vid = await video.convert_async(prompt=input_image, input_type="image_path")  # type: ignore
converted_vid

Found default environment files: ['/home/vscode/.pyrit/.env', '/home/vscode/.pyrit/.env.local']
Loaded environment file: /home/vscode/.pyrit/.env
Loaded environment file: /home/vscode/.pyrit/.env.local

ConverterResult(output_text='/workspace/dbdata/prompt-memory-entries/videos/1767054729513019.mp4', output_type='video_path')
Contents