import gradio as gr
‎import os
‎
‎
‎def generate_lipsync(image, audio):
‎    os.makedirs("input", exist_ok=True)
‎    os.makedirs("output", exist_ok=True)
‎
‎    image_path = "input/image.jpg"
‎    audio_path = "input/audio.wav"
‎    video_path = "input/video.mp4"
‎    result_path = "output/result_voice.mp4"
‎
‎    image.save(image_path)
‎
‎    os.system(f"cp '{audio}' {audio_path}")
‎
‎    os.system(
‎        f"ffmpeg -loop 1 -i {image_path} -c:v libx264 -t 5 -pix_fmt yuv420p -vf scale=512:512 {video_path} -y"
‎    )
‎
‎    command = f'''
‎    python wav2lip/inference.py \
‎    --checkpoint_path checkpoints/wav2lip_gan.pth \
‎    --face {video_path} \
‎    --audio {audio_path} \
‎    --outfile {result_path}
‎    '''
‎
‎    os.system(command)
‎
‎    return result_path
‎
‎
‎iface = gr.Interface(
‎    fn=generate_lipsync,
‎    inputs=[
‎        gr.Image(type="pil", label="Upload Image"),
‎        gr.Audio(type="filepath", label="Upload Audio")
‎    ],
‎    outputs=gr.Video(label="Generated Video"),
‎    title="TikTok AI LipSync"
‎)
‎
‎iface.launch(server_name="0.0.0.0", server_port=7860)