yellow bg output1

app.py

@@ -141,14 +141,23 @@ def extract_and_paste_insect(original_image: np.ndarray, detection: DetectionRes
     xmin, ymin, xmax, ymax = mask_to_min_max(mask)
     insect_crop = original_image[ymin:ymax, xmin:xmax]
     mask_crop = mask[ymin:ymax, xmin:xmax]
+
+    # Create an all-yellow background
+    yellow_background = np.full_like(insect_crop, (0, 255, 255), dtype=np.uint8)
+
+    # Extract insect using the mask
     insect = cv2.bitwise_and(insect_crop, insect_crop, mask=mask_crop)
-    x_offset, y_offset = detection.box.xmin, detection.box.ymin
-    x_end, y_end = x_offset + insect.shape[1], y_offset + insect.shape[0]
+
+    # Prepare the area on the background to accept the insect
+    bg_region = yellow_background.copy()
     inverse_mask = cv2.bitwise_not(mask_crop)
-    bg_region = background[y_offset:y_end, x_offset:x_end]
     bg_ready = cv2.bitwise_and(bg_region, bg_region, mask=inverse_mask)
+
+    # Combine the insect with the prepared area
     combined = cv2.add(insect, bg_ready)
-    background[y_offset:y_end, x_offset:x_end] = combined
+
+    # Paste the insect onto the yellow background
+    background[ymin:ymax, xmin:xmax] = combined
 
 def create_yellow_background_with_insects(image: np.ndarray, detections: List[DetectionResult]) -> np.ndarray:
     yellow_background = np.full((image.shape[0], image.shape[1], 3), (0, 255, 255), dtype=np.uint8)