Update video_processor.py

video_processor.py

@@ -15,21 +15,18 @@ 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
+    final_decision = "Not Out"
 
     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)
 
-        # Simple LBW decision logic
-        decision = "Not Out"  # Frame-level decision
+        decision = "Not Out"
         ball_bbox = None
         player_bbox = None
 
@@ -37,20 +34,19 @@ def process_video(video_path, output_path="output.mp4"):
             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
+            elif label in ["player", "pads"]:
                 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
+                ball_bbox[1] < player_bbox[3] and ball_bbox[3] > player_bbox[1]):
                 decision = "Out"
-                final_decision = "Out"  # Update final decision if "Out" is detected
+                final_decision = "Out"
 
         frame = overlay_decision_text(frame, decision)
         out.write(frame)
 
     cap.release()
     out.release()
-    return output_path, final_decision  # Return both output path and final decision
+    return output_path, final_decision