update bugfixes
This commit is contained in:
@@ -161,7 +161,6 @@ class TTS:
|
||||
DEVICE = None
|
||||
DEFAULTS = {
|
||||
"language": "Chinese",
|
||||
"speaker": "Uncle_Fu",
|
||||
"instruct": "语速缓慢而审慎,每个音节的语调都拿捏得恰到好处,宛如教授在指导新生。",
|
||||
"instruct": "Speak in a slow pace and enuntiate every word, as a teacher to a learning student",
|
||||
}
|
||||
@@ -189,11 +188,13 @@ class TTS:
|
||||
@staticmethod
|
||||
def generate(text: str):
|
||||
"""Generates a Waw using the defaulst values"""
|
||||
print(f"starting to generate {text}")
|
||||
speaker = random.choice(TTS.VOICES)
|
||||
return TTS.MODEL.generate_custom_voice(
|
||||
audio = TTS.MODEL.generate_custom_voice(
|
||||
text=text, **TTS.DEFAULTS, speaker=speaker
|
||||
)
|
||||
|
||||
print(f"finish to generate {text}")
|
||||
return audio
|
||||
|
||||
# Clases
|
||||
|
||||
@@ -207,6 +208,7 @@ class ProcessFolder:
|
||||
def __init__(self, input_folder: Path, language_id: str = None):
|
||||
self.input_folder = input_folder
|
||||
self._language_id = language_id
|
||||
self.absolute_input_folder = INPUT / self.input_folder
|
||||
# process file type
|
||||
self.out_folder = OUTPUT / input_folder
|
||||
self.out_folder.mkdir(parents=True, exist_ok=True)
|
||||
@@ -214,24 +216,23 @@ class ProcessFolder:
|
||||
self.resources.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
@property
|
||||
def output_name():
|
||||
def output_name(self):
|
||||
"""Posible name for the output file, still missing the filetype"""
|
||||
if self.language_id is None:
|
||||
raise ValueError("Not a valid language selected")
|
||||
return self.out_folder / f"{self.input_file.stem}.{self.language_id}.temp"
|
||||
|
||||
@property
|
||||
def absolute_input_file(self):
|
||||
"""Absolute input file"""
|
||||
return INPUT / self.input_file
|
||||
return self.out_folder / f"{self.input_folder.stem}.{self.language_id}.temp"
|
||||
|
||||
@property
|
||||
def input_files(self):
|
||||
input_files = []
|
||||
for file in self.absolute_input_file.glob(f"*.txt"):
|
||||
for file in self.absolute_input_folder.glob(f"*.txt"):
|
||||
for file_type in (DICT_TYPE, PHRASES_TYPE, DICTATION_TYPE):
|
||||
if file_type in file.suffixes:
|
||||
input_files.append(ProcessFile(file, language_id=self.language_id))
|
||||
input_files.append(
|
||||
ProcessFile(
|
||||
file.relative_to(INPUT), language_id=self.language_id
|
||||
)
|
||||
)
|
||||
return input_files
|
||||
|
||||
@property
|
||||
@@ -339,6 +340,7 @@ class DictionaryResult:
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
n: str,
|
||||
language_id: str,
|
||||
simplified: str,
|
||||
traditional: str,
|
||||
@@ -346,6 +348,7 @@ class DictionaryResult:
|
||||
meaning: str,
|
||||
audio_path: Path,
|
||||
):
|
||||
self.n = n
|
||||
self.language_id = language_id
|
||||
self.simplified = simplified
|
||||
self.traditional = traditional
|
||||
|
||||
Reference in New Issue
Block a user