Updated visibility change callbacks

app.py

@@ -211,7 +211,7 @@ editor = ImageEditor()
 def change_component_visibility(component_types, invert_choices):
 
     def visibility_impl(visible):
-        return [component_types[idx].update(visible ^ invert_choices[idx]) for idx in range(len(component_types))]
+        return [component_types[idx].update(visible=visible ^ invert_choices[idx]) for idx in range(len(component_types))]
     
     return visibility_impl
 
@@ -226,22 +226,23 @@ with blocks:
         "For more information about the paper and code for training your own models (with examples OR text), see below."
     )
 
-    with gr.Column():
-        input_img    = gr.inputs.Image(type="filepath", label="Input image")
-        style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), type="value", label="Choose your styles!")
-        
-        with gr.Row():
+    with gr.Row():
+
+        with gr.Column():
+            input_img    = gr.inputs.Image(type="filepath", label="Input image")
+            style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), type="value", label="Choose your styles!")
+
             video_choice = gr.inputs.Checkbox(default=False, label="Generate Video?", optional=False)
             
             edit_choice       = gr.inputs.Checkbox(default=False, label="With Editing?", optional=False)
             vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format")
-        with gr.Row():
+        
             img_button = gr.Button("Edit Image")
-            vid_button = gr.Button("Edit Image")
+            vid_button = gr.Button("Generate Video")
 
-    with gr.Column():
-        img_output = gr.outputs.Image(type="file")
-        vid_output = gr.outputs.Video()
+        with gr.Column():
+            img_output = gr.outputs.Image(type="file")
+            vid_output = gr.outputs.Video()
 
     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])