Added use as input image for output image. and also submits when user clicks enter at prompt.

app.py

@@ -108,8 +108,7 @@ with gr.Blocks(fill_height=True, fill_width=True) as demo:
             type="pil",
             label="Input Image",
             crop_size=(1024, 1024),
-            layers=False,
-            sources=["upload"],
+            layers=False
         )
 
         result = ImageSlider(
@@ -117,15 +116,52 @@ with gr.Blocks(fill_height=True, fill_width=True) as demo:
             label="Generated Image",
         )
 
+    use_as_input_button = gr.Button("Use as Input Image", visible=False)  
+
+    def use_output_as_input(output_image):
+        return gr.update(value=output_image[1])
+
+    use_as_input_button.click(
+        fn=use_output_as_input,
+        inputs=[result],
+        outputs=[input_image]
+    )
+
     run_button.click(
         fn=clear_result,
         inputs=None,
         outputs=result,
+    ).then(
+        fn=lambda: gr.update(visible=False),
+        inputs=None,
+        outputs=use_as_input_button,
     ).then(
         fn=fill_image,
         inputs=[prompt, input_image, model_selection],
         outputs=result,
+    ).then(
+        fn=lambda: gr.update(visible=True),
+        inputs=None,
+        outputs=use_as_input_button,
+    )
+
+    prompt.submit(
+        fn=clear_result,
+        inputs=None,
+        outputs=result,
+    ).then(
+        fn=lambda: gr.update(visible=False),
+        inputs=None,
+        outputs=use_as_input_button,
+    ).then(
+        fn=fill_image,
+        inputs=[prompt, input_image, model_selection],
+        outputs=result,
+    ).then(
+        fn=lambda: gr.update(visible=True),
+        inputs=None,
+        outputs=use_as_input_button,
     )
 
 
-demo.launch(share=False)
+demo.launch(share=False)