Fix key error when loading output formats from settings (#482)

chidiwilliams · Jun 5, 2023 · d63ccb9 · d63ccb9
1 parent 162c203
commit d63ccb9
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/buzz/gui.py b/buzz/gui.py
@@ -201,8 +201,11 @@ def __init__(self, file_paths: List[str], parent: Optional[QWidget] = None,
                                             default_value=DEFAULT_WHISPER_TEMPERATURE),
             word_level_timings=self.settings.value(key=Settings.Key.FILE_TRANSCRIBER_WORD_LEVEL_TIMINGS,
                                                    default_value=False))
+        default_export_format_states: List[str] = self.settings.value(key=Settings.Key.FILE_TRANSCRIBER_EXPORT_FORMATS,
+                                                                      default_value=[])
         self.file_transcription_options = FileTranscriptionOptions(
-            file_paths=self.file_paths)
+            file_paths=self.file_paths,
+            output_formats=set([OutputFormat(output_format) for output_format in default_export_format_states]))
 
         layout = QVBoxLayout(self)
 
@@ -221,11 +224,9 @@ def __init__(self, file_paths: List[str], parent: Optional[QWidget] = None,
         file_transcription_layout.addRow('', self.word_level_timings_checkbox)
 
         export_format_layout = QHBoxLayout()
-        default_export_format_states: List[str] = self.settings.value(key=Settings.Key.FILE_TRANSCRIBER_EXPORT_FORMATS,
-                                                                      default_value=[])
         for output_format in OutputFormat:
             export_format_checkbox = QCheckBox(f'{output_format.value.upper()}', parent=self)
-            export_format_checkbox.setChecked(output_format.value in default_export_format_states)
+            export_format_checkbox.setChecked(output_format in self.file_transcription_options.output_formats)
             export_format_checkbox.stateChanged.connect(self.get_on_checkbox_state_changed_callback(output_format))
             export_format_layout.addWidget(export_format_checkbox)