This PR allows to upload existing mask

app.py

@@ -43,7 +43,8 @@ EXAMPLES = [
         42,
         False,
         0.85,
-        20
+        20,
+        None
     ],
     [
         {
@@ -58,7 +59,8 @@ EXAMPLES = [
         42,
         False,
         0.9,
-        20
+        20,
+        None
     ]
 ]
 
@@ -163,24 +165,28 @@ def process(
     seed_slicer: int,
     randomize_seed_checkbox: bool,
     strength_slider: float,
-    num_inference_steps_slider: int
+    num_inference_steps_slider: int,
+    uploaded_mask: Image.Image
 ):
     if not inpainting_prompt_text:
         gr.Info("Please enter inpainting text prompt.")
         return None, None
 
     image_path = input_image_editor['background']
-    mask_path = input_image_editor['layers'][0]
 
     image = Image.open(image_path)
-    mask = Image.open(mask_path)
+    if uploaded_mask is None:
+        mask_path = input_image_editor['layers'][0]
+        mask = Image.open(mask_path)
+    else:
+        mask = uploaded_mask
 
     if not image:
         gr.Info("Please upload an image.")
         return None, None
 
     if is_mask_empty(mask) and not masking_prompt_text:
-        gr.Info("Please draw a mask or enter a masking prompt.")
+        gr.Info("Please draw a mask, upload a mask or enter a masking prompt.")
         return None, None
 
     if not is_mask_empty(mask) and masking_prompt_text:
@@ -240,6 +246,9 @@ with gr.Blocks() as demo:
                 submit_button_component = gr.Button(
                     value='Submit', variant='primary', scale=0)
 
+            with gr.Accordion("Upload a mask", open = False):
+                uploaded_mask_component = gr.Image(label = "Already made mask (white pixels will be preserved, black pixels will be redrawn)", sources = ["upload"], type = "pil")
+
             with gr.Accordion("Advanced Settings", open=False):
                 masking_prompt_text_component = gr.Text(
                     label="Masking prompt",
@@ -320,7 +329,8 @@ with gr.Blocks() as demo:
             seed_slicer_component,
             randomize_seed_checkbox_component,
             strength_slider_component,
-            num_inference_steps_slider_component
+            num_inference_steps_slider_component,
+            uploaded_mask_component
         ],
         outputs=[
             output_image_component,
@@ -341,7 +351,8 @@ with gr.Blocks() as demo:
             seed_slicer_component,
             randomize_seed_checkbox_component,
             strength_slider_component,
-            num_inference_steps_slider_component
+            num_inference_steps_slider_component,
+            uploaded_mask_component
         ],
         outputs=[
             output_image_component,