File size: 2,996 Bytes
dfd1909 | 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 | from typing import Literal, Tuple
import os
import subprocess
import numpy as np
from moviepy.editor import VideoFileClip, AudioFileClip, ImageClip
try: from pydub import AudioSegment
except: print('pydub is not installed. Please install it using `pip install pydub`')
class UtilVideo:
@staticmethod
def extract_audio_from_video(video_path:str,
output_path:str = './tmp/tmp.wav',
) -> str:
video = VideoFileClip(video_path)
audio = video.audio
os.makedirs(os.path.dirname(output_path), exist_ok=True)
audio.write_audiofile(output_path, codec='pcm_s16le')
return output_path
@staticmethod
def attach_audio_to_video(video_path:str,
audio_path:str,
output_path:str,
fps:int=30,
video_duration_sec:float = None,
audio_codec:Literal['aac', 'pcm_s16le', 'pcm_s32le'] = 'aac',
) -> VideoFileClip:
os.makedirs(os.path.dirname(output_path), exist_ok=True)
video_clip = VideoFileClip(video_path).set_fps(fps)
if video_duration_sec is not None:
video_clip = video_clip.subclip(0, video_duration_sec)
video_clip = video_clip.set_audio(AudioFileClip(audio_path))
video_clip.write_videofile(
output_path,
audio=True,
audio_codec = audio_codec,
fps=fps,
verbose=False,
logger=None
)
return video_clip
@staticmethod
def attach_audio_to_img(image_path:str,
audio_path:str,
output_path:str = 'output.mkv',
audio_codec:Literal['aac', 'pcm_s16le', 'pcm_s32le'] = 'pcm_s32le',
audio_fps:int=44100,
video_size:Tuple[int,int]=(1920,1080),
module:Literal['moviepy', 'ffmpeg'] = 'moviepy'
):
if module == 'moviepy':
import PIL
PIL.Image.ANTIALIAS = PIL.Image.LANCZOS
audio = AudioFileClip(audio_path)
image_clip:ImageClip = ImageClip(image_path).set_duration(audio.duration).resize(newsize=video_size)
video = image_clip.set_audio(audio)
video.write_videofile(output_path,
codec='libx264',
audio_fps = audio_fps,
audio_codec=audio_codec,
fps=24)
elif module == 'ffmpeg':
subprocess.run([
'ffmpeg', '-loop', '1', '-i', image_path, '-i', audio_path,
'-vf', f'scale={video_size[0]}:{video_size[1]}', '-c:v', 'libx264',
'-c:a', 'aac', '-shortest', output_path
])
|