You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
293 lines
12 KiB
293 lines
12 KiB
import os
|
|
import re
|
|
import multiprocessing
|
|
from os.path import exists
|
|
from typing import Tuple, Any, Final
|
|
import translators as ts
|
|
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()
|
|
|
|
|
|
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)
|
|
|
|
lang = settings.config["reddit"]["thread"]["post_lang"]
|
|
if lang:
|
|
print_substep("Translating filename...")
|
|
translated_name = ts.google(name, to_language=lang)
|
|
return translated_name
|
|
else:
|
|
return name
|
|
|
|
|
|
def prepare_background(reddit_id: str, W: int, H: int) -> VideoFileClip:
|
|
clip = (
|
|
VideoFileClip(f"assets/temp/{reddit_id}/background.mp4")
|
|
.without_audio()
|
|
.resize(height=H)
|
|
)
|
|
|
|
# calculate the center of the background clip
|
|
c = clip.w // 2
|
|
|
|
# calculate the coordinates where to crop
|
|
half_w = W // 2
|
|
x1 = c - half_w
|
|
x2 = c + half_w
|
|
|
|
return clip.crop(x1=x1, y1=0, x2=x2, y2=H)
|
|
|
|
|
|
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.
|
|
"""
|
|
# settings values
|
|
W: Final[int] = int(settings.config["settings"]["resolution_w"])
|
|
H: Final[int] = int(settings.config["settings"]["resolution_h"])
|
|
|
|
# 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
|
|
|
|
reddit_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 = prepare_background(reddit_id, W=W, H=H)
|
|
|
|
# Gather all audio clips
|
|
if settings.config["settings"]["storymode"]:
|
|
if settings.config["settings"]["storymodemethod"] == 0:
|
|
audio_clips = [AudioFileClip(f"assets/temp/{reddit_id}/mp3/title.mp3")]
|
|
audio_clips.insert(1, AudioFileClip(f"assets/temp/{reddit_id}/mp3/postaudio.mp3"))
|
|
elif settings.config["settings"]["storymodemethod"] == 1:
|
|
audio_clips = [
|
|
AudioFileClip(f"assets/temp/{reddit_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/{reddit_id}/mp3/title.mp3"))
|
|
|
|
else:
|
|
audio_clips = [
|
|
AudioFileClip(f"assets/temp/{reddit_id}/mp3/{i}.mp3")
|
|
for i in range(number_of_clips)
|
|
]
|
|
audio_clips.insert(0, AudioFileClip(f"assets/temp/{reddit_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)
|
|
)
|
|
screenshot_width = int((W * 90) // 100)
|
|
image_clips.insert(
|
|
0,
|
|
ImageClip(f"assets/temp/{reddit_id}/png/title.png")
|
|
.set_duration(audio_clips[0].duration)
|
|
.resize(width=screenshot_width)
|
|
.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/{reddit_id}/png/story_content.png")
|
|
.set_duration(audio_clips[1].duration)
|
|
.set_position("center")
|
|
.resize(width=screenshot_width)
|
|
.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/{reddit_id}/png/img{i}.png")
|
|
.set_duration(audio_clips[i + 1].duration)
|
|
.resize(width=screenshot_width)
|
|
.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/{reddit_id}/png/comment_{i}.png")
|
|
.set_duration(audio_clips[i + 1].duration)
|
|
.resize(width=screenshot_width)
|
|
.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/{reddit_id}/thumbnail.png")
|
|
print_substep(f"Thumbnail - Building Thumbnail in assets/temp/{reddit_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/{reddit_id}/thumbnail.png")
|
|
print_substep(f"Thumbnail - Building Thumbnail in assets/temp/{reddit_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)
|
|
|
|
final = Video(final).add_watermark(
|
|
text=f"Background credit: {background_config[2]}",
|
|
opacity=0.4,
|
|
redditid=reddit_obj,
|
|
)
|
|
final.write_videofile(
|
|
f"assets/temp/{reddit_id}/temp.mp4",
|
|
fps=int(settings.config["settings"]["fps"]),
|
|
audio_codec="aac",
|
|
audio_bitrate="192k",
|
|
verbose=False,
|
|
threads=multiprocessing.cpu_count(),
|
|
#preset="ultrafast", # for testing purposes
|
|
)
|
|
ffmpeg_extract_subclip(
|
|
f"assets/temp/{reddit_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(reddit_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]}'
|
|
)
|