update bugfixes

This commit is contained in:
Wolfang Torres
2026-06-27 16:29:36 +08:00
parent 7cb0894abd
commit 9437d5ed7a
5 changed files with 93 additions and 59 deletions

View File

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