🐛 fix apply_normalize missing sr

lenML · Jun 23, 2024 · 2db6d65 · 2db6d65
1 parent 26122d4
commit 2db6d65
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/modules/api/impl/handler/SSMLHandler.py b/modules/api/impl/handler/SSMLHandler.py
@@ -93,8 +93,7 @@ def enqueue(self) -> tuple[np.ndarray, int]:
 
         if adjust_config.normalize:
             sample_rate, audio_data = audio.apply_normalize(
-                audio_data=audio_data,
-                headroom=adjust_config.headroom,
+                audio_data=audio_data, headroom=adjust_config.headroom, sr=sample_rate
             )
 
         return audio_data, sample_rate
diff --git a/modules/api/impl/handler/TTSHandler.py b/modules/api/impl/handler/TTSHandler.py
@@ -104,6 +104,7 @@ def enqueue(self) -> tuple[np.ndarray, int]:
             sample_rate, audio_data = apply_normalize(
                 audio_data=audio_data,
                 headroom=adjust_config.headroom,
+                sr=sample_rate,
             )
 
         return audio_data, sample_rate

diff --git a/modules/utils/audio.py b/modules/utils/audio.py
@@ -117,8 +117,9 @@ def apply_prosody_to_audio_data(
 def apply_normalize(
     audio_data: np.ndarray,
     headroom: float = 1,
+    sr: int = 24000,
 ):
-    segment = ndarray_to_segment(audio_data, 24000)
+    segment = ndarray_to_segment(audio_data, sr)
     segment = effects.normalize(seg=segment, headroom=headroom)
 
     return pydub_to_np(segment)