saving the state selected by the voice formula dialog

abogen/gui.py

@@ -459,6 +459,7 @@ class abogen(QWidget):
         self.last_output_path = None
         self.selected_voice = self.config.get("selected_voice", "af_heart")
         self.selected_lang = self.selected_voice[0]
+        self.mixed_voice_state = None  # Store the mixed voice state
         self.is_converting = False
         self.subtitle_mode = self.config.get("subtitle_mode", "Sentence")
         self.max_subtitle_words = self.config.get(
@@ -1662,7 +1663,17 @@ class abogen(QWidget):
         save_config(self.config)
         
     def show_voice_formula_dialog(self):
-        VoiceFormulaDialog(self).exec_()
+        # get the current voice mix
+        if self.mixed_voice_state is None:
+            # if no voice mix is set, use the selected voice
+            self.mixed_voice_state = [(self.selected_voice, 1.0)]
+        
+        dialog = VoiceFormulaDialog(self, initial_state=self.mixed_voice_state)
+        if dialog.exec_() == QDialog.Accepted:
+            self.mixed_voice_state = dialog.get_selected_voices()
+            print("Selected voices and weights:", self.mixed_voice_state)
+        else:
+            print("Dialog canceled")
        
     def show_about_dialog(self):
         """Show an About dialog with program information including GitHub link."""

abogen/voice_formula_gui.py

@@ -21,7 +21,7 @@ from constants import (
 voice_mixer_width = 160
 
 class VoiceMixer(QWidget):
-    def __init__(self, voice_name, language_icon, update_formula_callback):
+    def __init__(self, voice_name, language_icon, update_formula_callback, initial_status=False, initial_weight=0.0):
         super().__init__()
 
         self.voice_name = voice_name
@@ -38,7 +38,7 @@ class VoiceMixer(QWidget):
 
         # Checkbox at the top
         self.checkbox = QCheckBox()
-        self.checkbox.setChecked(True)
+        self.checkbox.setChecked(initial_status)
         self.checkbox.stateChanged.connect(self.update_formula_callback)
         layout.addWidget(self.checkbox, alignment=Qt.AlignCenter)
 
@@ -56,11 +56,12 @@ class VoiceMixer(QWidget):
         self.spin_box.setRange(0, 1)
         self.spin_box.setSingleStep(0.01)
         self.spin_box.setDecimals(2)
+        self.spin_box.setValue(initial_weight)
         self.spin_box.valueChanged.connect(self.update_formula_callback)
 
         self.slider = QSlider(Qt.Vertical)  # Set slider orientation to vertical
         self.slider.setRange(0, 100)
-        self.slider.setValue(100)  # Default to 1.0
+        self.slider.setValue(int(initial_weight * 100))
         self.slider.setFixedHeight(180)
         self.slider.valueChanged.connect(
             lambda val: self.spin_box.setValue(val / 100)
@@ -92,8 +93,14 @@ class VoiceMixer(QWidget):
         return ""
 
 
+    def get_voice_weight(self):
+        """Return the voice and its weight if selected."""
+        if self.checkbox.isChecked():
+            return self.voice_name, self.spin_box.value()
+        return None
+
 class VoiceFormulaDialog(QDialog):
-    def __init__(self, parent=None):
+    def __init__(self, parent=None, initial_state=None):
         super().__init__(parent)
 
         self.setWindowTitle("Voice Mixer")
@@ -106,7 +113,7 @@ class VoiceFormulaDialog(QDialog):
         # Scroll Area for Voice Panels
         self.scroll_area = QScrollArea()
         self.scroll_area.setWidgetResizable(True)
-        self.scroll_area.setFixedSize(1000, 500)  # Keep scroll area height within 500
+        self.scroll_area.setFixedSize(1000, 400)  # Keep scroll area height within 500
         self.voice_list_widget = QWidget()
         self.voice_list_layout = QHBoxLayout()  
         self.voice_list_widget.setLayout(self.voice_list_layout)
@@ -138,16 +145,16 @@ class VoiceFormulaDialog(QDialog):
         self.setLayout(main_layout)
         
         # Initialize with fixed voices
-        self.add_voices()
-
-
-    def add_voices(self):
         for voice in VOICES_INTERNAL:
             flag = FLAGS.get(voice[0], "")
-            self.add_voice(voice, flag)
+            matching_voice = next((item for item in initial_state if item[0] == voice), None)
+            initial_status = matching_voice is not None
+            initial_weight = matching_voice[1] if matching_voice else 1.0
+            self.add_voice(voice, flag, initial_status, initial_weight)
 
-    def add_voice(self, voice_name, language_icon):
+        
-        voice_mixer = VoiceMixer(voice_name, language_icon, self.update_formula)
+    def add_voice(self, voice_name, language_icon, initial_status=False, initial_weight=1.0):
+        voice_mixer = VoiceMixer(voice_name, language_icon, self.update_formula, initial_status, initial_weight)
         self.voice_mixers.append(voice_mixer)
         self.voice_list_layout.addWidget(voice_mixer)
         self.update_formula()
@@ -163,3 +170,10 @@ class VoiceFormulaDialog(QDialog):
         ]
         formula = " + ".join(filter(None, formula_components))
         self.formula_label.setText(f"Voice Formula: {formula}")
+
+    def get_selected_voices(self):
+        """Return the list of selected voices and their weights."""
+        selected_voices = [
+            mixer.get_voice_weight() for mixer in self.voice_mixers
+        ]
+        return [voice for voice in selected_voices if voice]  # Filter out None