Update app.py

app.py

@@ -28,6 +28,12 @@ def process_video(video_path):
     frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     fps = int(cap.get(cv2.CAP_PROP_FPS))
+    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    if frame_count < 10:  # Ensure video has enough frames
+        logger.error("Video too short, requires at least 10 frames")
+        cap.release()
+        return None, "Error: Video too short", None
+
     output_path = "replay.mp4"
     output_video = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (frame_width, frame_height))
 
@@ -61,7 +67,7 @@ def process_video(video_path):
         ball_detected = False
         for result in results:
             for box in result.boxes:
-                if result.names[int(box.cls)] == 'cricket_ball' and box.conf > 0.7:
+                if result.names[int(box.cls)] == 'cricket_ball' and box.conf > 0.5:  # Lowered threshold
                     x, y, w, h = box.xywh[0]
                     if last_valid_pos is None or (
                         abs(x - last_valid_pos[0]) < 100 and abs(y - last_valid_pos[1]) < 100
@@ -73,6 +79,8 @@ def process_video(video_path):
                         ball_detected = True
                         logger.info(f"Frame {frame_idx}: Detected ball at ({x:.2f}, {y:.2f}), conf={box.conf:.2f}")
                     break
+            if ball_detected:
+                break
 
         if not ball_detected:
             prediction = kalman.predict()
@@ -84,23 +92,26 @@ def process_video(video_path):
         # Pose detection
         pose_results = pose.process(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
         if pose_results.pose_landmarks:
-            if frame_idx < 10 and release_frame is None:
-                hand = pose_results.pose_landmarks.landmark[mp_pose.PoseLandmark.RIGHT_HAND]
-                if hand.visibility > 0.7:
-                    release_x, release_y = hand.x * frame_width, hand.y * frame_height
-                    release_frame = frame_idx
-                    logger.info(f"Release point detected at frame {frame_idx}: ({release_x:.2f}, {release_y:.2f})")
+            try:
+                if frame_idx < 10 and release_frame is None:
+                    hand = pose_results.pose_landmarks.landmark[mp_pose.PoseLandmark.RIGHT_WRIST]
+                    if hand.visibility > 0.7:
+                        release_x, release_y = hand.x * frame_width, hand.y * frame_height
+                        release_frame = frame_idx
+                        logger.info(f"Release point detected at frame {frame_idx}: ({release_x:.2f}, {release_y:.2f})")
 
-            if ball_detected and impact_frame is None:
-                for landmark in [mp_pose.PoseLandmark.LEFT_KNEE, mp_pose.PoseLandmark.RIGHT_KNEE]:
-                    knee = pose_results.pose_landmarks.landmark[landmark]
-                    if knee.visibility > 0.7:
-                        knee_x, knee_y = knee.x * frame_width, knee.y * frame_height
-                        if abs(knee_x - x) < 50 and abs(knee_y - y) < 50:
-                            impact_x, impact_y = x, y
-                            impact_frame = frame_idx
-                            logger.info(f"Impact point detected at frame {frame_idx}: ({impact_x:.2f}, {impact_y:.2f})")
-                            break
+                if ball_detected and impact_frame is None:
+                    for landmark in [mp_pose.PoseLandmark.LEFT_KNEE, mp_pose.PoseLandmark.RIGHT_KNEE]:
+                        knee = pose_results.pose_landmarks.landmark[landmark]
+                        if knee.visibility > 0.7:
+                            knee_x, knee_y = knee.x * frame_width, knee.y * frame_height
+                            if abs(knee_x - x) < 50 and abs(knee_y - y) < 50:
+                                impact_x, impact_y = x, y
+                                impact_frame = frame_idx
+                                logger.info(f"Impact point detected at frame {frame_idx}: ({impact_x:.2f}, {impact_y:.2f})")
+                                break
+            except Exception as e:
+                logger.error(f"Pose detection error: {str(e)}")
 
         # Pitch point
         if ball_detected and pitch_frame is None:
@@ -114,7 +125,15 @@ def process_video(video_path):
     cap.release()
 
     # Smooth trajectory
-    confident_positions = [(idx, x, y) for idx, x, y, conf in ball_positions if conf > 0.7]
+    confident_positions = [(idx, x, y) for idx, x, y, conf in ball_positions if conf > 0.5]
+    if not confident_positions:
+        logger.error("No confident ball detections found in the video")
+        output_video.release()
+        fig = go.Figure()
+        fig.add_annotation(text="No confident ball detections for trajectory plot", showarrow=False)
+        fig.update_layout(template="plotly_dark")
+        return output_path, "Error: No ball detections", fig
+
     smooth_trajectory = []
     if len(confident_positions) > 2:
         frames_range, x_coords, y_coords = zip(*confident_positions)
@@ -130,9 +149,9 @@ def process_video(video_path):
             logger.info(f"Generated smooth trajectory with {len(smooth_trajectory)} points")
         except Exception as e:
             logger.error(f"Interpolation failed: {str(e)}")
-            smooth_trajectory = [(int(x), int(y)) for idx, x, y, conf in ball_positions if conf > 0.7 and 0 <= x < frame_width and 0 <= y < frame_height]
+            smooth_trajectory = [(int(x), int(y)) for idx, x, y, conf in ball_positions if conf > 0.5 and 0 <= x < frame_width and 0 <= y < frame_height]
     else:
-        smooth_trajectory = [(int(x), int(y)) for idx, x, y, conf in ball_positions if conf > 0.7 and 0 <= x < frame_width and 0 <= y < frame_height]
+        smooth_trajectory = [(int(x), int(y)) for idx, x, y, conf in ball_positions if conf > 0.5 and 0 <= x < frame_width and 0 <= y < frame_height]
         logger.warning(f"Insufficient confident detections ({len(confident_positions)}), using raw confident positions")
 
     # LBW Decision
@@ -248,7 +267,7 @@ def process_video(video_path):
         fig.layout['yaxis2'].update(title="Y Coordinate (pixels)", range=[frame_height, 0])
     else:
         fig = go.Figure()
-        fig.add_annotation(text="No confident detections for trajectory plot", showarrow=False)
+        fig.add_annotation(text="No confident ball detections for trajectory plot", showarrow=False)
         fig.update_layout(template="plotly_dark")
 
     logger.info("Video processing completed")