support output to be moved to input

app.py

@@ -47,6 +47,14 @@ def run_invhalf(invhalfer, input_img, decoding_only, device="cuda"):
 def click_run(input_img, decoding_only):
     output = run_invhalf(invhalfer, input_img, decoding_only, device)
     return output
+    
+    
+def click_move(output_img, decoding_only):
+    if decoding_only:
+        radio_status = "Halftoning (Photo2Halftone)"
+    else:
+        radio_status = "Restoration (Halftone2Photo)"
+    return output_img, radio_status, None
 
 ## step 2: configure interface
 demo = gr.Blocks(title="ReversibleHalftoning")
@@ -63,8 +71,10 @@ with demo:
                 Button_run = gr.Button(value="Run")
         with gr.Column():
             Image_output = gr.Image(type="numpy", label="Output").style(height=480)
+            Button_move = gr.Button(value="Use it as input")
 
     Button_run.click(fn=click_run, inputs=[Image_input, Radio_mode], outputs=Image_output)
+    Button_move.click(fn=click_move, inputs=[Image_output, Radio_mode], outputs=[Image_input, Radio_mode, Image_output])
     
     if RUN_MODE != "local":
         gr.Examples(examples=[