voice/pytts/pytts_voice.py

"""
pytts voice service (offline)
"""

import time
import pyttsx3
from bridge.reply import Reply, ReplyType
from common.log import logger
from common.tmp_dir import TmpDir
from voice.voice import Voice


class PyttsVoice(Voice):
    engine = pyttsx3.init()

    def __init__(self):
        # 语速
        self.engine.setProperty('rate', 125)
        # 音量
        self.engine.setProperty('volume', 1.0)
        for voice in self.engine.getProperty('voices'):
            if "Chinese" in voice.name:
                self.engine.setProperty('voice', voice.id)

    def textToVoice(self, text):
        try:
            wavFile = TmpDir().path() + 'reply-' + str(int(time.time())) + '.wav'
            self.engine.save_to_file(text, wavFile)
            self.engine.runAndWait()
            logger.info(
                '[Pytts] textToVoice text={} voice file name={}'.format(text, wavFile))
            reply = Reply(ReplyType.VOICE, wavFile)
        except Exception as e:
            reply = Reply(ReplyType.ERROR, str(e))
        finally:
            return reply