Updated to new blocks SDK, testing visibility

README.md

@@ -4,7 +4,7 @@ emoji: 🌖
 colorFrom: blue
 colorTo: gray
 sdk: gradio
-sdk_version: 3.0b8
+sdk_version: 3.0.1
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -236,17 +236,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)
+# def group_visibility(visible):
+#     print("visible: ", visible)
 
-    return gr.Group.update(visibile=visible)
+#     return gr.Group.update(visibile=visible)
 
 blocks = gr.Blocks()
 
@@ -259,61 +259,59 @@ with blocks:
         "For more information about the paper and code for training your own models (with examples OR text), see below."
     )
 
-    # 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!")
+        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)
-            
-    #         video_options_group = gr.Group()
+            video_choice = gr.inputs.Checkbox(default=False, label="Generate Video?", optional=False)
             
-    #         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")
+            loop_styles       = gr.inputs.Checkbox(default=True, label="Loop video back to the initial style?", visible=False)
+            edit_choice       = gr.inputs.Checkbox(default=False, label="With Editing?", visible=False)
+            vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format", visible=False)
         
-    #         # img_button = gr.Button("Edit Image")
-    #         # vid_button = gr.Button("Generate Video")
-    #         img_button = gr.Button("Edit Image")
-    #         vid_button = gr.Button("Generate Video")
+            # img_button = gr.Button("Edit Image")
+            # vid_button = gr.Button("Generate Video")
+            img_button = gr.Button("Edit Image")
+            vid_button = gr.Button("Generate Video", visible=False)
 
-    #     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(visible=False)
 
-    # # 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])
+    visibility_fn = change_component_visibility(component_types=[gr.Checkbox, gr.Radio, gr.Video, gr.Button, gr.Image, gr.Button, gr.Checkbox],
+                                                invert_choices=[False, False, False, False, True, True, False])
 
-    # # 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=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)
+    img_button.click(fn=editor.edit_image, inputs=[input_img, style_choice], outputs=img_output)
+    vid_button.click(fn=editor.edit_video, inputs=[input_img, style_choice, edit_choice, vid_format_choice, loop_styles], outputs=vid_output)
     
-    with gr.Row():
-        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.Column():
-                img_button = gr.Button("Edit Image")
-            with gr.Column():
-                img_output = gr.outputs.Image(type="file", label="Output Image")
+    # with gr.Row():
+    #     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.Column():
+    #             img_button = gr.Button("Edit Image")
+    #         with gr.Column():
+    #             img_output = gr.outputs.Image(type="file", label="Output Image")
             
-        with gr.TabItem("Create Video"):
-            with gr.Column():
-                with gr.Row():
-                    vid_button = gr.Button("Generate Video")
-                    loop_styles       = gr.inputs.Checkbox(default=True, label="Loop video back to the initial style?")
-                    edit_choice       = gr.inputs.Checkbox(default=False, label="With latent space editing?")
-                    vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format")
+    #     with gr.TabItem("Create Video"):
+    #         with gr.Column():
+    #             with gr.Row():
+    #                 vid_button = gr.Button("Generate Video")
+    #                 loop_styles       = gr.inputs.Checkbox(default=True, label="Loop video back to the initial style?")
+    #                 edit_choice       = gr.inputs.Checkbox(default=False, label="With latent space editing?")
+    #                 vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format")
             
-            with gr.Column():
-                vid_output = gr.outputs.Video(label="Output Video")
+    #         with gr.Column():
+    #             vid_output = gr.outputs.Video(label="Output Video")
 
-    img_button.click(fn=editor.edit_image, inputs=[input_img, style_choice], outputs=img_output)
-    vid_button.click(fn=editor.edit_video, inputs=[input_img, style_choice, edit_choice, vid_format_choice, loop_styles], outputs=vid_output)
+    # img_button.click(fn=editor.edit_image, inputs=[input_img, style_choice], outputs=img_output)
+    # vid_button.click(fn=editor.edit_video, inputs=[input_img, style_choice, edit_choice, vid_format_choice, loop_styles], 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)

generate_videos.py

@@ -151,7 +151,7 @@ def video_from_interpolations(fps, output_dir):
                "-r", f"{fps}", 
                "-i", f"{output_dir}/%03d.jpg", 
                "-c:v", "libx264", 
-               "-vf", f"fps={fps}, loop=loop=-1", 
+               "-vf", f"fps={fps}", 
                "-pix_fmt", "yuv420p", 
                f"{output_dir}/out.mp4"]