retain original color

app.py

@@ -136,15 +136,12 @@ def extract_and_paste_insect(original_image: np.ndarray, detection: DetectionRes
     insect_crop = original_image[ymin:ymax, xmin:xmax]
     mask_crop = mask[ymin:ymax, xmin:xmax]
 
-    insect = cv2.bitwise_and(insect_crop, insect_crop, mask=mask_crop)
-
     # 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)
-
-    x_offset, y_offset = xmin, ymin
-    background[y_offset:y_offset + combined.shape[0], x_offset:x_offset + combined.shape[1]] = combined
+    insect = cv2.bitwise_and(insect_crop, insect_crop, mask=mask_crop)
+    
+    # Place the insect on the yellow background
+    for c in range(3):
+        background[ymin:ymax, xmin:xmax, c] = background[ymin:ymax, xmin:xmax, c] * (1 - mask_crop) + insect[:, :, c] * mask_crop
 
 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