Display/hide information

demos/musicgen_app.py

@@ -222,8 +222,8 @@ def predict_full(model, model_path, decoder, text, melody, duration, topk, topp,
         top_k=topk, top_p=topp, temperature=temperature, cfg_coef=cfg_coef,
         gradio_progress=progress)
     if USE_DIFFUSION:
-        return videos[0], wavs[0], videos[1], wavs[1]
-    return videos[0], wavs[0], None, None
+        return videos[0], wavs[0], "To download the output, right-click and click on Save as...", videos[1], wavs[1]
+    return videos[0], wavs[0], "To download the output, right-click and click on Save as...", None, None
 
 
 def toggle_audio_src(choice):
@@ -240,17 +240,6 @@ def toggle_diffusion(choice):
         return [gr.update(visible = False)] * 2
 
 
-def toggle_hint(choice):
-    if choice == "MultiBand_Diffusion":
-        return [gr.update(visible=True)] * 2
-    else:
-        return [gr.update(visible=False)] * 2
-
-
-def toggle_output_hint(isVisible):
-    return [gr.update(visible = isVisible)]
-
-
 def ui_full(launch_kwargs):
     with gr.Blocks() as interface:
         gr.Markdown(
@@ -307,17 +296,34 @@ def ui_full(launch_kwargs):
                     # Adapted from https://github.com/rkfg/audiocraft/blob/long/app.py, MIT license.
                     _ = gr.Button("Abort all", variant="stop").click(fn=interrupt, queue = False)
             with gr.Column():
-                output_hint = gr.Label("To download the output, right-click and click on Save as...", label = "To download the output, right-click and click on Save as...")
                 output = gr.Video(label="Generated Music")
                 audio_output = gr.Audio(label="Generated Music (wav)", type='filepath', autoplay = True, show_download_button = True)
+                output_hint = gr.Label()
                 diffusion_output = gr.Video(label="MultiBand Diffusion Decoder")
                 audio_diffusion = gr.Audio(label="MultiBand Diffusion Decoder (wav)", type='filepath')
-        submit.click(toggle_diffusion, decoder, [diffusion_output, audio_diffusion], queue=False,
-                     show_progress=False).then(toggle_hint, decoder, [output_hint, audio_diffusion], queue=False,
-                     show_progress=False).then(predict_full, inputs=[model, model_path, decoder, text, melody, duration, topk, topp,
-                                                                     temperature, cfg_coef, output_hint],
-                                               outputs=[output, audio_output, diffusion_output, audio_diffusion], scroll_to_output = True)
-        radio.change(toggle_audio_src, radio, [melody], queue=False, show_progress=False)
+        submit.click(toggle_diffusion, decoder, [
+            diffusion_output,
+            audio_diffusion
+        ], queue=False, show_progress=False).then(predict_full, inputs = [
+            model,
+            model_path,
+            decoder,
+            text,
+            melody,
+            duration,
+            topk,
+            topp,
+            temperature,
+            cfg_coef,
+            output_hint
+        ], outputs = [
+            output,
+            audio_output,
+            output_hint,
+            diffusion_output,
+            audio_diffusion
+        ], scroll_to_output = True)
+        radio.change(toggle_audio_src, radio, [melody], queue = False, show_progress = False)
 
         gr.Examples(
             fn=predict_full,