Update UI

app.py

@@ -14,10 +14,7 @@ def generate_video(text, auth_openai, auth_elevenlabs, auth_replicate, auth_rev)
     # generated_music = bgm.text2audio(text=text, duration=20, guidance_scale=5, random_seed=24, n_candidates=3)
 
     generated_video = join.join_artifacts()
-    return generated_video
-
-def download_video(video_path):
-    return video_path
+    return [generated_video, generated_video]
 
 with gr.Blocks() as demo:
     gr.Markdown("Generate a narrated horror story video from a text prompt.")
@@ -33,12 +30,10 @@ with gr.Blocks() as demo:
             generate_button = gr.Button("Generate Video")
 
         with gr.Column():
-            video_out = gr.Video(label="Output", interactive=False)
-            download_button = gr.Button("Download")
-            output_file = gr.outputs.File(label="Video Output")
+            video_out = gr.Video(label="Video Player", interactive=False)
+            output_file = gr.outputs.File(label="File Output")
             output_file.type = "video/mp4"
 
-    generate_button.click(generate_video, inputs=[prompt_input, auth_openai_input, auth_eleven_input, auth_replicate_input, auth_rev_input], outputs=[video_out])
-    download_button.click(download_video, inputs=[video_out], outputs=[output_file])
+    generate_button.click(generate_video, inputs=[prompt_input, auth_openai_input, auth_eleven_input, auth_replicate_input, auth_rev_input], outputs=[video_out, output_file])
 
 demo.launch(debug=True, enable_queue=True)