Update app.py

app.py

@@ -19,7 +19,6 @@ 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):
@@ -143,10 +142,6 @@ 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))
@@ -155,15 +150,6 @@ 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 +194,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), Stumps (White)")
+        gr.Video(label="Very Slow-Motion Replay with Ball Detection (Green), Trajectory (Blue Line), Pitch Point (Red), Impact Point (Yellow)")
     ],
     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), impact point (yellow circle), and stumps (white lines)."
+    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)."
 )
 
 if __name__ == "__main__":