From 006ab10b4439a755a9044b0953765e8aad890b73 Mon Sep 17 00:00:00 2001 From: Sara Date: Fri, 13 Oct 2023 10:50:46 +0200 Subject: [PATCH] use numpy in waveform calculation --- server/reflector/utils/audio_waveform.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/server/reflector/utils/audio_waveform.py b/server/reflector/utils/audio_waveform.py index 7a6fdb3e..8fd184a8 100644 --- a/server/reflector/utils/audio_waveform.py +++ b/server/reflector/utils/audio_waveform.py @@ -57,12 +57,9 @@ def get_audio_waveform(path: Path | str, segments_count: int = 256) -> list[int] # number of decimals to use when rounding the peak value digits = 2 - max_val = float(max(volumes)) - new_volumes = [] - for x in volumes: - new_volumes.append(round(x / max_val, digits)) + volumes = np.round(volumes / volumes.max(), digits) - return new_volumes + return volumes if __name__ == "__main__":