Spaces:
Runtime error
Runtime error
File size: 2,919 Bytes
9867d34 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
"""Media utilities for audio/video processing."""
import os
import subprocess
from pathlib import Path
from typing import Optional
from loguru import logger
class MediaProcessingError(Exception):
"""Exception raised for media processing errors."""
pass
def merge_audio_video(
audio_path: str,
video_path: str,
output_path: str,
overwrite: bool = True,
quality: str = "high"
) -> str:
"""
Merge audio and video files using ffmpeg.
Args:
audio_path: Path to input audio file
video_path: Path to input video file
output_path: Path for output video file
overwrite: Whether to overwrite existing output file
quality: Quality setting ('high', 'medium', 'low')
Returns:
Path to the output file
Raises:
MediaProcessingError: If input files don't exist or ffmpeg fails
FileNotFoundError: If ffmpeg is not installed
"""
# Validate input files
if not os.path.exists(audio_path):
raise MediaProcessingError(f"Audio file not found: {audio_path}")
if not os.path.exists(video_path):
raise MediaProcessingError(f"Video file not found: {video_path}")
# Create output directory if needed
output_dir = Path(output_path).parent
output_dir.mkdir(parents=True, exist_ok=True)
# Quality settings
quality_settings = {
"high": ["-b:a", "192k"],
"medium": ["-b:a", "128k"],
"low": ["-b:a", "96k"]
}
# Build ffmpeg command
ffmpeg_command = [
"ffmpeg",
"-i", video_path,
"-i", audio_path,
"-c:v", "copy",
"-c:a", "aac",
"-ac", "2",
"-af", "pan=stereo|c0=c0|c1=c0",
"-map", "0:v:0",
"-map", "1:a:0",
*quality_settings.get(quality, quality_settings["high"]),
]
if overwrite:
ffmpeg_command.append("-y")
ffmpeg_command.append(output_path)
try:
logger.info(f"Merging audio '{audio_path}' with video '{video_path}'")
process = subprocess.Popen(
ffmpeg_command,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
text=True
)
stdout, stderr = process.communicate()
if process.returncode != 0:
error_msg = f"FFmpeg failed with return code {process.returncode}: {stderr}"
logger.error(error_msg)
raise MediaProcessingError(error_msg)
else:
logger.info(f"Successfully merged video saved to: {output_path}")
except FileNotFoundError:
raise FileNotFoundError(
"ffmpeg not found. Please install ffmpeg: "
"https://ffmpeg.org/download.html"
)
except Exception as e:
raise MediaProcessingError(f"Unexpected error during media processing: {e}")
return output_path
|