69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
|
import logger
|
|||
|
|
import pyttsx3
|
|||
|
|
from PyQt5.QtWidgets import QWidget, QPushButton, QVBoxLayout, QLabel
|
|||
|
|
from uas.io.base_io import BaseOutput, IOType, IOData
|
|||
|
|
from uas.io.outputs.text_output import HAS_TTS
|
|||
|
|
from uas.io.signals import io_signals
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SpeechOutput(BaseOutput):
|
|||
|
|
def __init__(self):
|
|||
|
|
super().__init__("speech_output", [IOType.TEXT, IOType.SPEECH])
|
|||
|
|
self._engine = None
|
|||
|
|
if HAS_TTS:
|
|||
|
|
self._init_tts()
|
|||
|
|
|
|||
|
|
def _init_tts(self):
|
|||
|
|
try:
|
|||
|
|
self._engine = pyttsx3.init()
|
|||
|
|
self._engine.setProperty("rate", 180)
|
|||
|
|
self._engine.setProperty("volume", 0.9)
|
|||
|
|
logger.info("✅ TTS 引擎初始化成功")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"TTS 初始化失败: {e}")
|
|||
|
|
|
|||
|
|
def render(self, data: IOData) -> None:
|
|||
|
|
text = str(data.processed or data.raw)
|
|||
|
|
self._tts_label.setText(f"朗读: {text[:80]}...")
|
|||
|
|
if not self._mute_btn.isChecked() and self._engine:
|
|||
|
|
try:
|
|||
|
|
self._engine.say(text)
|
|||
|
|
self._engine.runAndWait()
|
|||
|
|
io_signals.tts_finished.emit(text)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"TTS 朗读失败: {e}")
|
|||
|
|
# ─────────────────────────────────────────────────────────────────
|
|||
|
|
# 语音输出（TTS）
|
|||
|
|
# ─────────────────────────────────────────────────────────────────
|
|||
|
|
|
|||
|
|
class SpeechOutputWidget(QWidget, SpeechOutput):
|
|||
|
|
"""
|
|||
|
|
语音输出（TTS）组件
|
|||
|
|
基于 pyttsx3，在独立 QThread 中朗读，不阻塞 UI
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(self, parent=None):
|
|||
|
|
QWidget.__init__(self, parent)
|
|||
|
|
SpeechOutput.__init__(self)
|
|||
|
|
self._build_ui()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _build_ui(self):
|
|||
|
|
layout = QVBoxLayout(self)
|
|||
|
|
layout.setContentsMargins(8, 8, 8, 8)
|
|||
|
|
|
|||
|
|
title = QLabel("🔊 语音输出 (TTS)")
|
|||
|
|
title.setStyleSheet("font-weight: bold; font-size: 13px;")
|
|||
|
|
layout.addWidget(title)
|
|||
|
|
|
|||
|
|
self._tts_label = QLabel("等待输出...")
|
|||
|
|
self._tts_label.setWordWrap(True)
|
|||
|
|
self._tts_label.setStyleSheet(
|
|||
|
|
"background:#e8f5e9; padding:6px; border-radius:4px;"
|
|||
|
|
)
|
|||
|
|
layout.addWidget(self._tts_label)
|
|||
|
|
|
|||
|
|
self._mute_btn = QPushButton("🔇 静音")
|
|||
|
|
self._mute_btn.setCheckable(True)
|
|||
|
|
layout.addWidget(self._mute_btn)
|