From 6e96fb64050e1ce74a152ead59c8380577e6c458 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Deniz=20=C5=9Eafak?= <denizsafak98@gmail.com>
Date: Thu, 22 May 2025 02:42:45 +0300
Subject: [PATCH] Fix previewed voice states when cancel

---
 abogen/voice_formula_gui.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/abogen/voice_formula_gui.py b/abogen/voice_formula_gui.py
index c912904..b182b09 100644
--- a/abogen/voice_formula_gui.py
+++ b/abogen/voice_formula_gui.py
@@ -308,6 +308,12 @@ class HoverLabel(QLabel):
 class VoiceFormulaDialog(QDialog):
     def __init__(self, parent=None, initial_state=None, selected_profile=None):
         super().__init__(parent)
+        # Store original profile/mix state for restoration on cancel
+        self._original_profile_name = None
+        self._original_mixed_voice_state = None
+        if parent is not None:
+            self._original_profile_name = getattr(parent, "selected_profile_name", None)
+            self._original_mixed_voice_state = getattr(parent, "mixed_voice_state", None)
         profiles = load_profiles()
         self._virtual_new_profile = False
         if not profiles:
@@ -914,6 +920,13 @@ class VoiceFormulaDialog(QDialog):
         super().accept()
 
     def reject(self):
+        # Restore parent's profile/mix state on cancel
+        parent = self.parent()
+        if parent is not None:
+            if hasattr(self, "_original_profile_name"):
+                parent.selected_profile_name = self._original_profile_name
+            if hasattr(self, "_original_mixed_voice_state"):
+                parent.mixed_voice_state = self._original_mixed_voice_state
         # Prompt to save if unsaved changes, then check for zero-weight error after save
         if self._has_unsaved_changes():
             if not self._prompt_save_changes():
@@ -923,6 +936,13 @@ class VoiceFormulaDialog(QDialog):
         super().reject()
 
     def closeEvent(self, event):
+        # Restore parent's profile/mix state on close
+        parent = self.parent()
+        if parent is not None:
+            if hasattr(self, "_original_profile_name"):
+                parent.selected_profile_name = self._original_profile_name
+            if hasattr(self, "_original_mixed_voice_state"):
+                parent.mixed_voice_state = self._original_mixed_voice_state
         # Prompt to save if unsaved changes, then check for zero-weight error after save
         if self._has_unsaved_changes():
             if not self._prompt_save_changes():