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.
84 lines
2.9 KiB
84 lines
2.9 KiB
#!/usr/bin/env python3
|
|
from os import getenv
|
|
from pathlib import Path
|
|
|
|
import sox
|
|
from mutagen import MutagenError
|
|
from mutagen.mp3 import MP3, HeaderNotFoundError
|
|
from rich.console import Console
|
|
from rich.progress import track
|
|
|
|
from TTS.swapper import TTS
|
|
|
|
from utils.console import print_step, print_substep
|
|
from utils.voice import sanitize_text
|
|
|
|
console = Console()
|
|
|
|
|
|
VIDEO_LENGTH: int = 40 # secs
|
|
|
|
|
|
def save_text_to_mp3(reddit_obj):
|
|
"""Saves text to MP3 files. Goes through the reddit_obj and generates the title MP3 file and a certain number of comments until the total amount of time exceeds VIDEO_LENGTH seconds.
|
|
|
|
Args:
|
|
reddit_obj (dict[str]): Reddit object received from reddit API in askreddit.py
|
|
|
|
Returns:
|
|
tuple[int,int]: (total length of the audio, the number of comments audio was generated for)
|
|
"""
|
|
print_step("Saving Text to MP3 files...")
|
|
length = 0
|
|
|
|
# Create a folder for the mp3 files.
|
|
Path("assets/temp/mp3").mkdir(parents=True, exist_ok=True)
|
|
TextToSpeech = TTS()
|
|
TextToSpeech.tts(
|
|
sanitize_text(reddit_obj["thread_title"]),
|
|
filename="assets/temp/mp3/title.mp3",
|
|
random_speaker=False,
|
|
)
|
|
try:
|
|
length += MP3("assets/temp/mp3/title.mp3").info.length
|
|
except HeaderNotFoundError: # note to self AudioFileClip
|
|
length += sox.file_info.duration("assets/temp/mp3/title.mp3")
|
|
if getenv("STORYMODE").casefold() == "true":
|
|
TextToSpeech.tts(
|
|
sanitize_text(reddit_obj["thread_content"]),
|
|
filename="assets/temp/mp3/story_content.mp3",
|
|
random_speaker=False,
|
|
)
|
|
# 'story_content'
|
|
com = 0
|
|
for comment in track((reddit_obj["comments"]), "Saving..."):
|
|
# ! Stop creating mp3 files if the length is greater than VIDEO_LENGTH seconds. This can be longer
|
|
# but this is just a good_voices starting point
|
|
if length > VIDEO_LENGTH:
|
|
break
|
|
|
|
TextToSpeech.tts(
|
|
sanitize_text(comment["comment_body"]),
|
|
filename=f"assets/temp/mp3/{com}.mp3",
|
|
random_speaker=False,
|
|
)
|
|
try:
|
|
length += MP3(f"assets/temp/mp3/{com}.mp3").info.length
|
|
com += 1
|
|
except (HeaderNotFoundError, MutagenError, Exception):
|
|
try:
|
|
length += sox.file_info.duration(f"assets/temp/mp3/{com}.mp3")
|
|
com += 1
|
|
except (OSError, IOError):
|
|
print(
|
|
"would have removed"
|
|
f"assets/temp/mp3/{com}.mp3"
|
|
f"assets/temp/png/comment_{com}.png"
|
|
)
|
|
# remove(f"assets/temp/mp3/{com}.mp3")
|
|
# remove(f"assets/temp/png/comment_{com}.png")# todo might cause odd un-syncing
|
|
|
|
print_substep("Saved Text to MP3 files Successfully.", style="bold green")
|
|
# ! Return the index, so we know how many screenshots of comments we need to make.
|
|
return length, com
|