import os, sys
import tempfile
import gradio as gr
from gradio_demo import SadTalker

def get_source_image(image):
        return image


if __name__ == "__main__":
    src_image_path = './examples/source_image/ABOUT_test_00001.jpg'
    driving_audio_path = './examples/driven_audio/RD_Radio31_000.wav'


    sadtalker = SadTalker(lazy_load=True)
    return_path = sadtalker.test(source_image=src_image_path, driven_audio=driving_audio_path)
    print(return_path)