adds mask inversion checkbox

app.py

@@ -96,9 +96,14 @@ def fn_update_mask(
         masks_enabled: List[int], 
         max_kernel: int,
         min_kernel: int,
+        invert_mask: bool
     ):
     masks_enabled = [int(m.split(':')[0]) for m in masks_enabled]
     combined_mask = reduce(lambda x, y: x | y, [masks[i] for i in masks_enabled], np.zeros_like(masks[0], dtype=bool))
+
+    if invert_mask:
+        combined_mask = ~combined_mask
+
     combined_mask = clean_mask(combined_mask, max_kernel, min_kernel)
 
     masked_image = np.array(image).copy()
@@ -169,6 +174,7 @@ with demo:
     with gr.Row():
         max_slider = gr.Slider(minimum=1, maximum=99, value=23, step=2, label="Mask Overflow")
         min_slider = gr.Slider(minimum=1, maximum=99, value=5, step=2, label="Mask Denoising")
+        invert_mask = gr.Checkbox(label="Invert Mask")
 
         mask_checkboxes = gr.CheckboxGroup(interactive=True, label="Mask Selection")
 
@@ -184,11 +190,12 @@ with demo:
 
         inpainted_image = gr.Image(type='pil', label="Inpainted Image")
 
-    update_mask_inputs = [input_image, mask_storage, mask_checkboxes, max_slider, min_slider]
+    update_mask_inputs = [input_image, mask_storage, mask_checkboxes, max_slider, min_slider, invert_mask]
     update_mask_outputs = [mask_image, masked_image]
 
     # Clear checkbox group on input image change
     input_image.change(lambda: gr.CheckboxGroup.update(choices=[], value=[]), outputs=mask_checkboxes)
+    input_image.change(lambda: gr.Checkbox.update(value=False), outputs=invert_mask)
 
     # Segmentation button callback
     bt_masks.click(fn_segmentation, inputs=[input_image, max_slider, min_slider], outputs=[mask_storage, mask_checkboxes, mask_image, masked_image])
@@ -198,6 +205,7 @@ with demo:
     max_slider.change(fn_update_mask, inputs=update_mask_inputs, outputs=update_mask_outputs, show_progress=False)
     min_slider.change(fn_update_mask, inputs=update_mask_inputs, outputs=update_mask_outputs, show_progress=False)
     mask_checkboxes.change(fn_update_mask, inputs=update_mask_inputs, outputs=update_mask_outputs, show_progress=False)
+    invert_mask.change(fn_update_mask, inputs=update_mask_inputs, outputs=update_mask_outputs, show_progress=False)
 
     # Diffusion button callback
     bt_diffusion.click(fn_diffusion, inputs=[