From 621ae7d681c17842b570a7db05668f8d5b15e71d Mon Sep 17 00:00:00 2001 From: Mathieu Virbel Date: Wed, 23 Aug 2023 19:24:05 +0200 Subject: [PATCH] server: update to set 256 points by default --- server/reflector/utils/audio_waveform.py | 4 ++-- server/reflector/views/transcripts.py | 2 +- server/tests/test_transcripts_audio_download.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/server/reflector/utils/audio_waveform.py b/server/reflector/utils/audio_waveform.py index c94a4f72..97f00b53 100644 --- a/server/reflector/utils/audio_waveform.py +++ b/server/reflector/utils/audio_waveform.py @@ -4,7 +4,7 @@ import av import numpy as np -def get_audio_waveform(path: Path | str, segments_count: int = 1000) -> list[int]: +def get_audio_waveform(path: Path | str, segments_count: int = 256) -> list[int]: if isinstance(path, Path): path = path.as_posix() @@ -67,7 +67,7 @@ if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument("path", type=Path) - parser.add_argument("--segments-count", type=int, default=1000) + parser.add_argument("--segments-count", type=int, default=256) args = parser.parse_args() print(get_audio_waveform(args.path, args.segments_count)) diff --git a/server/reflector/views/transcripts.py b/server/reflector/views/transcripts.py index deaa0567..c92079a6 100644 --- a/server/reflector/views/transcripts.py +++ b/server/reflector/views/transcripts.py @@ -121,7 +121,7 @@ class Transcript(BaseModel): # move temporary file to final location shutil.move(tmp.name, fn.as_posix()) - def convert_audio_to_waveform(self, segments_count=1000): + def convert_audio_to_waveform(self, segments_count=256): fn = self.audio_waveform_filename if fn.exists(): return diff --git a/server/tests/test_transcripts_audio_download.py b/server/tests/test_transcripts_audio_download.py index 2af34ee7..a33ecceb 100644 --- a/server/tests/test_transcripts_audio_download.py +++ b/server/tests/test_transcripts_audio_download.py @@ -104,4 +104,4 @@ async def test_transcript_audio_download_waveform(fake_transcript): assert response.status_code == 200 assert response.headers["content-type"] == "application/json" assert isinstance(response.json()["data"], list) - assert len(response.json()["data"]) == 1000 + assert len(response.json()["data"]) == 256