Update app.py

app.py

@@ -132,7 +132,7 @@ def prepare_image_and_mask(image, width, height, overlap_percentage, resize_opti
     return background, mask
 
 @spaces.GPU
-def inpaint(image, width, height, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom):
+def inpaint(image, width, height, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, progress=gr.Progress(track_tqdm=True)):
     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)
     
     if not can_expand(background.width, background.height, width, height, alignment):
@@ -162,7 +162,7 @@ def inpaint(image, width, height, overlap_percentage, num_inference_steps, resiz
     result = result.convert("RGBA")
     cnet_image.paste(result, (0, 0), mask)
 
-    return background, cnet_image
+    return cnet_image, 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)
@@ -305,9 +305,9 @@ with gr.Blocks(css=css) as demo:
                     label="Generated Image",
                 )
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)
-
-                history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
-                preview_image = gr.Image(label="Preview")
+                with gr.Accordion("History and Mask", open=False):
+                    history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
+                    preview_image = gr.Image(label="Mask preview")
 
     def use_output_as_input(output_image):
         return gr.update(value=output_image[1])
@@ -376,7 +376,7 @@ with gr.Blocks(css=css) as demo:
                 overlap_left, overlap_right, overlap_top, overlap_bottom],
         outputs=[result, preview_image],
     ).then(
-        fn=lambda x, history: update_history(x[1], history),
+        fn=lambda x, history: update_history(x, history),
         inputs=[result, history_gallery],
         outputs=history_gallery,
     ).then(