Updated visibility change callbacks

app.py

@@ -120,7 +120,8 @@ class ImageEditor(object):
         print("setup complete")
 
     def get_style_list(self):
-        style_list = ['all', 'list - enter below']
+        # style_list = ['all', 'list - enter below']
+        style_list = []
 
         for key in self.generators:
             style_list.append(key)
@@ -208,12 +209,17 @@ class ImageEditor(object):
 
 editor = ImageEditor()
 
-def change_component_visibility(component_types, invert_choices):
+# def change_component_visibility(component_types, invert_choices):
 
-    def visibility_impl(visible):
-        return [component_types[idx].update(visible=visible ^ invert_choices[idx]) for idx in range(len(component_types))]
+#     def visibility_impl(visible):
+#         return [component_types[idx].update(visible=visible ^ invert_choices[idx]) for idx in range(len(component_types))]
     
-    return visibility_impl
+#     return visibility_impl
+
+def group_visibility(visible):
+    print("visible: ", visible)
+
+    return gr.Group.update(visibile=visible)
 
 blocks = gr.Blocks()
 
@@ -234,9 +240,14 @@ with blocks:
 
             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")
+            video_options_group = gr.Group()
+            
+            with video_options_group:
+                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")
         
+            # img_button = gr.Button("Edit Image")
+            # vid_button = gr.Button("Generate Video")
             img_button = gr.Button("Edit Image")
             vid_button = gr.Button("Generate Video")
 
@@ -244,13 +255,45 @@ with blocks:
             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])
-
-    video_choice.change(fn=visibility_fn, inputs=video_choice, outputs=[edit_choice, vid_format_choice, vid_output, vid_button, img_output, img_button])
+    # 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])
+    video_choice.change(fn=group_visibility, inputs=video_choice, outputs=video_options_group)
     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)
+    
+    # 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.Tabs():
+    #     with gr.TabItem("Edit Images"):
+    #         with gr.Row():
+    #             with gr.Column():
+                    
+            
+
+    #         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")
+        
+    #         img_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()
+
+    # 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)
 
     article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.00946' target='_blank'>StyleGAN-NADA: CLIP-Guided Domain Adaptation of Image Generators</a> | <a href='https://stylegan-nada.github.io/' target='_blank'>Project Page</a> | <a href='https://github.com/rinongal/StyleGAN-nada' target='_blank'>Code</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=rinong_sgnada' alt='visitor badge'></center>"
     gr.Markdown(article)