Update app.py

app.py

@@ -15,7 +15,7 @@ STUMPS_WIDTH = 0.2286  # meters (width of stumps)
 BALL_DIAMETER = 0.073  # meters (approx. cricket ball diameter)
 FRAME_RATE = 30  # Input video frame rate
 SLOW_MOTION_FACTOR = 6  # For very slow motion (6x slower)
-CONF_THRESHOLD = 0.3  # Confidence threshold for detection
+CONF_THRESHOLD = 0.25  # Lowered confidence threshold to improve detection
 IMPACT_ZONE_Y = 0.85  # Fraction of frame height where impact is likely (near stumps)
 
 def process_video(video_path):
@@ -50,13 +50,33 @@ def process_video(video_path):
     else:
         debug_log.append(f"Total ball detections: {len(ball_positions)}")
 
+    # Interpolate missing detections
+    if ball_positions:
+        ball_positions = interpolate_missing_positions(ball_positions, frame_count)
+
     return frames, ball_positions, "\n".join(debug_log)
 
+def interpolate_missing_positions(ball_positions, total_frames):
+    if len(ball_positions) < 2:
+        return ball_positions
+    times = np.linspace(0, total_frames / FRAME_RATE, total_frames)
+    detected_times = [i / FRAME_RATE for i, _ in enumerate(ball_positions)]
+    x_coords = [pos[0] for pos in ball_positions]
+    y_coords = [pos[1] for pos in ball_positions]
+
+    try:
+        fx = interp1d(detected_times, x_coords, kind='linear', fill_value="extrapolate")
+        fy = interp1d(detected_times, y_coords, kind='linear', fill_value="extrapolate")
+        interpolated_positions = [(fx(t), fy(t)) for t in times if t <= detected_times[-1]]
+        return interpolated_positions
+    except:
+        return ball_positions
+
 def estimate_trajectory(ball_positions, frames):
     if len(ball_positions) < 2:
         return None, None, None, "Error: Fewer than 2 ball detections for trajectory"
     frame_height = frames[0].shape[0]
-    
+
     # Extract x, y coordinates
     x_coords = [pos[0] for pos in ball_positions]
     y_coords = [pos[1] for pos in ball_positions]
@@ -71,8 +91,9 @@ def estimate_trajectory(ball_positions, frames):
     if impact_idx is None:
         impact_idx = len(ball_positions) - 1  # Fallback to last detection
 
+    pitch_point = ball_positions[0]
     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]
@@ -88,7 +109,36 @@ def estimate_trajectory(ball_positions, frames):
     t_full = np.linspace(times[0], times[-1] + 0.5, len(times) + 10)
     x_full = fx(t_full)
     y_full = fy(t_full)
-    trajectory = listagus = f"Out (Ball hits stumps, Pitch at x: {pitch_x:.1f}, y: {pitch_y:.1f}, Impact at x: {impact_x:.1f}, y: {impact_y:.1f})", trajectory, pitch_point, impact_point
+    trajectory = list(zip(x_full, y_full))
+
+    return trajectory, pitch_point, impact_point, "Trajectory estimated successfully"
+
+def lbw_decision(ball_positions, trajectory, frames, pitch_point, impact_point):
+    if not frames:
+        return "Error: No frames processed", None, None, None
+    if not trajectory or len(ball_positions) < 2:
+        return "Not enough data (insufficient ball detections)", None, None, None
+
+    frame_height, frame_width = frames[0].shape[:2]
+    stumps_x = frame_width / 2
+    stumps_y = frame_height * 0.9
+    stumps_width_pixels = frame_width * (STUMPS_WIDTH / 3.0)
+
+    pitch_x, pitch_y = pitch_point
+    impact_x, impact_y = impact_point
+
+    # Check pitching point
+    if pitch_x < stumps_x - stumps_width_pixels / 2 or pitch_x > stumps_x + stumps_width_pixels / 2:
+        return f"Not Out (Pitched outside line at x: {pitch_x:.1f}, y: {pitch_y:.1f})", trajectory, pitch_point, impact_point
+
+    # Check impact point
+    if impact_x < stumps_x - stumps_width_pixels / 2 or impact_x > stumps_x + stumps_width_pixels / 2:
+        return f"Not Out (Impact outside line at x: {impact_x:.1f}, y: {impact_y:.1f})", trajectory, pitch_point, impact_point
+
+    # Check trajectory hitting stumps
+    for x, y in trajectory:
+        if abs(x - stumps_x) < stumps_width_pixels / 2 and abs(y - stumps_y) < frame_height * 0.1:
+            return f"Out (Ball hits stumps, Pitch at x: {pitch_x:.1f}, y: {pitch_y:.1f}, Impact at x: {impact_x:.1f}, y: {impact_y:.1f})", trajectory, pitch_point, impact_point
     return f"Not Out (Missing stumps, Pitch at x: {pitch_x:.1f}, y: {pitch_y:.1f}, Impact at x: {impact_x:.1f}, y: {impact_y:.1f})", trajectory, pitch_point, impact_point
 
 def generate_slow_motion(frames, trajectory, pitch_point, impact_point, output_path):