Update app.py

app.py

@@ -263,16 +263,28 @@ def get_detection_data(image_data):
             # Convert BGR to RGB
             color_rgb = (int(color_bgr[2]), int(color_bgr[1]), int(color_bgr[0]))
 
+            #detections.append({
+            #"class_id": cls_id,
+            #"class_name": cls_name,
+            #"bounding_box": {
+            #   "vertices": get_box_vertices([x1, y1, x2, y2])
+            #},
+            #"position_3d": get_3d_position(center, depth_value, get_camera_matrix(depth_estimator)),
+            #"distance": depth_value,
+            #"color": color_rgb,
+            #"confidence": float(confidence)
+
             detections.append({
-            "class_id": cls_id,
-            "class_name": cls_name,
-            "bounding_box": {
-                "vertices": get_box_vertices([x1, y1, x2, y2])
-            },
-            "position_3d": get_3d_position(center, depth_value, get_camera_matrix(depth_estimator)),
-            "distance": depth_value,
-            "color": color_rgb,
-            "confidence": float(confidence)
+                "class_id": cls_id,
+                "class_name": cls_name,
+                "bounding_box": {
+                    "vertices": get_box_vertices([x1, y1, x2, y2])
+                },
+                "center_2d": center_2d,  # Add 2D center in screen space
+                "distance": depth_value,  # Depth in meters
+                "color": color_rgb,
+                "confidence": float(confidence)
+            })
     })
 
         response = {