File size: 1,732 Bytes
cf72e88 |
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 |
import multiprocessing
import os
import subprocess
import random
from utils import *
HOME = Path.cwd()
def prepare_background(background_mp4: str, filename_mp3: str, filename_srt: str, verbose: bool = False) -> str:
video_info = get_info(background_mp4, kind='video')
video_duration = int(round(video_info.get('duration'), 0))
audio_info = get_info(filename_mp3, kind='audio')
audio_duration = int(round(audio_info.get('duration'), 0))
ss = random.randint(0, (video_duration-audio_duration))
audio_duration = convert_time(audio_duration)
if ss < 0:
ss = 0
srt_raw = filename_srt
srt_filename = filename_srt.name
srt_path = filename_srt.parent.absolute()
directory = HOME / 'output'
if not directory.exists():
directory.mkdir()
outfile = f"{HOME}{os.sep}output{os.sep}output_{ss}.mp4"
if verbose:
rich_print(
f"{filename_srt = }\n{background_mp4 = }\n{filename_mp3 = }\n", style='bold green')
args = [
"ffmpeg",
"-ss", str(ss),
"-t", str(audio_duration),
"-i", background_mp4,
"-i", filename_mp3,
"-map", "0:v",
"-map", "1:a",
"-vf", f"crop=ih/16*9:ih, scale=w=1080:h=1920:flags=lanczos, gblur=sigma=2, ass='{srt_raw.absolute()}'",
"-c:v", "libx264",
"-crf", "23",
"-c:a", "aac",
"-ac", "2",
"-b:a", "192K",
f"{outfile}",
"-y",
"-threads", f"{multiprocessing.cpu_count()}"]
if verbose:
rich_print('[i] FFMPEG Command:\n'+' '.join(args)+'\n', style='yellow')
with KeepDir() as keep_dir:
keep_dir.chdir(srt_path)
subprocess.run(args, check=True)
return outfile
|