This PR allows to optionally upload a pre-existing mask

app.py

@@ -17,20 +17,23 @@ pipe = FluxInpaintPipeline.from_pretrained(
 
 
 @spaces.GPU()
-def process(input_image_editor, input_text, strength, progress=gr.Progress(track_tqdm=True)):
+def process(input_image_editor, uploaded_mask, input_text, strength, progress=gr.Progress(track_tqdm=True)):
     if not input_text:
         gr.Info("Please enter a text prompt.")
         return None
 
     image = input_image_editor['background']
-    mask_image = input_image_editor['layers'][0]
+    if uploaded_mask is None:
+        mask_image = input_image_editor['layers'][0]
+    else:
+        mask_image = uploaded_mask
 
     if not image:
         gr.Info("Please upload an image.")
         return None
 
     if not mask_image:
-        gr.Info("Please draw a mask on the image.")
+        gr.Info("Please draw or upload a mask on the image.")
         return None
 
     width, height = image.size
@@ -70,6 +73,8 @@ with gr.Blocks() as demo:
                 step=0.01,
                 label="Strength"
             )
+            with gr.Accordion("Upload a mask", open = False):
+                uploaded_mask_component = gr.Image(label = "Already made mask (black pixels will be preserved, white pixels will be redrawn)", sources = ["upload"], type = "pil")
             submit_button_component = gr.Button(
                 value='Submit', variant='primary')
         with gr.Column():
@@ -80,6 +85,7 @@ with gr.Blocks() as demo:
         fn=process,
         inputs=[
             input_image_editor_component,
+            uploaded_mask_component,
             input_text_component,
             strength_slider
         ],