Update app.py

app.py

@@ -59,17 +59,14 @@ def predict_from_landinglens(image_path):
                 coverage_percentage = (mask_area / total_pixels) * 100
                 label_name = bitmap_data.get("label_name", f"Mask {i}")
                 coverage_info.append(f"{label_name}: {coverage_percentage:.2f}%")
+
                 
-                # Create colored overlay
-                colored_mask = np.zeros_like(img_array)
-                colored_mask[mask > 0] = [255, 0, 0]  # Red overlay for mask
-                
-                # Combine original image with colored mask
-                alpha = 0.5  # Transparency of the overlay
-                combined = cv2.addWeighted(img_array, 1, colored_mask, alpha, 0)
                 
-                # Convert to PIL Image
-                masked_image = Image.fromarray(combined)
+                # Create colored overlay using Pillow
+                mask_image = Image.fromarray((mask > 0).astype(np.uint8) * 255)
+                overlay = Image.new("RGB", mask_image.size, (255, 0, 0))  # Red overlay
+                mask_image = mask_image.convert("RGB")
+                masked_image = Image.blend(original_img, overlay, alpha=0.5 * mask_image.convert("L").point(lambda x: x > 0 and 1))
                 masked_images.append(masked_image)
 
             except Exception as e: