Update app.py

app.py

@@ -62,12 +62,12 @@ def live_feed_generator(video_type, confidence_threshold=0.9):
 
     while cap.isOpened():
         ret, frame = cap.read()
-        if not ret:
-            print(f"Failed to read frame at count {state.frame_count}")
+        if not ret or frame is None or frame.size == 0:
+            print(f"Frame read failed or empty at frame count {state.frame_count}")
             cap.set(cv2.CAP_PROP_POS_FRAMES, 0)  # Loop the video
             continue
         else:
-            print(f"Failed to read frame at count {state.frame_count}")
+            print(f"Frame shape: {frame.shape} at frame count {state.frame_count}")
 
         state.frame_count += 1
         frame_pil = video_service.frame_to_pil(frame)
@@ -89,6 +89,7 @@ def live_feed_generator(video_type, confidence_threshold=0.9):
         # Draw detections on frame
         annotated_frame = video_service.draw_detections(frame, detections)
         annotated_frame_rgb = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
+        print(f"Annotated frame shape (RGB): {annotated_frame_rgb.shape}, dtype: {annotated_frame_rgb.dtype}")
         cv2.imshow("Debug Frame", annotated_frame_rgb)
         cv2.waitKey(1)
         print(f"Annotated frame shape: {annotated_frame_rgb.shape}")