From 5896f0db666893abe0b3c64f7207a794d2dcb3fd Mon Sep 17 00:00:00 2001 From: Wolfang Torres Date: Thu, 25 Jun 2026 21:25:49 +0800 Subject: [PATCH] change topurch audio to soundfile --- pyproject.toml | 2 +- src/anki_hsk_creator/proccessor.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index e5cbce2..18a7a33 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -31,7 +31,7 @@ dependencies = [ "genanki", "argostranslate", "torch", - "torchaudio", + "soundfile", "torchcodec", "python-dotenv", "qwen-tts", diff --git a/src/anki_hsk_creator/proccessor.py b/src/anki_hsk_creator/proccessor.py index 23d9889..110b4a2 100644 --- a/src/anki_hsk_creator/proccessor.py +++ b/src/anki_hsk_creator/proccessor.py @@ -5,7 +5,7 @@ import csv # Pip import argostranslate.translate -import torchaudio +import soundfile as sf # Local from .constants import LANGUAGES @@ -32,7 +32,7 @@ def dictation_process( audio_path = process_file.resources / f"N{n:03n}.wav" if not audio_path.exists(): wavs, sr = TTS.generate(f"{audio_line}。") - torchaudio.save(audio_path, wavs[0], sr) + sf.write(audio_path, wavs[0], sr) translated = argostranslate.translate.translate( line, LANGUAGES.CN, process_file.language_id ) @@ -55,7 +55,7 @@ def translator_process( audio_path = process_file.resources / f"N{n:03n}.wav" if not audio_path.exists(): wavs, sr = TTS.generate(f"{audio_line}。") - torchaudio.save(audio_path, wavs[0], sr) + sf.write(audio_path, wavs[0], sr) translated = argostranslate.translate.translate( line, LANGUAGES.CN, process_file.language_id ) @@ -118,7 +118,7 @@ def dictionary_process(process_file: ProcessFile) -> list[DictionaryResult]: audio_path = process_file.resources / f"{line['pinyin']}.wav" if not audio_path.exists(): wavs, sr = TTS.generate(f"{line['simplified']}。") - torchaudio.save(audio_path, wavs[0], sr) + sf.write(audio_path, wavs[0], sr) print(line) result = DictionaryResult( **line, audio_path=audio_path, language_id=process_file.language_id