#!/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...', ): 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 enumerate( indexes_for_videos, start=index_offset, ): 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).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]}' )