hueso/gajim-plugins
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Introduce dataclasses for configuring model settings

Browse Source
This commit is contained in:
mesonium
2024-06-18 15:15:52 +02:00
committed by hueso
parent f5db7f9809
commit 36c96b4a16
6 changed files with 68 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
stt_voice_messages/gtk/sttbox.py
View File

@@ -49,26 +49,29 @@ class STTBox(Gtk.Box):
self.add(self._transcribe_button)
self.add(self._transcription_label)
self._result = helper.Results('')
self._transcribe_button.connect('clicked', self._on_transcribe_clicked)
self.show_all()
def _on_transcribe_clicked(self, _button: Gtk.Button) -> None:
log.debug('config.data = %s', self._config.data)
model_class = self._config.data['model_class']
if model_class is None:
model = self._config.data['model_instance']
if model is None:
return
self._model = model_class()
self._model = model
transcription_task = helper.BackgroundTask(
self._model.transcribe(self._audio_file),
self._model.transcribe(self._result, self._audio_file),
self._show_result
)
transcription_task.start()
def _show_result(self):
self._text = self._model.result
assert self._model is not None
self._text = self._result.text
if self._text.strip() != '':
self._transcription_label.set_text(self._text.strip())
else: