#!/usr/bin/env python3 import multiprocessing import os import re from os.path import exists from typing import Tuple, Any, Union from moviepy.editor import ( VideoFileClip, AudioFileClip, ImageClip, CompositeAudioClip, CompositeVideoClip, ) from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip from rich.console import Console from rich.progress import track from utils.cleanup import cleanup from utils.console import print_step, print_substep from utils.video import Video from utils.videos import save_data from utils import settings from video_creation.background import download_background, chop_background_video console = Console() def name_normalize( name: str ) -> str: name = re.sub(r'[?\\"%*:|<>]', "", name) name = re.sub(r"( [w,W]\s?\/\s?[o,O,0])", r" without", name) name = re.sub(r"( [w,W]\s?\/)", r" with", name) name = re.sub(r"(\d+)\s?\/\s?(\d+)", r"\1 of \2", name) name = re.sub(r"(\w+)\s?\/\s?(\w+)", r"\1 or \2", name) name = re.sub(r"\/", "", name) # name[:30] # the hell this little guy does? commented until explained lang = settings.config["reddit"]["thread"]["post_lang"] if lang: import translators as ts print_substep("Translating filename...") translated_name = ts.google(name, to_language=lang) return translated_name return name def make_final_video( indexes_of_clips: list, reddit_obj: dict, background_config: Tuple[str, str, str, Any], ) -> None: """ Gathers audio clips, gathers all screenshots, stitches them together and saves the final video to assets/temp Args: indexes_of_clips (list): Indexes of voiced comments reddit_obj (dict): The reddit object that contains the posts to read. background_config (Tuple[str, str, str, Any]): The background config to use. """ W: int = int(settings.config["settings"]["video_width"]) H: int = int(settings.config["settings"]["video_height"]) if not W or not H: W, H = 1080, 1920 max_length: int = int(settings.config["settings"]["video_length"]) time_before_first_picture: float = settings.config["settings"]["time_before_first_picture"] time_before_tts: float = settings.config["settings"]["time_before_tts"] time_between_pictures: float = settings.config["settings"]["time_between_pictures"] delay_before_end: float = settings.config["settings"]["delay_before_end"] print_step("Creating the final video 🎥") VideoFileClip.reW = lambda clip: clip.resize(width=W) VideoFileClip.reH = lambda clip: clip.resize(width=H) opacity = settings.config["settings"]["opacity"] / 100 def create_audio_clip( clip_title: Union[str, int], clip_start: float, ) -> 'AudioFileClip': return ( AudioFileClip(f"assets/temp/mp3/{clip_title}.mp3") .set_start(clip_start) ) video_duration = 0 # Gather all audio clips audio_clips = list() correct_audio_offset = time_before_tts * 2 + time_between_pictures audio_title = create_audio_clip( "title", time_before_first_picture + time_before_tts, ) video_duration += audio_title.duration + time_before_first_picture + time_before_tts audio_clips.append(audio_title) indexes_for_videos = list() for audio_title in track( indexes_of_clips, description="Gathering audio clips...", total=indexes_of_clips.__len__() ): temp_audio_clip = create_audio_clip( audio_title, correct_audio_offset + video_duration, ) if video_duration + temp_audio_clip.duration + correct_audio_offset + delay_before_end <= max_length: video_duration += temp_audio_clip.duration + correct_audio_offset audio_clips.append(temp_audio_clip) indexes_for_videos.append(audio_title) video_duration += delay_before_end + time_before_tts # Can't use concatenate_audioclips here, it resets clips' start point audio_composite = CompositeAudioClip(audio_clips) console.log("[bold green] Video Will Be: %.2f Seconds Long" % video_duration) # Gather all images new_opacity = 1 if opacity is None or opacity >= 1 else opacity def create_image_clip( image_title: Union[str, int], audio_start: float, audio_duration: float, ) -> 'ImageClip': return ( ImageClip(f"assets/temp/png/{image_title}.png") .set_start(audio_start - time_before_tts) .set_duration(time_before_tts * 2 + audio_duration) .set_opacity(new_opacity) .resize(width=W - 100) ) # add title to video image_clips = list() # Accounting for title and other stuff if audio_clips index_offset = 1 image_clips.append( create_image_clip( "title", audio_clips[0].start, audio_clips[0].duration ) ) for idx, photo_idx in track( enumerate( indexes_for_videos, start=index_offset, ), description="Gathering audio clips...", total=indexes_for_videos[index_offset:].__len__() ): image_clips.append( create_image_clip( f"comment_{photo_idx}", audio_clips[idx].start, audio_clips[idx].duration ) ) # if os.path.exists("assets/mp3/posttext.mp3"): # image_clips.insert( # 0, # ImageClip("assets/png/title.png") # .set_duration(audio_clips[0].duration + audio_clips[1].duration) # .set_position("center") # .resize(width=W - 100) # .set_opacity(float(opacity)), # ) # else: story mode stuff # Can't use concatenate_videoclips here, it resets clips' start point image_concat = CompositeVideoClip(image_clips) image_concat.set_position(background_config[3]) download_background(background_config) chop_background_video(background_config, video_duration) background_clip = ( VideoFileClip("assets/temp/background.mp4") .set_start(0) .set_end(video_duration) .without_audio() .resize(height=H) ) back_video_width, back_video_height = background_clip.size # Fix for crop with vertical videos if back_video_width < H: background_clip = ( background_clip .resize(width=W) ) back_video_width, back_video_height = background_clip.size background_clip = background_clip.crop( x1=0, x2=back_video_width, y1=back_video_height / 2 - H / 2, y2=back_video_height / 2 + H / 2 ) else: background_clip = background_clip.crop( x1=back_video_width / 2 - W / 2, x2=back_video_width / 2 + W / 2, y1=0, y2=back_video_height ) final = CompositeVideoClip([background_clip, image_concat]) final.audio = audio_composite title = re.sub(r"[^\w\s-]", "", reddit_obj["thread_title"]) idx = re.sub(r"[^\w\s-]", "", reddit_obj["thread_id"]) filename = f"{name_normalize(title)}.mp4" subreddit = str(settings.config["reddit"]["thread"]["subreddit"]) if not exists(f"./results/{subreddit}"): print_substep("The results folder didn't exist so I made it") os.makedirs(f"./results/{subreddit}") # if settings.config["settings"]['background']["background_audio"] and exists(f"assets/backgrounds/background.mp3"): # audioclip = mpe.AudioFileClip(f"assets/backgrounds/background.mp3").set_duration(final.duration) # audioclip = audioclip.fx( volumex, 0.2) # final_audio = mpe.CompositeAudioClip([final.audio, audioclip]) # # lowered_audio = audio_background.multiply_volume( # todo get this to work # # VOLUME_MULTIPLIER) # lower volume by background_audio_volume, use with fx # final.set_audio(final_audio) final = Video(final).add_watermark( text=f"Background credit: {background_config[2]}", opacity=0.4 ) final.write_videofile( "assets/temp/temp.mp4", fps=30, audio_codec="aac", audio_bitrate="192k", verbose=False, threads=multiprocessing.cpu_count(), ) ffmpeg_extract_subclip( "assets/temp/temp.mp4", 0, video_duration, targetname=f"results/{subreddit}/{filename}", ) save_data(subreddit, filename, title, idx, background_config[2]) print_step("Removing temporary files 🗑") cleanups = cleanup() print_substep(f"Removed {cleanups} temporary files 🗑") print_substep("See result in the results folder!") print_step( f'Reddit title: {reddit_obj["thread_title"]} \n Background Credit: {background_config[2]}' )