Update video_processor.py

video_processor.py

@@ -5,6 +5,7 @@ from detector import LBWDetector
 from utils import draw_boxes, overlay_decision_text
 
 def process_video(video_path, output_path="output.mp4"):
+    """Process a video, detect objects, make LBW decisions, and return output path and decision."""
     detector = LBWDetector()
     cap = cv2.VideoCapture(video_path)
 
@@ -14,17 +15,42 @@ def process_video(video_path, output_path="output.mp4"):
 
     out = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height))
 
+    # Assume stumps are in the center third of the frame (adjust as needed)
+    stump_zone = (width // 3, 2 * width // 3)
+    final_decision = "Not Out"  # Default decision
+
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
             break
+
+        # Detect objects (ball, player/pads)
         detections, class_names = detector.detect_objects(frame)
         frame = draw_boxes(frame, detections, class_names)
-        # Placeholder decision; replace with actual LBW logic later
-        decision = "Pending"  # Example: "Out" or "Not Out"
+
+        # Simple LBW decision logic
+        decision = "Not Out"  # Frame-level decision
+        ball_bbox = None
+        player_bbox = None
+
+        for x1, y1, x2, y2, conf, cls_id in detections:
+            label = class_names[int(cls_id)]
+            if label == "ball":
+                ball_bbox = (x1, y1, x2, y2)
+            elif label in ["player", "pads"]:  # Adjust based on your model's class names
+                player_bbox = (x1, y1, x2, y2)
+
+        if ball_bbox and player_bbox:
+            ball_x_center = (ball_bbox[0] + ball_bbox[2]) / 2
+            # Check if ball hits player in stump zone
+            if (stump_zone[0] <= ball_x_center <= stump_zone[1] and
+                ball_bbox[1] < player_bbox[3] and ball_bbox[3] > player_bbox[1]):  # Vertical overlap
+                decision = "Out"
+                final_decision = "Out"  # Update final decision if "Out" is detected
+
         frame = overlay_decision_text(frame, decision)
         out.write(frame)
 
     cap.release()
     out.release()
-    return output_path
+    return output_path, final_decision  # Return both output path and final decision