Make output video display properly

app.py

@@ -10,13 +10,13 @@ def generate_video(text, auth_openai, auth_elevenlabs, auth_replicate, auth_rev)
     generated_story = story.text2story(text, auth_openai)
     narrator.text2voice(generated_story, "audio_out.mp3", auth_elevenlabs, 5)
     deforum_str, max_frames = subtitles.audio2subtitle(auth_rev)
-    generated_animation = animation.story2video(deforum_str, max_frames, auth_replicate)
+    animation.story2video(deforum_str, max_frames, auth_replicate)
     # generated_music = bgm.text2audio(text=text, duration=20, guidance_scale=5, random_seed=24, n_candidates=3)
-    generated_video = join.join_artifacts()
 
+    generated_video = join.join_artifacts()
     return generated_video
 
-def download_video(v):
+def download_video(video_path):
     pass
 
 with gr.Blocks() as demo:
@@ -37,6 +37,6 @@ with gr.Blocks() as demo:
             download_button = gr.Button("Download")
 
     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)
+    download_button.click(download_video, inputs=[video_out])
 
 demo.launch(debug=True, enable_queue=True)

join.py

@@ -1,6 +1,9 @@
 import ffmpeg
+import os
 
 def join_artifacts():
     audio_in = ffmpeg.input('audio_out.mp3')
     video_in = ffmpeg.input('video_out.mp4')
-    ffmpeg.concat(video_in.filter("subtitles", "audio_out.srt"), audio_in, v=1, a=1).output('av_out.mp4').run()
+    ffmpeg.concat(video_in.filter("subtitles", "audio_out.srt"), audio_in, v=1, a=1).output('av_out.mp4').run()
+    av_path = os.path.join(os.path.dirname(__file__), "av_out.mp4")
+    return av_path