Update modules/app.py

modules/app.py

@@ -104,22 +104,29 @@ async def concatenate_videos(input_data: VideosInput):
 
         # Download videos to temporary files
         for video_url in input_data.video_urls:
-            temp_video_paths.append(await download_file(str(video_url), ".mp4"))
+            temp_video_paths.append(await download_file(video_url, ".mp4"))
 
-        # Generate a unique filename
+        # Generate a unique filename for the output
         output_filename = f"{uuid.uuid4()}.mp4"
         output_path = os.path.join(OUTPUT_DIR, output_filename)
 
-        # Concatenate video clips using ffmpeg
-        inputs = [ffmpeg.input(path) for path in temp_video_paths]
-        concat_input = ffmpeg.concat(*inputs, v=1, a=1).node
-        video = concat_input[0]
-        audio = concat_input[1]
-        ffmpeg.output(video, audio, output_path, vcodec='libx264', acodec='aac').run()
+        # Separate video and audio streams
+        video_and_audio_streams = []
+        for path in temp_video_paths:
+            video = ffmpeg.input(path).video
+            audio = ffmpeg.input(path).audio
+            video_and_audio_streams.append(video)
+            video_and_audio_streams.append(audio)
+
+        # Concatenate video and audio streams
+        joined = ffmpeg.concat(*video_and_audio_streams, v=1, a=1).node
+
+        # Merge video and audio
+        ffmpeg.output(joined[0], joined[1], output_path, vcodec='libx264', acodec='aac').run()
 
         # Clean up temporary files
         for path in temp_video_paths:
-            os.unlink(path)
+            os.remove(path)
 
         # Return the URL path to the output file
         return f"/output/{output_filename}"