Update app.py

app.py

@@ -469,17 +469,16 @@ with gr.Blocks(css=css) as demo:
                 resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
                 overlap_left, overlap_right, overlap_top, overlap_bottom],
         outputs=result,
-    ).then(  # Update the history gallery
-        fn=lambda x, history: update_history(x[1], history),
-        inputs=[result, history_gallery],
-        outputs=history_gallery,
-        show_api=False
     ).then(  # Show the "Use as Input Image" button
         fn=lambda: gr.update(visible=True),
         inputs=None,
         outputs=use_as_input_button,
         show_api=False
-    )
+    ).then(  # Update the history gallery
+        fn=lambda x, history: update_history(x[1], history),
+        inputs=[result, history_gallery],
+        outputs=history_gallery,
+        show_api=False
 
     prompt_input.submit(  # Clear the result
         fn=clear_result,