You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.7 KiB
51 lines
1.7 KiB
from gtts import gTTS
|
|
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.
|
|
|
|
Args:
|
|
reddit_obj : The reddit object you received from the reddit API in the askreddit.py file.
|
|
"""
|
|
print_step("Saving Text to MP3 files...")
|
|
length = 0
|
|
|
|
# Create a folder for the mp3 files.
|
|
Path("assets/mp3").mkdir(parents=True, exist_ok=True)
|
|
|
|
tts = gTTS(text=reddit_obj["thread_title"], lang="en", slow=False)
|
|
tts.save(f"assets/mp3/title.mp3")
|
|
length += MP3(f"assets/mp3/title.mp3").info.length
|
|
|
|
try:
|
|
Path(f"assets/mp3/posttext.mp3").unlink()
|
|
except OSError as e:
|
|
pass
|
|
|
|
if reddit_obj["thread_post"] != "":
|
|
tts = gTTS(text=reddit_obj["thread_post"], lang="en", slow=False)
|
|
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"])):
|
|
|
|
#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
|
|
tts = gTTS(text=comment["comment_body"], lang="en", slow=False)
|
|
tts.save(f"assets/mp3/{idx}.mp3")
|
|
length += MP3(f"assets/mp3/{idx}.mp3").info.length
|
|
|
|
#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
|