Update app.py

app.py

@@ -65,29 +65,7 @@ def estimate_trajectory(ball_positions, frames):
     y_coords = [pos[1] for pos in ball_positions]
     times = np.arange(len(ball_positions)) / FRAME_RATE
 
-    # Detect the pitch point: find when the ball touches the ground
-    pitch_point = None
-    for i, y in enumerate(y_coords):
-        if y > frame_height * 0.75:  # Threshold for ground contact (near the bottom of the frame)
-            pitch_point = ball_positions[i]
-            break
-    
-    # Find impact point (closest to batsman, near stumps)
-    impact_idx = None
-    for i, y in enumerate(y_coords):
-        if y > frame_height * IMPACT_ZONE_Y:  # Ball is near stumps/batsman
-            impact_idx = i
-            break
-    if impact_idx is None:
-        impact_idx = len(ball_positions) - 1  # Fallback to last detection
-    
-    impact_point = ball_positions[impact_idx]
-
     # Use positions up to impact for interpolation
-    x_coords = x_coords[:impact_idx + 1]
-    y_coords = y_coords[:impact_idx + 1]
-    times = times[:impact_idx + 1]
-
     try:
         fx = interp1d(times, x_coords, kind='linear', fill_value="extrapolate")
         fy = interp1d(times, y_coords, kind='quadratic', fill_value="extrapolate")
@@ -100,7 +78,29 @@ def estimate_trajectory(ball_positions, frames):
     y_full = fy(t_full)
     trajectory = list(zip(x_full, y_full))
 
-    return trajectory, pitch_point, impact_point, "Trajectory estimated successfully"
+    return trajectory, None, None, "Trajectory estimated successfully"
+
+def detect_pitch_and_impact(ball_positions, frames, frame_height):
+    pitch_point = None
+    impact_point = None
+    
+    # Pitch detection: Ball hits the ground (vertical velocity and low height)
+    pitch_threshold_y = frame_height * 0.75  # Ball reaches near the ground
+
+    # For the impact point, we assume that the region of interest is near the stumps
+    impact_zone_y_min = frame_height * 0.80
+    impact_zone_y_max = frame_height * 0.85
+
+    # Detect pitch and impact points
+    for i, (x, y) in enumerate(ball_positions):
+        if y > pitch_threshold_y and not pitch_point:
+            pitch_point = (x, y)  # Ball has hit the ground
+
+        # Check if the ball is near the batsman's impact zone (e.g., near stumps)
+        if impact_zone_y_min <= y <= impact_zone_y_max and not impact_point:
+            impact_point = (x, y)  # Ball has impacted the batsman (bat or pad)
+
+    return pitch_point, impact_point
 
 def lbw_decision(ball_positions, trajectory, frames, pitch_point, impact_point):
     if not frames:
@@ -186,6 +186,10 @@ def drs_review(video):
     if not frames:
         return f"Error: Failed to process video", None
     trajectory, pitch_point, impact_point, trajectory_log = estimate_trajectory(ball_positions, frames)
+    
+    # Detect pitch and impact points based on ball positions
+    pitch_point, impact_point = detect_pitch_and_impact(ball_positions, frames, frames[0].shape[0])
+
     decision, trajectory, pitch_point, impact_point = lbw_decision(ball_positions, trajectory, frames, pitch_point, impact_point)
 
     output_path = f"output_{uuid.uuid4()}.mp4"