121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
import pyttsx3
|
||
import threading
|
||
import queue
|
||
import time
|
||
|
||
class VoicePrompt:
|
||
def __init__(self, rate=150, volume=1.0, voice_id=None):
|
||
"""初始化语音提示组件
|
||
|
||
Args:
|
||
rate: 语速
|
||
volume: 音量 (0.0 到 1.0)
|
||
voice_id: 语音ID (None表示使用默认值)
|
||
"""
|
||
self.rate = rate
|
||
self.volume = volume
|
||
self.voice_id = voice_id
|
||
|
||
# 创建语音消息队列
|
||
self.speech_queue = queue.Queue()
|
||
|
||
# 启动单独的线程来处理语音播报
|
||
self.speech_thread = threading.Thread(target=self._speech_worker, daemon=True)
|
||
self.speech_thread.start()
|
||
|
||
# 防止多次初始化
|
||
self.initialized = False
|
||
|
||
def _init_engine(self):
|
||
"""初始化语音引擎"""
|
||
if not hasattr(self, 'engine') or self.engine is None:
|
||
try:
|
||
self.engine = pyttsx3.init()
|
||
self.engine.setProperty('rate', self.rate)
|
||
self.engine.setProperty('volume', self.volume)
|
||
|
||
# 设置语音（如果提供了voice_id）
|
||
if self.voice_id:
|
||
self.engine.setProperty('voice', self.voice_id)
|
||
else:
|
||
# 尝试设置中文语音（如果有的话）
|
||
voices = self.engine.getProperty('voices')
|
||
for voice in voices:
|
||
if 'chinese' in voice.id.lower() or 'zh' in voice.id.lower():
|
||
self.engine.setProperty('voice', voice.id)
|
||
break
|
||
self.initialized = True
|
||
except Exception as e:
|
||
print(f"初始化语音引擎失败: {e}")
|
||
|
||
def _speech_worker(self):
|
||
"""后台线程，从队列中获取文本并播报"""
|
||
# 初始化引擎（仅在这个区域创建和使用引擎）
|
||
self._init_engine()
|
||
|
||
while True:
|
||
try:
|
||
# 从队列中读取文本
|
||
text = self.speech_queue.get()
|
||
if text == "__EXIT__": # 退出信号
|
||
break
|
||
|
||
if not self.initialized:
|
||
self._init_engine()
|
||
|
||
# 仅当初始化成功时播放语音
|
||
if self.initialized:
|
||
try:
|
||
self.engine.say(text)
|
||
self.engine.runAndWait()
|
||
except Exception as e:
|
||
print(f"语音播报错误: {e}")
|
||
# 重新初始化引擎
|
||
try:
|
||
self.engine = None
|
||
self._init_engine()
|
||
except Exception as e2:
|
||
print(f"重新初始化引擎失败: {e2}")
|
||
time.sleep(1) # 避免循环过快
|
||
|
||
# 通知队列任务完成
|
||
self.speech_queue.task_done()
|
||
except Exception as e:
|
||
print(f"语音工作线程错误: {e}")
|
||
time.sleep(0.5) # 防止错误时CPU资源过度消耗
|
||
|
||
def speak(self, text, block=False):
|
||
"""将要播报的文本添加到队列
|
||
|
||
Args:
|
||
text: 要播报的文本
|
||
block: 是否等待播报完成
|
||
"""
|
||
try:
|
||
# 将文本加入队列
|
||
self.speech_queue.put(text)
|
||
|
||
# 如果是阻塞模式，等待这个任务完成
|
||
if block:
|
||
self.speech_queue.join()
|
||
except Exception as e:
|
||
print(f"添加语音文本到队列失败: {e}")
|
||
|
||
def get_available_voices(self):
|
||
"""获取所有可用的语音
|
||
|
||
Returns:
|
||
list: 可用语音列表
|
||
"""
|
||
voices = self.engine.getProperty('voices')
|
||
return [(voice.id, voice.name) for voice in voices]
|
||
|
||
# 测试代码
|
||
if __name__ == "__main__":
|
||
voice = VoicePrompt()
|
||
print("可用语音:")
|
||
for voice_id, name in voice.get_available_voices():
|
||
print(f" - {name} ({voice_id})")
|
||
|
||
voice.speak("您好，欢迎使用人脸识别系统", block=True)
|