Update voices.py

pull/354/head
null3000 3 years ago committed by GitHub
parent bafe8c3616
commit 9b25603bba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save