Display or not the debug items

app.py

@@ -133,6 +133,12 @@ def predict(source_img, enlarge_top, enlarge_right, enlarge_bottom, enlarge_left
         mask_image
     ]
 
+def toggle_debug(is_debug_mode):
+    if is_debug_mode:
+        return [gr.update(visible = True)] * 3
+    else:
+        return [gr.update(visible = False)] * 3
+
 with gr.Blocks() as interface:
     gr.Markdown(
         """
@@ -195,13 +201,14 @@ with gr.Blocks() as interface:
     with gr.Row():
         information = gr.Label(label = "Information")
     with gr.Row():
-        original_image = gr.Image(label = "Original image")
+        original_image = gr.Image(label = "Original image", visible = False)
     with gr.Row():
-        enlarged_image = gr.Image(label = "Enlarged image")
+        enlarged_image = gr.Image(label = "Enlarged image", visible = False)
     with gr.Row():
-        mask_image = gr.Image(label = "Mask image")
+        mask_image = gr.Image(label = "Mask image", visible = False)
 
-    submit.click(predict, inputs = [
+    submit.click(toggle_debug, debug_mode, [original_image, enlarged_image, mask_image], queue = False,
+                     show_progress = False).then(predict, inputs = [
         source_img,
         enlarge_top,
         enlarge_right,