Update app.py

app.py

@@ -165,9 +165,10 @@ def inpaint(image, width, height, overlap_percentage, num_inference_steps, resiz
     
     # blend foreground with background and alpha, less color bleeding
     #new_image = blend(foreground, np.array(background), alpha)
+    cnet_pure = cnet_image
     cnet_image.paste(result, (0, 0), mask)
 
-    return cnet_image, background, mask
+    return cnet_image, background, mask, result, cnet_pure, background
 
 def preview_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom):
     background, mask = prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom)
@@ -311,6 +312,18 @@ with gr.Blocks(css=css) as demo:
                     interactive=False,
                     label="Generated Mask",
                 )
+                result_pure = gr.Image(
+                    interactive=False,
+                    label="Generated Result",
+                )
+                cnet_pure = gr.Image(
+                    interactive=False,
+                    label="Cnet Pure",
+                )
+                background = gr.Image(
+                    interactive=False,
+                    label="background",
+                )
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)
                 with gr.Accordion("History and Mask", open=False):
                     history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
@@ -362,7 +375,7 @@ with gr.Blocks(css=css) as demo:
         inputs=[input_image, width_slider, height_slider, overlap_percentage, num_inference_steps,
                 resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
                 overlap_left, overlap_right, overlap_top, overlap_bottom],
-        outputs=[result, preview_image, result_mask],
+        outputs=[result, preview_image, result_mask, result_pure, cnet_pure, background],
     ).then(
         fn=lambda x, history: update_history(x, history),
         inputs=[result, history_gallery],