make directory

main.py

@@ -82,12 +82,11 @@ def show_tracking(video_content):
 
 
 def pose3d(video):
-    if video[:-5] == ".webm":
-        print("Format is webm, changing it to mp4...")
-        clip = moviepy.VideoFileClip(video)
+    clip = moviepy.VideoFileClip(video)
+
+    video = video[:-5]+".mp4"
+    clip.write_videofile(video)
 
-        new_video = video[:-5]+".mp4"
-        clip.write_videofile(new_video)
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
@@ -110,19 +109,17 @@ def pose3d(video):
 
 def pose2d(video):
 
-    if video[:-5] == ".webm":
-        print("Format is webm, changing it to mp4...")
-        clip = moviepy.VideoFileClip(video)
+    clip = moviepy.VideoFileClip(video)
 
-        new_video = video[:-5]+".mp4"
-        clip.write_videofile(new_video)
+    video = video[:-5]+".mp4"
+    clip.write_videofile(video)
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())
     vis_out_dir = os.path.join("/".join(video.split("/")[:-1]), add_dir)
     os.makedirs(vis_out_dir)
 
-    result_generator = human(new_video, 
+    result_generator = human(video, 
                             vis_out_dir = vis_out_dir,
                             return_vis=True,
                             thickness=2,
@@ -139,12 +136,11 @@ def pose2d(video):
 
 
 def pose2dhand(video):
-    if video[:-5] == ".webm":
-        print("Format is webm, changing it to mp4...")
-        clip = moviepy.VideoFileClip(video)
+    clip = moviepy.VideoFileClip(video)
+
+    video = video[:-5]+".mp4"
+    clip.write_videofile(video)
 
-        new_video = video[:-5]+".mp4"
-        clip.write_videofile(new_video)
 
     # Define new unique folder
     add_dir = str(uuid.uuid4())