TranscribeWorker.py

from PySide6 import QtCore
import threading
import traceback
from AzureConversationTranscriber import AzureConversationTranscriber
import logging

logger = logging.getLogger(__name__)
class TranscribeWorker(QtCore.QObject):
    """
    用來做語音轉文字的工作執行緒
    在前端透過QThread啟動此工作者
    """
    TranscribeWorker_Transcribed_Signal = QtCore.Signal(str)
    TranscribeWorker_FinishedSignal = QtCore.Signal()

    def __init__(self, transcriber : AzureConversationTranscriber):
        super().__init__()
        self.transcriber = transcriber
        self._stop_event = threading.Event()

    def run(self):
        logger.info(f'TranscribeWorker.run() started.')

        try:
            # transcribe_from_mic_with_diarization must support on_transcribed and stop_event
            self.transcriber.transcribe_from_mic_with_diarization(on_transcribed=self._emit, stop_event=self._stop_event)
        except Exception:
            logger.exception('Exception in TranscribeWorker.run():\n' + traceback.format_exc())
        finally:
            try:
                self.TranscribeWorker_FinishedSignal.emit()
            except Exception:
                logger.exception('Failed to emit finished signal from TranscribeWorker.')

    def _emit(self, text):
        try:
            self.TranscribeWorker_Transcribed_Signal.emit(text)
        except Exception:
            logger.exception('Failed to emit transcribed_signal from TranscribeWorker.')

    def stop(self):
        try:
            self._stop_event.set()
            logger.info('Stop event set in TranscribeWorker.')
        except Exception:
            logger.exception('Failed to set stop event in TranscribeWorker.')

        
        try:
            if hasattr(self.transcriber, 'stop_transcribing_async'):
                try:
                    self.transcriber.stop_transcribing_async()
                except Exception:
                    logger.exception('Failed to call stop_transcribing_async in TranscribeWorker.')
        except Exception:
            logger.exception('Failed to check or call stop_transcribing_async in TranscribeWorker.')