support mask

app.py

@@ -15,6 +15,7 @@ def sanitize_prompt(prompt):
 #@spaces.GPU(duration=120)
 def process_images(image, image2=None,inpaint_radius=3,blur_radius=25,edge_expand=8,inpaint_mode="Telea",progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Start Inpainting")
+    print("process_images")
     # I'm not sure when this happen
     if not isinstance(image, dict):
         if image2 == None:
@@ -32,7 +33,7 @@ def process_images(image, image2=None,inpaint_radius=3,blur_radius=25,edge_expan
          mask = image['layers'][0]
 
 
-    output = opencvinpaint.process_image(image["background"],mask,inpaint_radius,blur_radius,edge_expand,inpaint_mode)
+    output = opencvinpaint.process_cvinpaint(image["background"],mask,inpaint_radius,blur_radius,edge_expand,inpaint_mode)
         
     return output,mask
     
@@ -58,13 +59,13 @@ css="""
 """
 
 #css=css,
-demo_blocks = gr.Blocks(css=css, elem_id="demo-container")
-with demo_blocks as demo:
+
+with gr.Blocks(css=css, elem_id="demo-container") as demo:
     with gr.Column():
         gr.HTML(read_file("demo_header.html"))
     with gr.Row():
                 with gr.Column():
-                    image = gr.ImageEditor(sources=['upload','clipboard'],transforms=[],image_mode='RGB', layers=False,  elem_id="image_upload", type="pil", label="Upload",brush=gr.Brush(colors=["#333"], color_mode="fixed"))
+                    image = gr.ImageEditor(sources=['upload','clipboard'],transforms=[],image_mode='RGB', layers=False,  elem_id="image_upload", type="pil", label="Upload",brush=gr.Brush(colors=["#fff"], color_mode="fixed"))
                     with gr.Row(elem_id="prompt-container",  equal_height=False):
                         with gr.Row():
                             btn = gr.Button("Inpaint!", elem_id="run_button")
@@ -100,4 +101,4 @@ with demo_blocks as demo:
         """
     )
 
-demo_blocks.queue(max_size=25).launch(share=True)
+    demo.launch()

opencvinpaint.py

@@ -43,7 +43,8 @@ def blend_rgb_images(image1: np.ndarray, image2: np.ndarray, mask: np.ndarray) -
 
     return blended.astype(np.uint8)
 
-def process_image(image,mask_image,inpaint_radius,blur_radius,edge_expand,inpaint_mode):
+def process_cvinpaint(image,mask_image,inpaint_radius,blur_radius,edge_expand,inpaint_mode):
+    print("process cvinpaint")
     #print(blur_radius,",",edge_expand)
     cv_image = pil_to_cv(image)
 
@@ -92,5 +93,5 @@ def process_image(image,mask_image,inpaint_radius,blur_radius,edge_expand,inpain
 if __name__ == "__main__":
     image = Image.open(sys.argv[1])
     mask  = Image.open(sys.argv[2])
-    output = process_image(image,mask)
+    output = process_cvinpaint(image,mask)
     output.save(sys.argv[3])