55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
import os
|
||
import sys
|
||
|
||
import numpy as np
|
||
import soundfile as sf
|
||
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
import config
|
||
import effects
|
||
import tts
|
||
|
||
|
||
PHRASE = "Приветствую, сэр. Все системы в норме."
|
||
OUT_RAW = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
|
||
'sound', '_preview_silero_raw.wav')
|
||
OUT_FX = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
|
||
'sound', '_preview_silero_fx.wav')
|
||
|
||
|
||
def _render_raw(text: str) -> np.ndarray:
|
||
audio = tts.model.apply_tts(text=text + "..",
|
||
speaker=tts.speaker,
|
||
sample_rate=tts.sample_rate,
|
||
put_accent=tts.put_accent,
|
||
put_yo=tts.put_yo)
|
||
return np.asarray(audio, dtype=np.float32)
|
||
|
||
|
||
def _render_fx(raw: np.ndarray) -> np.ndarray:
|
||
return effects.process(
|
||
raw, tts.sample_rate,
|
||
bandpass=(config.TTS_BANDPASS_LOW_HZ, config.TTS_BANDPASS_HIGH_HZ),
|
||
reverb=(config.TTS_REVERB_WET, config.TTS_REVERB_DECAY_MS),
|
||
pitch_semitones=config.TTS_PITCH_SEMITONES,
|
||
)
|
||
|
||
|
||
def main():
|
||
print(f"Synthesizing: {PHRASE!r}")
|
||
raw = _render_raw(PHRASE)
|
||
fx = _render_fx(raw)
|
||
|
||
sf.write(OUT_RAW, raw, tts.sample_rate, subtype='PCM_16')
|
||
sf.write(OUT_FX, fx, tts.sample_rate, subtype='PCM_16')
|
||
|
||
dur_raw = len(raw) / tts.sample_rate
|
||
dur_fx = len(fx) / tts.sample_rate
|
||
print(f"raw: {OUT_RAW} ({len(raw)} samples, {dur_raw:.2f}s)")
|
||
print(f"fx : {OUT_FX} ({len(fx)} samples, {dur_fx:.2f}s)")
|
||
print("Open both in a media player and A/B compare.")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|