duration

app.py

@@ -30,6 +30,30 @@ print("[INFO]: Imported modules!")
 track_model = YOLO('yolov8n.pt')  # Load an official Detect model
 print("[INFO]: Downloaded models!")
 
+
+
+def check_extension(video):
+    
+    clip = moviepy.VideoFileClip(video)
+
+    if clip.duration > 10:
+            raise gr.Error("Please provide or record a video shorter than 10 seconds...")
+
+    split_tup = os.path.splitext(video)
+
+    # extract the file name and extension
+    file_name = split_tup[0]
+    file_extension = split_tup[1]
+
+    if file_extension != ".mp4":
+        print("Converting to mp4")
+
+        video = file_name+".mp4"
+        clip.write_videofile(video, threads = 8)
+    
+    return video
+
+
 def tracking(video, model, boxes=True):
     print("[INFO] Is cuda available? ", torch.cuda.is_available())
     print(device)
@@ -45,17 +69,12 @@ def tracking(video, model, boxes=True):
     return annotated_frame
 
 def show_tracking(video_content):
+        
+        video = check_extension(video_content)
 
         # https://docs.ultralytics.com/datasets/detect/coco/
         video = cv2.VideoCapture(video_content)
 
-        fps = video.get(cv2.CAP_PROP_FPS)      # OpenCV v2.x used "CV_CAP_PROP_FPS"
-        frame_count = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
-        duration = frame_count/fps
-
-        if duration > 10:
-            raise gr.Error("Please provide or record a video shorter than 10 seconds...")
-
         # Track
         video_track = tracking(video_content, track_model.track)