diff --git a/TTS/aws_polly.py b/TTS/aws_polly.py index 703aa6a..6cbe4c5 100644 --- a/TTS/aws_polly.py +++ b/TTS/aws_polly.py @@ -35,9 +35,9 @@ class AWSPolly: if random_voice: voice = self.randomvoice() else: - if not os.getenv("VOICE"): + if not os.getenv("AWS_VOICE"): return ValueError( - f"Please set the environment variable VOICE to a valid voice. options are: {voices}" + f"Please set the environment variable AWS_VOICE to a valid voice. options are: {voices}" ) voice = str(os.getenv("AWS_VOICE")).capitalize() try: diff --git a/TTS/engine_wrapper.py b/TTS/engine_wrapper.py index 77f6eab..13ff850 100644 --- a/TTS/engine_wrapper.py +++ b/TTS/engine_wrapper.py @@ -72,7 +72,7 @@ class TTSEngine: print_substep("Saved Text to MP3 files successfully.", style="bold green") return self.length, idx - def split_post(self, text: str, idx: int) -> str: + def split_post(self, text: str, idx: int): split_files = [] split_text = [ x.group().strip() @@ -88,9 +88,15 @@ class TTSEngine: f"{self.path}/{idx}.mp3", fps=44100, verbose=False, logger=None ) - for i in range(0, idy + 1): + for i in split_files: + name = i.filename + i.close() + Path(name).unlink() + + # for i in range(0, idy + 1): # print(f"Cleaning up {self.path}/{idx}-{i}.part.mp3") - Path(f"{self.path}/{idx}-{i}.part.mp3").unlink() + + # Path(f"{self.path}/{idx}-{i}.part.mp3").unlink() def call_tts(self, filename: str, text: str): self.tts_module.run(text=process_text(text), filepath=f"{self.path}/{filename}.mp3") @@ -100,6 +106,7 @@ class TTSEngine: # self.length += sox.file_info.duration(f"{self.path}/{filename}.mp3") clip = AudioFileClip(f"{self.path}/{filename}.mp3") self.length += clip.duration + clip.close() def process_text(text: str): lang = getenv("POSTLANG", "") diff --git a/TTS/streamlabs_polly.py b/TTS/streamlabs_polly.py index 41fe269..066fa53 100644 --- a/TTS/streamlabs_polly.py +++ b/TTS/streamlabs_polly.py @@ -35,9 +35,9 @@ class StreamlabsPolly: if random_voice: voice = self.randomvoice() else: - if not os.getenv("VOICE"): + if not os.getenv("STREAMLABS_VOICE"): return ValueError( - f"Please set the environment variable VOICE to a valid voice. options are: {voices}" + f"Please set the environment variable STREAMLABS_VOICE to a valid voice. options are: {voices}" ) voice = str(os.getenv("STREAMLABS_VOICE")).capitalize() body = {"voice": voice, "text": text, "service": "polly"} diff --git a/video_creation/background.py b/video_creation/background.py index 7bf6ae2..2654499 100644 --- a/video_creation/background.py +++ b/video_creation/background.py @@ -2,6 +2,7 @@ import random from os import listdir, environ from pathlib import Path from random import randrange +from typing import Tuple from moviepy.editor import VideoFileClip from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip @@ -10,7 +11,7 @@ from pytube import YouTube from utils.console import print_step, print_substep -def get_start_and_end_times(video_length: int, length_of_clip: int) -> tuple[int, int]: +def get_start_and_end_times(video_length: int, length_of_clip: int) -> Tuple[int, int]: """Generates a random interval of time to be used as the background of the video. Args: