diff --git a/reddit/subreddit.py b/reddit/subreddit.py index 0171749..28f1a8d 100644 --- a/reddit/subreddit.py +++ b/reddit/subreddit.py @@ -1,6 +1,6 @@ from numpy import Infinity from rich.console import Console -from utils.console import print_step, print_substep, print_markdown +from utils.console import print_step, print_substep from dotenv import load_dotenv import os, random, praw, re diff --git a/video_creation/screenshot_downloader.py b/video_creation/screenshot_downloader.py index 88fcaa1..7e81630 100644 --- a/video_creation/screenshot_downloader.py +++ b/video_creation/screenshot_downloader.py @@ -49,8 +49,8 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num, theme): path="assets/png/title.png" ) - for idx, comment in track( - enumerate(reddit_object["comments"]) + for idx, comment in enumerate( + track(reddit_object["comments"]), "Downloading screenshots..." ): #allow user to see what comment is being saved diff --git a/video_creation/voices.py b/video_creation/voices.py index 6ae88ab..b3278b8 100644 --- a/video_creation/voices.py +++ b/video_creation/voices.py @@ -34,11 +34,7 @@ def save_text_to_mp3(reddit_obj): tts.save("assets/mp3/posttext.mp3") length += MP3("assets/mp3/posttext.mp3").info.length - for idx, comment in track(enumerate(reddit_obj["comments"])): - - #allow user to see what comment is being saved - print_substep(f"Saving MP3 {idx + 1} ") - + for idx, comment in enumerate(track(reddit_obj["comments"]), "Saving..."): # ! Stop creating mp3 files if the length is greater than 50 seconds. This can be longer, but this is just a good starting point if length > 50: break