Update app.py

app.py

@@ -265,14 +265,19 @@ def get_detection_data(image_data):
             # Debug: Log center and vertices to verify resolution
             print(f"Debug - Object {idx}: Center = {center}, Vertices = {get_box_vertices([x1, y1, x2, y2])}")
 
-            # Scale from 1536x1024 to 1512x1008
+            # Scale from 1536x1024 to 1512x1008, ensuring values fit within bounds
             scale_x = 1512 / 1536  # Width scaling factor
             scale_y = 1008 / 1024  # Height scaling factor
 
-            # Scale center and vertices to 1512x1008
-            scaled_center = (int(center[0] * scale_x), int(center[1] * scale_y))
-            scaled_vertices = [[int(x * scale_x), int(y * scale_y)] for x, y in get_box_vertices([x1, y1, x2, y2])]
-
+            # Scale and clamp center and vertices to 1512x1008
+            scaled_center = (
+                int(min(max(center[0] * scale_x, 0), 1511)),  # Clamp to [0, 1511]
+                int(min(max(center[1] * scale_y, 0), 1007))    # Clamp to [0, 1007]
+            )
+            scaled_vertices = [
+                [int(min(max(x * scale_x, 0), 1511)), int(min(max(y * scale_y, 0), 1007))]
+                for x, y in get_box_vertices([x1, y1, x2, y2])
+            ]
             # Debug: Log scaled center and vertices
             print(f"Debug - Object {idx}: Scaled Center = {scaled_center}, Scaled Vertices = {scaled_vertices}")