Update utils/video_processing.py

utils/video_processing.py

@@ -4,14 +4,30 @@ import numpy as np
 from ultralytics import YOLO
 import os
 
-# Load YOLO model (assumed to be trained for cricket objects)
-model = YOLO('models/yolov8_model.pt')
+# Path to the YOLO model
+MODEL_PATH = 'models/yolov8_model.pt'
+
+# Check if model file exists
+if not os.path.exists(MODEL_PATH):
+    raise FileNotFoundError(f"YOLO model file not found at {MODEL_PATH}. Please ensure 'yolov8_model.pt' is in the 'models/' directory.")
+
+# Load YOLO model
+try:
+    model = YOLO(MODEL_PATH)
+except Exception as e:
+    raise RuntimeError(f"Failed to load YOLO model from {MODEL_PATH}: {str(e)}")
 
 def track_ball(video_path: str) -> list:
     """
     Track the ball in the video and return its trajectory as a list of (x, y) coordinates.
     """
+    if not os.path.exists(video_path):
+        raise FileNotFoundError(f"Video file not found at {video_path}")
+
     cap = cv2.VideoCapture(video_path)
+    if not cap.isOpened():
+        raise ValueError(f"Failed to open video file: {video_path}")
+
     tracker = cv2.TrackerKCF_create()
     trajectory = []
     init = False
@@ -45,7 +61,13 @@ def generate_replay(video_path: str, trajectory: list, decision: str) -> str:
     """
     Generate a slow-motion replay video with ball trajectory and decision overlay.
     """
+    if not os.path.exists(video_path):
+        raise FileNotFoundError(f"Video file not found at {video_path}")
+
     cap = cv2.VideoCapture(video_path)
+    if not cap.isOpened():
+        raise ValueError(f"Failed to open video file: {video_path}")
+
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     fps = cap.get(cv2.CAP_PROP_FPS) / 2  # Slow motion (half speed)