From 00413e8a2fa3c3f306ad3bba235395c4d751dde3 Mon Sep 17 00:00:00 2001 From: sajawalhassan Date: Mon, 12 Jun 2023 11:32:19 +0500 Subject: [PATCH] fix console output --- video_generator.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/video_generator.py b/video_generator.py index c2ff62b..f6e1659 100644 --- a/video_generator.py +++ b/video_generator.py @@ -1,5 +1,6 @@ import os, shutil from moviepy.editor import concatenate_videoclips, VideoFileClip +from utils.console import print_markdown from main import run class VideoGenerator(): @@ -21,22 +22,19 @@ class VideoGenerator(): if os.path.exists("results"): shutil.rmtree("results") - for _ in range(num_of_shorts): - print("="*50) + for i in range(num_of_shorts): + print_markdown(f"# Iteration {i}") run(self.config) print(f"Videos have been made!") def generate_long_video(self, num_of_videos_to_be_conc, remove_past_results=True): - if remove_past_results: - if os.path.exists("results"): - shutil.rmtree("results") - + if remove_past_results: if os.path.exists("long_form_videos_results"): shutil.rmtree("long_form_videos_results") - for _ in range(num_of_videos_to_be_conc): - print("="*50) + for i in range(num_of_videos_to_be_conc): + print_markdown(f"# Iteration {i}") run(self.config) print("Concatinating videos...")