Update app.py

app.py

@@ -253,8 +253,15 @@ def process_video(video_data, progress_callback=None):
                     seen_violations.add(label)
 
                     xywh_array = box.xywh.cpu().numpy()
-                    if xywh_array.shape[0] == 0:
-                        continue
+                    if xywh_array.size == 0 or xywh_array.shape[1] < 4:
+                        xyxy_array = box.xyxy.cpu().numpy()
+                        if xyxy_array.size == 0 or xyxy_array.shape[1] < 4:
+                            logger.warning("No valid bounding box info found, skipping box")
+                            continue
+                        xmin, ymin, xmax, ymax = xyxy_array[0]
+                        width = xmax - xmin
+                        height = ymax - ymin
+                        xywh_array = np.array([[xmin, ymin, width, height]])
 
                     violation = {
                         "frame": frame_count,