Update app.py

app.py

@@ -44,22 +44,18 @@ def zero_background(image, new_bgr=None):
     return make_flatten_background(image, img_cut, mask)
 
 
-
 def parse_face(image):
     return clients["FaceParser"].predict(image, api_name="/predict")
 
 
-def mirror(x):
-    return x
+def mirror(x, y):
+    return x, y
 
 
 def active_first():
     return gr.Tabs.update(selected=0)
 
 
-def clear():
-    return None, None
-
 
 footer = r"""
 <center>
@@ -94,14 +90,12 @@ with gr.Blocks(title="Face Shine") as app:
 
                 with gr.Column(scale=2):
                     with gr.Row():
-                        img_input = gr.Image(label="Input", type="filepath")
-                    with gr.Row():
-                        btn_reset = gr.Button(value="Reset", variant="stop")
-                        btn_swap = gr.Button(value="Ok", variant="primary")
-
+                        img_input = gr.Image(label="Image workspace", type="filepath")
                 with gr.Column(scale=2):
                     with gr.Row():
-                        img_output = gr.Image(label="Result", type="filepath", interactive=False)
+                        img_output = gr.Image(label="Image result", type="filepath", interactive=False)
+                    with gr.Row():
+                        btn_swap = gr.Button(value="<= Swap =>", variant="primary")
 
         with gr.TabItem("Examples", id=1):
             gr.Examples(examples=glob("lowres/*"), inputs=[img_input], label="Low resolution")
@@ -126,11 +120,10 @@ with gr.Blocks(title="Face Shine") as app:
     btn_color.click(colorize_photo, inputs=[img_input], outputs=[img_output])
     btn_light.click(enhance_light, inputs=[img_input], outputs=[img_output])
     btn_clear.click(zero_background, inputs=[img_input], outputs=[img_output])
-    btn_swap.click(mirror, inputs=[img_output], outputs=[img_input])
-    btn_reset.click(clear, outputs=[img_input, img_output])
+    btn_swap.click(mirror, inputs=[img_output, img_input], outputs=[img_input, img_output])
 
     with gr.Row():
         gr.HTML(footer)
 
 app.launch(share=False, debug=True, show_error=True)
-app.queue()
+app.queue()