experimenting with extraction of bbx

app.py

@@ -59,13 +59,14 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
         
         if mask is not None:
             mask_uint8 = (mask * 255).astype(np.uint8)
-            contours, _ = cv2.findContours(mask_uint8, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-            cv2.drawContours(yellow_background, contours, -1, (0, 0, 0), cv2.FILLED)
-
-            # Extract insect region using mask
-            insect_region = cv2.bitwise_and(image_cv2, image_cv2, mask=mask)
-
-            yellow_background[mask > 0] = insect_region[mask > 0]
+            mask_inv = cv2.bitwise_not(mask_uint8)
+            
+            # Extract insect using mask
+            insect_region = cv2.bitwise_and(image_cv2, image_cv2, mask=mask_uint8)
+            yellow_background_masked = cv2.bitwise_and(yellow_background, yellow_background, mask=mask_inv)
+            insect_combined = cv2.add(insect_region, yellow_background_masked)
+            
+            yellow_background = cv2.add(yellow_background, insect_combined)
 
     return cv2.cvtColor(yellow_background, cv2.COLOR_BGR2RGB)
 
@@ -217,4 +218,4 @@ gr.Interface(
     outputs=[gr.Image(type="numpy"), gr.Textbox()],
     title="InsectSAM 🐞",
     examples=examples
-).launch()
+).launch()