import cv2

from src.data import load_data
from src.img_utils import apply_color_overlay, blur_background, desaturate_background


def main():
    print("Starting Abduction Demo")
    labeled_images = load_data()
    image = labeled_images[0].image
    mask = labeled_images[0].create_mask([labeled_images[0].labels[0]])

    overlay_result = apply_color_overlay(image, mask, color=(255, 100, 0), alpha=0.4)
    cv2.imwrite(".tmp-data/highlight_overlay.jpg", overlay_result)

    blur_result = blur_background(image, mask, blur_intensity=(51, 51))
    cv2.imwrite(".tmp-data/highlight_blurred.jpg", blur_result)

    desaturate_result = desaturate_background(image, mask)
    cv2.imwrite(".tmp-data/highlight_desaturated.jpg", desaturate_result)


if __name__ == "__main__":
    main()