This PR allows to upload existing mask (#1)

- This PR allows to upload existing mask (91805f726809ed0cdf35eb697b2f205cbcb4a7ee)


Co-authored-by: Fabrice TIERCELIN <[email protected]>

app.py

@@ -48,7 +48,8 @@ EXAMPLES = [
         42,
         False,
         0.85,
-        30
+        30,
+        None
     ],
     [
         {
@@ -60,7 +61,8 @@ EXAMPLES = [
         42,
         False,
         0.85,
-        30
+        30,
+        None
     ]
 ]
 
@@ -101,6 +103,7 @@ def process(
     randomize_seed_checkbox: bool,
     strength_slider: float,
     num_inference_steps_slider: int,
+    uploaded_mask: Image.Image,
     progress=gr.Progress(track_tqdm=True)
 ):
     if not input_text:
@@ -108,14 +111,17 @@ def process(
         return None, None
 
     image = input_image_editor['background']
-    mask = input_image_editor['layers'][0]
+    if uploaded_mask is None:
+        mask = input_image_editor['layers'][0]
+    else:
+        mask = uploaded_mask
 
     if not image:
         gr.Info("Please upload an image.")
         return None, None
 
     if not mask:
-        gr.Info("Please draw a mask on the image.")
+        gr.Info("Please draw a mask on the image or upload a mask.")
         return None, None
 
     width, height = resize_image_dimensions(original_resolution_wh=image.size)
@@ -162,6 +168,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):
                 seed_slicer_component = gr.Slider(
                     label="Seed",
@@ -211,7 +220,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,
@@ -229,7 +239,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,