import os
from auto_media_publisher.generators.video_generator_taijiniao import VideoGeneratorTaijiniao
from auto_media_publisher.generators.video_generator_mashup import VideoGeneratorMashup
import time
import sys


if __name__ == '__main__':

    if len(sys.argv) >2:
        type = sys.argv[1]
        tmp_time = sys.argv[2]
    else:
        type = "taiji"
        tmp_time = "2025-07-24 18:00:00"


        

    if type == 'taiji':
        required_keys = {"user_id":"test","scheduled_time":tmp_time,"proxy_full":"None"}

        processor = VideoGeneratorTaijiniao(**required_keys)
    else:
        required_keys = {"user_id":"test","scheduled_time":tmp_time,"proxy_full":"None","video_type":"05-ocean-sunset","video_width":576,"video_height":1024,"font_size":30}
        processor = VideoGeneratorMashup(**required_keys)

    start_time = time.time()

    processor.generate()

    #worker.swap_face('./temp/lilizhen.jpg', './temp/g1.mp4', './temp/g1_no_enhancer.mp4',1)

    print(time.time()-start_time)