basyx commited on
Commit
decbfbf
·
verified ·
1 Parent(s): 4cf9d8c

Update utils/ffmpeg.py

Browse files
Files changed (1) hide show
  1. utils/ffmpeg.py +12 -31
utils/ffmpeg.py CHANGED
@@ -1,31 +1,12 @@
1
- import ffmpeg
2
- import tempfile
3
- from utils.logger import logger
4
-
5
-
6
- def extract_audio(video_path: str):
7
-
8
- temp_audio = tempfile.NamedTemporaryFile(
9
- delete=False,
10
- suffix=".wav"
11
- )
12
-
13
- output_path = temp_audio.name
14
-
15
- logger.info("Extracting audio via FFmpeg")
16
-
17
- (
18
- ffmpeg
19
- .input(video_path)
20
- .output(
21
- output_path,
22
- ar=16000,
23
- ac=1
24
- )
25
- .overwrite_output()
26
- .run(quiet=True)
27
- )
28
-
29
- logger.info("Audio extraction completed")
30
-
31
- return output_path
 
1
+ import subprocess
2
+ import os
3
+
4
+ def preprocess_audio(input_path: str, output_path: str):
5
+ """Optional: Normalize audio or extract audio from video via FFmpeg."""
6
+ command = [
7
+ "ffmpeg", "-i", input_path,
8
+ "-vn", "-acodec", "pcm_s16le",
9
+ "-ar", "16000", "-ac", "1",
10
+ output_path, "-y"
11
+ ]
12
+ subprocess.run(command, check=True, capture_output=True)