app v0.0 add video interface

app.py

@@ -1,7 +1,12 @@
 import gradio as gr
+from videopose_PSTMO import gr_video2mc
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def Video2MC(video):
+    output_path = gr_video2mc(video)
+    return output_path, output_path
+
+iface = gr.Interface(fn=Video2MC, 
+                     inputs=gr.Video(), 
+                     outputs=["file", "text"])
 iface.launch()

videopose_PSTMO.py

@@ -200,6 +200,14 @@ def inference_video(video_path, detector_2d):
     return output_dir_dict
 
 
+def gr_video2mc(video_path):
+    files = os.listdir('./input_videos')
+    FPS_mine_imator = 30
+    output_dir_dict = inference_video(video_path, 'alpha_pose')
+    Hk.hpe2keyframes(output_dir_dict['npy'], FPS_mine_imator, f"output_miframes/{output_dir_dict['video_name']}.miframes")
+    return os.path.abspath(f"output_miframes/{output_dir_dict['video_name']}.miframes")
+
+
 if __name__ == '__main__':
 
     files = os.listdir('./input_videos')