From 40a88f362198b9aef0d9f93f939df370a28a66b8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Victor=20SDV?= Date: Sat, 15 Oct 2022 13:32:13 -0300 Subject: [PATCH] get the first image(.png) with any name --- video_creation/final_video.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/video_creation/final_video.py b/video_creation/final_video.py index 0c88bc7..6261ade 100755 --- a/video_creation/final_video.py +++ b/video_creation/final_video.py @@ -5,6 +5,7 @@ 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 @@ -21,7 +22,6 @@ 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 PIL import Image console = Console() W, H = 1080, 1920 @@ -144,16 +144,27 @@ def make_final_video( 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 exists(f"assets/backgrounds/thumbnail.png"): + 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/thumbnail.png") + 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("Thumbnail - Building Thumbnail in 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)