make concatenation happen on clips that were generated at the execution instead of the entire results folder

pull/1694/head
sajawalhassan 2 years ago
parent 0f83a0ca1e
commit 640763e387

@ -37,15 +37,17 @@ class VideoGenerator():
if os.path.exists("long_form_videos_results"): if os.path.exists("long_form_videos_results"):
shutil.rmtree("long_form_videos_results") shutil.rmtree("long_form_videos_results")
subreddit = self.config["reddit"]["thread"]["subreddit"]
videos_folder = f"results/{subreddit}"
videos_to_be_conc = []
for i in range(num_of_videos_to_be_conc): for i in range(num_of_videos_to_be_conc):
print_markdown(f"# Iteration {i+1}") print_markdown(f"# Iteration {i+1}")
run(self.config) run(self.config)
videos_to_be_conc.append(os.listdir(videos_folder)[-1])
print_markdown("# Concatinating videos...") print_markdown("# Concatinating videos...")
videos = [] videos = []
subreddit = self.config["reddit"]["thread"]["subreddit"] for video in videos_to_be_conc:
videos_folder = f"results/{subreddit}"
for video in os.listdir(videos_folder):
videos.append(os.path.join(videos_folder, video)) videos.append(os.path.join(videos_folder, video))
if not os.path.exists("long_form_videos_results"): if not os.path.exists("long_form_videos_results"):

Loading…
Cancel
Save