runtime error fix

app.py

@@ -1,5 +1,5 @@
 import os
-os.system('pip install gradio==4.29.0')  # as gradio==4.29.0 doesn't work in requirements.txt
+os.system('pip install gradio==4.29.0') # as gradio==4.29.0 doesn't work in requirements.txt
 
 import random
 from dataclasses import dataclass
@@ -141,15 +141,14 @@ 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]
-
-    # Ensure that we keep the original colors of the insect
     insect = cv2.bitwise_and(insect_crop, insect_crop, mask=mask_crop)
-
-    x_offset, y_offset = xmin, ymin
+    x_offset, y_offset = detection.box.xmin, detection.box.ymin
     x_end, y_end = x_offset + insect.shape[1], y_offset + insect.shape[0]
-
-    # Place the insect onto the yellow background
-    background[y_offset:y_end, x_offset:x_end] = insect
+    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)
+    combined = cv2.add(insect, bg_ready)
+    background[y_offset:y_end, x_offset:x_end] = 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)
@@ -163,7 +162,7 @@ def draw_classification_boxes(image_with_insects, detections):
         label = detection.label
         score = detection.score
         box = detection.box
-        color = (0, 255, 255)  # Yellow color for bounding box
+        color = np.random.randint(0, 256, size=3).tolist()
 
         cv2.rectangle(image_with_insects, (box.xmin, box.ymin), (box.xmax, box.ymax), color, 2)
         (text_width, text_height), baseline = cv2.getTextSize(f"{label}: {score:.2f}", cv2.FONT_HERSHEY_SIMPLEX, 0.5, 2)
@@ -181,7 +180,7 @@ def draw_classification_boxes(image_with_insects, detections):
             cv2.FONT_HERSHEY_SIMPLEX,
             0.5,
             (255, 255, 255),
-            2
+             2
         )
     return image_with_insects
 
@@ -198,4 +197,4 @@ gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=[gr.Image(type="numpy"), gr.Image(type="numpy")],
     title="🐞 InsectSAM + GroundingDINO Inference",
-).launch()
+).launch()