Ensure the insect retains its original color

app.py

@@ -138,11 +138,13 @@ def extract_and_paste_insect(original_image: np.ndarray, detection: DetectionRes
 
     insect = cv2.bitwise_and(insect_crop, insect_crop, mask=mask_crop)
 
-    x_offset, y_offset = xmin, ymin
-    x_end, y_end = x_offset + insect.shape[1], y_offset + insect.shape[0]
+    # Ensure the insect retains its original color
+    color_insect = cv2.bitwise_and(insect_crop, insect_crop, mask=mask_crop)
+    non_mask_area = cv2.bitwise_and(background[ymin:ymax, xmin:xmax], background[ymin:ymax, xmin:xmax], mask=cv2.bitwise_not(mask_crop))
+    combined = cv2.add(non_mask_area, color_insect)
 
-    insect_area = background[y_offset:y_end, x_offset:x_end]
-    insect_area[mask_crop == 1] = insect[mask_crop == 1]
+    x_offset, y_offset = xmin, ymin
+    background[y_offset:y_offset + combined.shape[0], x_offset:x_offset + combined.shape[1]] = 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)  # BGR for yellow