From 9b25603bba6b5d4b3b426655b2f583bd8fc1d705 Mon Sep 17 00:00:00 2001
From: null3000 <76852813+null3000@users.noreply.github.com>
Date: Tue, 7 Jun 2022 01:52:44 +0200
Subject: [PATCH] Update voices.py

---
 video_creation/voices.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/video_creation/voices.py b/video_creation/voices.py
index c0df8b7..9ac6aac 100644
--- a/video_creation/voices.py
+++ b/video_creation/voices.py
@@ -3,7 +3,8 @@ from pathlib import Path
 from mutagen.mp3 import MP3
 from utils.console import print_step, print_substep
 from rich.progress import track
-
+from rich.console import Console
+console = Console()
 
 def save_text_to_mp3(reddit_obj):
     """Saves Text to MP3 files.
@@ -31,7 +32,11 @@ def save_text_to_mp3(reddit_obj):
         tts.save(f"assets/mp3/posttext.mp3")
         length += MP3(f"assets/mp3/posttext.mp3").info.length
 
-    for idx, comment in track(enumerate(reddit_obj["comments"]), "Saving..."):
+    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} ")
+
         # ! 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
@@ -39,6 +44,7 @@ def save_text_to_mp3(reddit_obj):
         tts.save(f"assets/mp3/{idx}.mp3")
         length += MP3(f"assets/mp3/{idx}.mp3").info.length
 
-    print_substep("Saved Text to MP3 files successfully.", style="bold green")
+    #let user know that the MP3 files are saved
+    console.log(f"[bold green]Saved {idx + 1} MP3 Files.")
     # ! Return the index so we know how many screenshots of comments we need to make.
     return length, idx