Better hide information

demos/musicgen_app.py

@@ -221,9 +221,11 @@ def predict_full(model, model_path, decoder, text, melody, duration, topk, topp,
         [text], [melody], duration, progress=True,
         top_k=topk, top_p=topp, temperature=temperature, cfg_coef=cfg_coef,
         gradio_progress=progress)
+    information = "To download the output, right-click and click on Save as..."
+
     if USE_DIFFUSION:
-        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
+        return videos[0], wavs[0], information, videos[1], wavs[1]
+    return videos[0], wavs[0], information, None, None
 
 
 def toggle_audio_src(choice):
@@ -240,6 +242,14 @@ def toggle_diffusion(choice):
         return [gr.update(visible = False)] * 2
 
 
+def show_information(choice):
+    return [gr.update(visible = True)]
+
+
+def hide_information(choice):
+    return [gr.update(visible = False)]
+
+
 def ui_full(launch_kwargs):
     with gr.Blocks() as interface:
         gr.Markdown(
@@ -297,13 +307,15 @@ def ui_full(launch_kwargs):
                     _ = gr.Button("Abort all", variant="stop").click(fn=interrupt, queue = False)
             with gr.Column():
                 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()
+                audio_output = gr.Audio(label = "Generated Music (wav)", type='filepath', autoplay = True, show_download_button = True)
+                output_hint = gr.Label(label = "Information")
                 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(hide_information, decoder, [
+            output_hint
         ], queue=False, show_progress=False).then(predict_full, inputs = [
             model,
             model_path,
@@ -322,7 +334,10 @@ def ui_full(launch_kwargs):
             output_hint,
             diffusion_output,
             audio_diffusion
-        ], scroll_to_output = True)
+        ], scroll_to_output = True).then(show_information, decoder, [
+            output_hint
+        ], queue=False, show_progress=False)
+
         radio.change(toggle_audio_src, radio, [melody], queue = False, show_progress = False)
 
         gr.Examples(