7. Video Converters

7. Video Converters#

Adding Images to a Video#

Adds an image to a video. 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

ConverterResult(output_text='C:\\git\\PyRIT\\dbdata\\prompt-memory-entries\\videos\\1764988282052021.mp4', output_type='video_path')
Contents