output file nav

main.py

@@ -27,6 +27,7 @@ print("[INFO]: Downloaded models!")
 
 def poses(photo):
     print("[INFO]: Running inference!")
+    #vis_out_dir
     result_generator = inferencer(photo, 
                                   vis_out_dir =".",
                                   return_vis=True,
@@ -59,32 +60,10 @@ def poses(photo):
 
     # out_writer.release()
     # cv2.destroyAllWindows() # Closing window
-    output_file = glob.glob("*.mp4")
-    print(glob.glob("*.mp4"))
-    print(glob.glob("*.avi"))
-
-    cap = cv2.VideoCapture(os.path.join(output_file[0]))
-    formatfile = "test.mp4"
-    fourcc = cv2.VideoWriter_fourcc(*"mp4v")  # Codec for MP4 video
-    fps = 32
-    height = 480
-    width = 640
-    size = (width,height)
-
-    out_writer = cv2.VideoWriter(formatfile, fourcc, fps, size)
-    
-    while cap.isOpened():
-        ret, frame = cap.read()
-        out_writer.write(frame)
-
-
-    # Release everything if job is finished
-    cap.release()
-    out_writer.release()
-    cv2.destroyAllWindows()
-
+    output_files = glob.glob(os.path.join(".","*.mp4"))
+    print(output_files)
 
-    return formatfile #output_file[0]
+    return output_files[0]