Update app.py

app.py

@@ -50,30 +50,25 @@ def annotate(image: Union[Image.Image, np.ndarray], detection_results: List[Dete
     image_cv2 = np.array(image) if isinstance(image, Image.Image) else image
     image_cv2 = cv2.cvtColor(image_cv2, cv2.COLOR_RGB2BGR)
     
-    # Create an empty yellow background
+    # Create a completely yellow background
     yellow_background = np.full(image_cv2.shape, (0, 255, 255), dtype=np.uint8)
     
     for detection in detection_results:
+        box = detection.box
         mask = detection.mask
         
+        # Drawing bounding box with yellow fill
+        cv2.rectangle(yellow_background, (box.xmin, box.ymin), (box.xmax, box.ymax), (0, 255, 255), cv2.FILLED)
+        
         if mask is not None:
-            mask_uint8 = (mask * 255).astype(np.uint8)
-            mask_3_channel = cv2.merge([mask_uint8, mask_uint8, mask_uint8])
-            
-            # Invert the mask to get the background part
-            inverted_mask = cv2.bitwise_not(mask_uint8)
-            inverted_mask_3_channel = cv2.merge([inverted_mask, inverted_mask, inverted_mask])
+            mask = mask.astype(bool)
+            mask_3_channel = np.stack([mask, mask, mask], axis=-1)
             
             # Extract insect region using mask
-            insect_region = cv2.bitwise_and(image_cv2, mask_3_channel)
-            yellow_background_region = cv2.bitwise_and(yellow_background, inverted_mask_3_channel)
-            
-            # Combine the insect region and the yellow background region
-            combined_region = cv2.add(insect_region, yellow_background_region)
+            insect_region = image_cv2 * mask_3_channel
             
-            # Apply combined region onto the yellow background
-            yellow_background = cv2.bitwise_and(yellow_background, inverted_mask_3_channel)
-            yellow_background = cv2.add(yellow_background, combined_region)
+            # Overlay insect region within the bounding box on the yellow background
+            yellow_background[mask_3_channel] = insect_region[mask_3_channel]
 
     return cv2.cvtColor(yellow_background, cv2.COLOR_BGR2RGB)
 
@@ -82,7 +77,7 @@ def plot_detections(image: Union[Image.Image, np.ndarray], detections: List[Dete
     return annotated_image
 
 def load_image(image: Union[str, Image.Image]) -> Image.Image:
-    if isinstance(image, str) and image.startswith("http"):
+    if isinstance(image, str) and image.startsWith("http"):
         image = Image.open(requests.get(image, stream=True).raw).convert("RGB")
     elif isinstance(image, str):
         image = Image.open(image).convert("RGB")