Update app.py

app.py

@@ -227,11 +227,11 @@ def wav_list_to_wav(wav_list):
 
 def on_submit(prompt_1, prompt_2, steps, num_iterations, feel, seed):
     if prompt_1 == "":
-        return None, gr.update(value="First prompt is required.")
+        return None, gr.update(value="First prompt is required."), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
     if prompt_2 == "":
-        return generate(prompt_1, steps, num_iterations, feel, seed), None, gr.update(visible=True)
+        return generate(prompt_1, steps, num_iterations, feel, seed), None, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
     else:
-        return generate_riffuse(prompt_1, steps, num_iterations, feel, prompt_end=prompt_2, seed_start=seed), None, gr.update(visible=True)
+        return generate_riffuse(prompt_1, steps, num_iterations, feel, prompt_end=prompt_2, seed_start=seed), None, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 
 def on_num_iterations_change(n, prompt_2):
@@ -289,13 +289,13 @@ with gr.Blocks(css=css) as app:
         with gr.Column():
             video = gr.Video(elem_id="riff-video")
 
-            with gr.Column(elem_id="share-btn-container", visible=False) as share_group:
+            with gr.Group(elem_id="share-btn-container"):
                 community_icon = gr.HTML(community_icon_html, visible=False)
                 loading_icon = gr.HTML(loading_icon_html, visible=False)
                 share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
 
     inputs = [prompt_1, prompt_2, steps, num_iterations, feel, seed]
-    outputs = [video, info, share_group]
+    outputs = [video, info, community_icon, loading_icon, share_button]
 
     num_iterations.change(on_num_iterations_change, [num_iterations, prompt_2], [info])
     prompt_1.submit(on_submit, inputs, outputs)