import gradio as gr
from videopose_PSTMO import gr_video2mc
import os


def Video2MC(video, progress=gr.Progress(track_tqdm=True)):

    progress(1.0, desc="Step 0: Starting")
    output_path, output_video = gr_video2mc(video, progress)

    return output_path, output_path, output_video

    
iface = gr.Interface(fn=Video2MC, 
                     inputs=gr.Video(), 
                     outputs=["file", "text", gr.Video()],
                     examples=[os.path.join(os.path.dirname(__file__), 
                                     "input_videos/kun_test_5sec.mp4")], 
                     cache_examples=False,
                     )

iface.queue(concurrency_count=10).launch()