Update modules/app.py

modules/app.py

@@ -72,48 +72,58 @@ async def download_file(url: str, suffix: str):
 
 @app.post("/add_audio_to_image")
 async def add_audio_to_image(input_data: AudioImageInput):
-    # Download image and audio files
-    temp_image_path = await download_file(str(input_data.image_url), ".png")
-    temp_audio_path = await download_file(str(input_data.audio_url), ".mp3")
+    try:
+        # Download image and audio files
+        temp_image_path = await download_file(str(input_data.image_url), ".png")
+        temp_audio_path = await download_file(str(input_data.audio_url), ".mp3")
 
-    # Generate a unique filename
-    output_filename = f"{uuid.uuid4()}.mp4"
-    output_path = os.path.join(OUTPUT_DIR, output_filename)
+        # Generate a unique filename
+        output_filename = f"{uuid.uuid4()}.mp4"
+        output_path = os.path.join(OUTPUT_DIR, output_filename)
 
-    # Use ffmpeg to combine image and audio into a video
-    input_image = ffmpeg.input(temp_image_path, loop=1, t=5)  # 5 seconds duration
-    input_audio = ffmpeg.input(temp_audio_path)
-    ffmpeg.concat(input_image, input_audio, v=1, a=1).output(output_path, vcodec='libx264', acodec='aac').run()
+        # Use ffmpeg to combine image and audio into a video
+        input_image = ffmpeg.input(temp_image_path, loop=1, t=5)  # 5 seconds duration
+        input_audio = ffmpeg.input(temp_audio_path)
+        ffmpeg.concat(input_image, input_audio, v=1, a=1).output(output_path, vcodec='libx264', acodec='aac').run()
 
-    # Clean up temporary files
-    os.unlink(temp_image_path)
-    os.unlink(temp_audio_path)
+        # 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}"
+        # 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)}")
 
 @app.post("/concatenate_videos")
 async def concatenate_videos(input_data: VideosInput):
-    temp_video_paths = []
+    try:
+        temp_video_paths = []
 
-    # Download videos to temporary files
-    for video_url in input_data.video_urls:
-        temp_video_paths.append(await download_file(str(video_url), ".mp4"))
+        # Download videos to temporary files
+        for video_url in input_data.video_urls:
+            temp_video_paths.append(await download_file(str(video_url), ".mp4"))
 
-    # Generate a unique filename
-    output_filename = f"{uuid.uuid4()}.mp4"
-    output_path = os.path.join(OUTPUT_DIR, output_filename)
+        # Generate a unique filename
+        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]
-    ffmpeg.concat(*inputs, v=1, a=1).output(output_path, vcodec='libx264', acodec='aac').run()
+        # Concatenate video clips using ffmpeg
+        inputs = [ffmpeg.input(path) for path in temp_video_paths]
+        ffmpeg.concat(*inputs, v=1, a=1).output(output_path, vcodec='libx264', acodec='aac').run()
 
-    # Clean up temporary files
-    for path in temp_video_paths:
-        os.unlink(path)
+        # 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}"
+        # 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)}")
 
 @app.get("/get_audio")
 async def get_audio(url: str):