Update main.py

main.py

@@ -358,6 +358,87 @@ async def make_video(request: Request):
         print(traceback.format_exc())
         raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
 
+@app.post("/concatenate_videos_with_music")
+async def concatenate_videos_with_music(request: Request):
+    try:
+        # Generate a unique filename for the output
+        output_filename = f"{uuid.uuid4()}.mp4"
+        output_path = os.path.join(OUTPUT_DIR, output_filename)
+        temp_dir = os.path.join(tempfile.gettempdir(), generate_hash())
+        os.makedirs(temp_dir, exist_ok=True)
+
+        data = await request.json()
+        video_urls = data.get("video_urls")
+        music_url = data.get("music_url")
+        volume_adjustment = data.get("volume_adjustment", 1.0)  # Default volume if not specified
+
+        if not video_urls or not isinstance(video_urls, list):
+            raise HTTPException(status_code=400, detail="Invalid video_urls in request. Must be a list of URLs.")
+
+        if not music_url:
+            raise HTTPException(status_code=400, detail="Missing music_url in request.")
+
+        # Download the video files
+        video_files = []
+        for i, url in enumerate(video_urls):
+            video_file = await download_file(url, f"_{i}.mp4")
+            video_files.append(video_file)
+
+        # Create a temporary file with the list of input files
+        concat_list_path = os.path.join(temp_dir, "concat_list.txt")
+        with open(concat_list_path, "w") as f:
+            for file in video_files:
+                f.write(f"file '{file}'\n")
+
+        # First concatenate all videos
+        concat_output = os.path.join(temp_dir, "concat_output.mp4")
+        concat_cmd = f"ffmpeg -f concat -safe 0 -i {concat_list_path} -c:v copy -an {concat_output}"
+        process = await asyncio.create_subprocess_shell(
+            concat_cmd,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE
+        )
+        stdout, stderr = await process.communicate()
+
+        if process.returncode != 0:
+            print(f"FFmpeg concat error: {stderr.decode()}")
+            raise HTTPException(status_code=500, detail=f"FFmpeg concat failed: {stderr.decode()}")
+
+        # Download music file
+        music_file = await download_file(music_url, ".mp3")
+
+        # Add music to the concatenated video
+        final_cmd = (
+            f'ffmpeg -i {concat_output} -i {music_file} -map 0:v -map 1:a '
+            f'-c:v copy -c:a aac -shortest -af "volume={volume_adjustment}" {output_path}'
+        )
+        process = await asyncio.create_subprocess_shell(
+            final_cmd,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE
+        )
+        stdout, stderr = await process.communicate()
+
+        if process.returncode != 0:
+            print(f"FFmpeg music mix error: {stderr.decode()}")
+            raise HTTPException(status_code=500, detail=f"FFmpeg music mixing failed: {stderr.decode()}")
+
+        # Clean up temporary files
+        for file in video_files:
+            os.remove(file)
+        os.remove(concat_list_path)
+        os.remove(concat_output)
+        os.remove(music_file)
+        os.rmdir(temp_dir)
+
+        # Return the URL path to the output file
+        return f"/output/{output_filename}"
+
+    except Exception as e:
+        print(f"An error occurred: {str(e)}")
+        print(traceback.format_exc())
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)