Updated visibility change callbacks

app.py

@@ -1,4 +1,5 @@
 import os
+from posixpath import basename
 
 import torch
 import gradio as gr
@@ -207,11 +208,12 @@ class ImageEditor(object):
 
 editor = ImageEditor()
 
-def change_component_visibility(visible, component):
-    return component.update(visible=visible)
+def change_component_visibility(component_types, invert_choices):
 
-def change_component_visibility_flipped(visible, component):
-    return component.update(visible=not visible)
+    def visibility_impl(visible):
+        return [component_types[idx].update(visible ^ invert_choices[idx]) for idx in range(len(component_types))]
+    
+    return visibility_impl
 
 blocks = gr.Blocks()
 
@@ -241,10 +243,10 @@ with blocks:
         img_output = gr.outputs.Image(type="file")
         vid_output = gr.outputs.Video()
 
-    for component in [edit_choice, vid_format_choice, vid_output, vid_button]:
-        video_choice.change(fn=change_component_visibility, inputs=[video_choice, component], outputs=component)
-    for component in [img_output, img_button]:
-        video_choice.change(fn=change_component_visibility_flipped, inputs=[video_choice, component], outputs=component)
+    visibility_fn = change_component_visibility(component_types=[gr.Checkbox, gr.Radio, gr.Video, gr.Button, gr.Image, gr.Button],
+                                                invert_choices=[False, False, False, False, True, True])
+
+    video_choice.change(fn=visibility_fn, inputs=video_choice, outputs=[edit_choice, vid_format_choice, vid_output, vid_button, img_output, img_button])
 
     img_button.click(fn=editor.predict, inputs=[input_img, style_choice, video_choice, edit_choice, vid_format_choice], outputs=img_output)
     vid_button.click(fn=editor.predict, inputs=[input_img, style_choice, video_choice, edit_choice, vid_format_choice], outputs=vid_output)