from auto_media_publisher.modifiers.video_matting.robust_video_matting_converter import RobustVideoMattingConverter
import time
if __name__ == '__main__':
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument('--input', type=str, required=True)
    parser.add_argument('--bg_video', type=str)
    parser.add_argument('--bg_image', type=str)
    parser.add_argument('--output_video', type=str)
    parser.add_argument('--output_alpha', type=str)
    parser.add_argument('--output_fore', type=str)
       
    args = parser.parse_args()

    time_start = time.time()

    converter = RobustVideoMattingConverter()

    print(f"model loaded .. {time.time()-time_start}")
    time_start = time.time()

    converter.video_matting(
        input=args.input,
        bg_video=args.bg_video,
        bg_image=args.bg_image,
        output_video=args.output_video,
        output_alpha=args.output_alpha,
        output_fore=args.output_fore,
        )
    print(f"video matting finished .. {time.time()-time_start}")
    # conda activate mediapublish
    # f:
    # cd automedia
    #python -m test.test_video_matting --input ./temp/buhui.mp4 --bg_video ./temp/ocean.mp4 --output_video ./temp/buhui_ocean-36.mp4