Update modules/app.py

modules/app.py

@@ -2,6 +2,11 @@ from fastapi import FastAPI, HTTPException, Request, Depends
 from fastapi.staticfiles import StaticFiles
 from pytubefix import YouTube
 from pytubefix.exceptions import PytubeFixError
+from moviepy.editor import VideoFileClip, concatenate_videoclips, CompositeVideoClip, AudioFileClip, ImageClip
+from pydantic import BaseModel, HttpUrl
+from typing import List
+import requests
+import tempfile
 import time
 import uuid
 import aiohttp
@@ -39,9 +44,88 @@ async def read_root():
 AUDIO_DIR = "audio_files"
 os.makedirs(AUDIO_DIR, exist_ok=True)
 
+# Create a directory for storing output files
+OUTPUT_DIR = "output"
+os.makedirs(OUTPUT_DIR, exist_ok=True)
+
 # Mount the audio directory
 app.mount("/audio", StaticFiles(directory=AUDIO_DIR), name="audio")
 
+# Mount the output directory
+app.mount("/output", StaticFiles(directory=OUTPUT_DIR), name="output")
+
+class AudioImageInput(BaseModel):
+    image_url: HttpUrl
+    audio_url: HttpUrl
+
+class VideosInput(BaseModel):
+    video_urls: List[HttpUrl]
+
+def download_file(url: str, suffix: str):
+    response = requests.get(url)
+    if response.status_code != 200:
+        raise HTTPException(status_code=400, detail=f"Failed to download file from {url}")
+    with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
+        temp_file.write(response.content)
+        return temp_file.name
+
+@app.post("/add_audio_to_image/")
+async def add_audio_to_image(input_data: AudioImageInput):
+    # Download image and audio files
+    temp_image_path = download_file(str(input_data.image_url), ".png")
+    temp_audio_path = download_file(str(input_data.audio_url), ".mp3")
+
+    # Create a video from the image
+    image_clip = ImageClip(temp_image_path).set_duration(5)  # 5 seconds duration
+
+    # Load the audio
+    audio_clip = AudioFileClip(temp_audio_path)
+
+    # Set the audio of the video clip
+    video_with_audio = image_clip.set_audio(audio_clip)
+
+    # Generate a unique filename
+    output_filename = f"{uuid.uuid4()}.mp4"
+    output_path = os.path.join(OUTPUT_DIR, output_filename)
+
+    # Write the result to a file
+    video_with_audio.write_videofile(output_path, codec="libx264", audio_codec="aac")
+
+    # Clean up temporary files
+    os.unlink(temp_image_path)
+    os.unlink(temp_audio_path)
+
+    # Return the URL path to the output file
+    return f"/output/{output_filename}"
+
+@app.post("/concatenate_videos/")
+async def concatenate_videos(input_data: VideosInput):
+    temp_video_paths = []
+
+    # Download videos to temporary files
+    for video_url in input_data.video_urls:
+        temp_video_paths.append(download_file(str(video_url), ".mp4"))
+
+    # Load video clips
+    video_clips = [VideoFileClip(path) for path in temp_video_paths]
+
+    # Concatenate video clips
+    final_clip = concatenate_videoclips(video_clips)
+
+    # Generate a unique filename
+    output_filename = f"{uuid.uuid4()}.mp4"
+    output_path = os.path.join(OUTPUT_DIR, output_filename)
+
+    # Write the result to a file
+    final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac")
+
+    # Clean up temporary files
+    for path in temp_video_paths:
+        os.unlink(path)
+
+    # Return the URL path to the output file
+    return f"/output/{output_filename}"
+
 @app.get("/get_audio")
 async def get_audio(url: str):    
     if not url:
@@ -75,7 +159,7 @@ async def get_audio(url: str):
         else:
             os.remove(out_file)
             raise HTTPException(status_code=413, detail="Audio file is too large. Limited to about 1.5 hours.")
-    except PytubeError as e:
+    except PytubeFixError as e:
         print(f"PytubeError occurred: {str(e)}")
         raise HTTPException(status_code=400, detail=f"Error processing YouTube video: {str(e)}")
     except HTTPException as he: