# SPDX-License-Identifier: Apache-2.0
# Copyright 2026 Mellis, Inc.
"""Generate the record-shop dance from two images and a motion video with audio."""
from _common import arguments, asset, generate


def main():
    args = arguments("ref2va")
    from diffusers.modular_pipelines.pika_h3_flex import PikaH3FlexImageReference, PikaH3FlexVideoReference
    character = asset(args, "ref2va", "character.jpg")
    environment = asset(args, "ref2va", "environment.jpg")
    motion = asset(args, "ref2va", "motion.mp4")
    references = [PikaH3FlexImageReference.from_file(str(character)),
                  PikaH3FlexImageReference.from_file(str(environment)),
                  PikaH3FlexVideoReference.from_file(str(motion))]
    generate(args, "ref2va", {"references": references}, [character, environment, motion])


if __name__ == "__main__":
    main()