Update app.py

app.py

@@ -19,6 +19,7 @@ CONF_THRESHOLD = 0.25  # Confidence threshold for detection
 IMPACT_ZONE_Y = 0.85  # Fraction of frame height for impact zone
 PITCH_ZONE_Y = 0.75  # Fraction of frame height for pitch zone
 IMPACT_DELTA_Y = 50  # Pixels for detecting sudden y-position change
+STUMPS_HEIGHT = 0.711  # meters (height of stumps)
 
 def process_video(video_path):
     if not os.path.exists(video_path):
@@ -66,19 +67,12 @@ def estimate_trajectory(ball_positions, detection_frames, frames):
 
     # Pitch point: first valid detection or when y exceeds PITCH_ZONE_Y
     pitch_idx = 0
-    pitch_point = None
-    pitch_frame = None
     for i, y in enumerate(y_coords):
         if y > frame_height * PITCH_ZONE_Y:
             pitch_idx = i
-            pitch_point = ball_positions[pitch_idx]
-            pitch_frame = detection_frames[pitch_idx]
             break
-    if pitch_point is None:  # Handle full toss (no pitch point)
-        pitch_idx = 0
-        pitch_point = ball_positions[0]
-        pitch_frame = detection_frames[0]
-        debug_log = "Warning: No pitch point detected (possible full toss)"
+    pitch_point = ball_positions[pitch_idx]
+    pitch_frame = detection_frames[pitch_idx]
 
     # Impact point: sudden y-change or y exceeds IMPACT_ZONE_Y
     impact_idx = None
@@ -92,14 +86,7 @@ def estimate_trajectory(ball_positions, detection_frames, frames):
     impact_point = ball_positions[impact_idx]
     impact_frame = detection_frames[impact_idx]
 
-    # Ensure pitch point appears before impact point (unless full toss)
-    if pitch_frame >= impact_frame and pitch_idx != 0:
-        debug_log = f"Error: Pitch frame {pitch_frame + 1} after impact frame {impact_frame + 1}, adjusting pitch to first detection"
-        pitch_idx = 0
-        pitch_point = ball_positions[0]
-        pitch_frame = detection_frames[0]
-
-    # Use positions up to impact for trajectory
+    # Use only detected positions for trajectory
     x_coords = x_coords[:impact_idx + 1]
     y_coords = y_coords[:impact_idx + 1]
     times = times[:impact_idx + 1]
@@ -156,6 +143,10 @@ def generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impac
     if not frames:
         return 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)
+    stumps_height_pixels = frame_height * (STUMPS_HEIGHT / 3.0)
 
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_path, fourcc, FRAME_RATE / SLOW_MOTION_FACTOR, (frame_width, frame_height))
@@ -164,6 +155,15 @@ def generate_slow_motion(frames, vis_trajectory, pitch_point, pitch_frame, impac
     trajectory_points = np.array(vis_trajectory, dtype=np.int32).reshape((-1, 1, 2))
 
     for i, frame in enumerate(frames):
+        # Draw stumps (three white vertical lines)
+        stump_positions = [
+            (stumps_x - stumps_width_pixels / 2, stumps_y),  # Left stump
+            (stumps_x, stumps_y),                           # Middle stump
+            (stumps_x + stumps_width_pixels / 2, stumps_y)   # Right stump
+        ]
+        for x, y in stump_positions:
+            cv2.line(frame, (int(x), int(y)), (int(x), int(y - stumps_height_pixels)), (255, 255, 255), 2)
+
         # Draw trajectory (blue line) only for detected frames
         if i in detection_frames and trajectory_points.size > 0:
             idx = detection_frames.index(i) + 1
@@ -208,10 +208,10 @@ iface = gr.Interface(
     inputs=gr.Video(label="Upload Video Clip"),
     outputs=[
         gr.Textbox(label="DRS Decision and Debug Log"),
-        gr.Video(label="Very Slow-Motion Replay with Ball Detection (Green), Trajectory (Blue Line), Pitch Point (Red), Impact Point (Yellow)")
+        gr.Video(label="Very Slow-Motion Replay with Ball Detection (Green), Trajectory (Blue Line), Pitch Point (Red), Impact Point (Yellow), Stumps (White)")
     ],
     title="AI-Powered DRS for LBW in Local Cricket",
-    description="Upload a video clip of a cricket delivery to get an LBW decision and slow-motion replay showing ball detection (green boxes), trajectory (blue line), pitch point (red circle), and impact point (yellow circle)."
+    description="Upload a video clip of a cricket delivery to get an LBW decision and slow-motion replay showing ball detection (green boxes), trajectory (blue line), pitch point (red circle), impact point (yellow circle), and stumps (white lines)."
 )
 
 if __name__ == "__main__":