Organise layout and interactions

app.py

@@ -65,34 +65,45 @@ def image_inference(image: gr.Image, alpha: Optional[float] = None):
 
 
 with gr.Blocks(analytics_enabled=True, title=PAPER_TITLE, theme="soft") as demo:
+    # LAYOUT
     gr.Markdown(MARKDOWN_DESCRIPTION)
     with gr.Row():
         with gr.Column():
             curr_image = gr.Image(label="input", type="pil")
-            orig_image = gr.Image(
+            _orig_image = gr.Image(
                 label="orig. image", type="pil", visible=False, interactive=False
             )
             alpha_slider = gr.Slider(0.0, 1.0, value=0.7, step=0.1, label="alpha")
             with gr.Row():
-                clear_button = gr.ClearButton([curr_image, orig_image])
+                clear_button = gr.Button(value="Clear", variant="secondary")
                 run_button = gr.Button(value="Submit", variant="primary")
         with gr.Column():
             output_label = gr.Label(label="output", num_top_classes=5)
     examples = gr.Examples(
         examples=glob(os.path.join(os.path.dirname(__file__), "examples", "*.jpg")),
-        inputs=[orig_image],
+        inputs=[_orig_image],
         outputs=[output_label],
         fn=image_inference,
         cache_examples=True,
     )
     gr.Markdown(f"Check out the <a href={PAPER_URL}>original paper</a> for more information.")
 
-    curr_image.upload(prepare_image, [curr_image], [curr_image, orig_image])
-    curr_image.clear(lambda: None, [], [orig_image])
-    orig_image.change(prepare_image, [orig_image], [curr_image, orig_image])
+    # INTERACTIONS
+    # - change
+    _orig_image.change(prepare_image, [_orig_image], [curr_image, _orig_image])
+
+    # - upload
+    curr_image.upload(prepare_image, [curr_image], [curr_image, _orig_image])
+    curr_image.upload(lambda: None, [], [output_label])
+
+    # - clear
+    curr_image.clear(lambda: (None, None), [], [_orig_image, output_label])
+
+    # - click
+    clear_button.click(lambda: (None, None, None), [], [curr_image, _orig_image, output_label])
     run_button.click(image_inference, [curr_image, alpha_slider], [output_label])
 
 
 if __name__ == "__main__":
     demo.queue()
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch()