#!/usr/bin/env python3 import multiprocessing import os import re from os.path import exists import shutil from typing import Tuple, Any from PIL import Image from moviepy.audio.AudioClip import concatenate_audioclips, CompositeAudioClip from moviepy.audio.io.AudioFileClip import AudioFileClip from moviepy.video.VideoClip import ImageClip from moviepy.video.compositing.CompositeVideoClip import CompositeVideoClip from moviepy.video.compositing.concatenate import concatenate_videoclips from moviepy.video.io.VideoFileClip import VideoFileClip 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.thumbnail import create_thumbnail from utils import settings from utils.thumbnail import create_thumbnail console = Console() W, H = 1080, 1920 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"\/", r"", name) name[:30] 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 else: return name def make_final_video( number_of_clips: int, length: int, reddit_obj: dict, background_config: Tuple[str, str, str, Any], ): """Gathers audio clips, gathers all screenshots, stitches them together and saves the final video to assets/temp Args: number_of_clips (int): Index to end at when going through the screenshots' length (int): Length of the video reddit_obj (dict): The reddit object that contains the posts to read. background_config (Tuple[str, str, str, Any]): The background config to use. """ # try: # if it isn't found (i.e you just updated and copied over config.toml) it will throw an error # VOLUME_MULTIPLIER = settings.config["settings"]['background']["background_audio_volume"] # except (TypeError, KeyError): # print('No background audio volume found in config.toml. Using default value of 1.') # VOLUME_MULTIPLIER = 1 id = re.sub(r"[^\w\s-]", "", reddit_obj["thread_id"]) 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"] transition = settings.config["settings"]["transition"] background_clip = ( VideoFileClip(f"assets/temp/{id}/background.mp4") .without_audio() .resize(height=H) .crop(x1=1166.6, y1=0, x2=2246.6, y2=1920) ) # Gather all audio clips if settings.config["settings"]["storymode"]: if settings.config["settings"]["storymodemethod"] == 0: audio_clips = [AudioFileClip(f"assets/temp/{id}/mp3/title.mp3")] audio_clips.insert(1, AudioFileClip(f"assets/temp/{id}/mp3/postaudio.mp3")) elif settings.config["settings"]["storymodemethod"] == 1: audio_clips = [ AudioFileClip(f"assets/temp/{id}/mp3/postaudio-{i}.mp3") for i in track( range(number_of_clips + 1), "Collecting the audio files..." ) ] audio_clips.insert(0, AudioFileClip(f"assets/temp/{id}/mp3/title.mp3")) else: audio_clips = [ AudioFileClip(f"assets/temp/{id}/mp3/{i}.mp3") for i in range(number_of_clips) ] audio_clips.insert(0, AudioFileClip(f"assets/temp/{id}/mp3/title.mp3")) audio_concat = concatenate_audioclips(audio_clips) audio_composite = CompositeAudioClip([audio_concat]) console.log(f"[bold green] Video Will Be: {length} Seconds Long") # add title to video image_clips = [] # Gather all images new_opacity = 1 if opacity is None or float(opacity) >= 1 else float(opacity) new_transition = ( 0 if transition is None or float(transition) > 2 else float(transition) ) image_clips.insert( 0, ImageClip(f"assets/temp/{id}/png/title.png") .set_duration(audio_clips[0].duration) .resize(width=W - 100) .set_opacity(new_opacity) .crossfadein(new_transition) .crossfadeout(new_transition), ) if settings.config["settings"]["storymode"]: if settings.config["settings"]["storymodemethod"] == 0: image_clips.insert( 1, ImageClip(f"assets/temp/{id}/png/story_content.png") .set_duration(audio_clips[1].duration) .set_position("center") .resize(width=W - 100) .set_opacity(float(opacity)), ) elif settings.config["settings"]["storymodemethod"] == 1: for i in track( range(0, number_of_clips + 1), "Collecting the image files..." ): image_clips.append( ImageClip(f"assets/temp/{id}/png/img{i}.png") .set_duration(audio_clips[i + 1].duration) .resize(width=W - 100) .set_opacity(new_opacity) # .crossfadein(new_transition) # .crossfadeout(new_transition) ) else: for i in range(0, number_of_clips): image_clips.append( ImageClip(f"assets/temp/{id}/png/comment_{i}.png") .set_duration(audio_clips[i + 1].duration) .resize(width=W - 100) .set_opacity(new_opacity) .crossfadein(new_transition) .crossfadeout(new_transition) ) img_clip_pos = background_config[3] image_concat = concatenate_videoclips(image_clips).set_position( img_clip_pos ) # note transition kwarg for delay in imgs image_concat.audio = audio_composite final = CompositeVideoClip([background_clip, image_concat]) title = re.sub(r"[^\w\s-]", "", reddit_obj["thread_title"]) idx = re.sub(r"[^\w\s-]", "", reddit_obj["thread_id"]) title_thumb = reddit_obj["thread_title"] filename = f"{name_normalize(title)[:251]}" subreddit = 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}") # create a tumbnail for the video settingsbackground = settings.config["settings"]["background"] if settingsbackground["background_thumbnail"]: if not exists(f"./results/{subreddit}/thumbnails"): print_substep( "The results/thumbnails folder didn't exist so I made it") os.makedirs(f"./results/{subreddit}/thumbnails") # get the first file with the .png extension from assets/backgrounds and use it as a background for the thumbnail first_image = next( ( file for file in os.listdir("assets/backgrounds") if file.endswith(".png") ), None, ) if first_image is None: print_substep("No png files found in assets/backgrounds", "red") if settingsbackground["background_thumbnail"] and first_image: font_family = settingsbackground["background_thumbnail_font_family"] font_size = settingsbackground["background_thumbnail_font_size"] font_color = settingsbackground["background_thumbnail_font_color"] thumbnail = Image.open(f"assets/backgrounds/{first_image}") width, height = thumbnail.size thumbnailSave = create_thumbnail(thumbnail, font_family, font_size, font_color, width, height, title_thumb) thumbnailSave.save(f"./assets/temp/{id}/thumbnail.png") print_substep(f"Thumbnail - Building Thumbnail in assets/temp/{id}/thumbnail.png") # create a tumbnail for the video settingsbackground = settings.config["settings"]["background"] if settingsbackground["background_thumbnail"]: if not exists(f"./results/{subreddit}/thumbnails"): print_substep( "The results/thumbnails folder didn't exist so I made it") os.makedirs(f"./results/{subreddit}/thumbnails") # get the first file with the .png extension from assets/backgrounds and use it as a background for the thumbnail first_image = next( ( file for file in os.listdir("assets/backgrounds") if file.endswith(".png") ), None, ) if first_image is None: print_substep("No png files found in assets/backgrounds", "red") if settingsbackground["background_thumbnail"] and first_image: font_family = settingsbackground["background_thumbnail_font_family"] font_size = settingsbackground["background_thumbnail_font_size"] font_color = settingsbackground["background_thumbnail_font_color"] thumbnail = Image.open(f"assets/backgrounds/{first_image}") width, height = thumbnail.size thumbnailSave = create_thumbnail(thumbnail, font_family, font_size, font_color, width, height, title_thumb) thumbnailSave.save(f"./assets/temp/{id}/thumbnail.png") print_substep(f"Thumbnail - Building Thumbnail in assets/temp/{id}/thumbnail.png") # 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) # if final = Video(final).add_watermark( text=f"Background credit: {background_config[2]}", opacity=0.4, redditid=reddit_obj ) final.write_videofile( f"assets/temp/{id}/temp.mp4", fps=int(settings.config["settings"]["fps"]), audio_codec="aac", audio_bitrate="192k", verbose=False, threads=multiprocessing.cpu_count(), ) ffmpeg_extract_subclip( f"assets/temp/{id}/temp.mp4", 0, length, targetname=f"results/{subreddit}/{filename}.mp4", ) #get the thumbnail image from assets/temp/id/thumbnail.png and save it in results/subreddit/thumbnails if settingsbackground["background_thumbnail"] and exists(f"assets/temp/{id}/thumbnail.png"): shutil.move(f"assets/temp/{id}/thumbnail.png", f"./results/{subreddit}/thumbnails/{filename}.png") save_data(subreddit, filename+".mp4", title, idx, background_config[2]) print_step("Removing temporary files 🗑") cleanups = cleanup(id) 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]}' )