ai-on-wechat/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
import os

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)

        self.engine.setProperty("voice", "zh")

    def textToVoice(self, text):
        try:
            mp3FileName = "reply-" + str(int(time.time())) + ".mp3"
            mp3File = TmpDir().path() + mp3FileName
            self.engine.save_to_file(text, mp3File)
            self.engine.runAndWait()

            # engine.runAndWait() will return before the file created
            while mp3FileName not in os.listdir(TmpDir().path()):
                time.sleep(0.1)

            logger.info(
                "[Pytts] textToVoice text={} voice file name={}".format(text, mp3File)
            )
            reply = Reply(ReplyType.VOICE, mp3File)
        except Exception as e:
            reply = Reply(ReplyType.ERROR, str(e))
        finally:
            return reply