Spaces:
Running
Running
Commit
·
091d569
1
Parent(s):
abe2568
audio detection bug fixed
Browse files
extraction/media_extractor.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
import os
|
| 2 |
import cv2
|
|
|
|
| 3 |
import librosa
|
| 4 |
import subprocess
|
| 5 |
import numpy as np
|
|
@@ -62,7 +63,7 @@ class MediaExtractor:
|
|
| 62 |
|
| 63 |
def extract_audio(self, video_path: str, timeline: List[Dict]) -> List[Dict]:
|
| 64 |
|
| 65 |
-
temp_audio = "temp_audio.wav"
|
| 66 |
command = [
|
| 67 |
'ffmpeg', '-i', video_path,
|
| 68 |
'-vn', '-acodec', 'pcm_s16le',
|
|
@@ -71,9 +72,10 @@ class MediaExtractor:
|
|
| 71 |
]
|
| 72 |
|
| 73 |
try:
|
| 74 |
-
subprocess.run(command, stdout=subprocess.
|
| 75 |
has_audio = os.path.exists(temp_audio) and os.path.getsize(temp_audio) > 0
|
| 76 |
except subprocess.CalledProcessError:
|
|
|
|
| 77 |
has_audio = False
|
| 78 |
|
| 79 |
if not has_audio:
|
|
|
|
| 1 |
import os
|
| 2 |
import cv2
|
| 3 |
+
from pathlib import Path
|
| 4 |
import librosa
|
| 5 |
import subprocess
|
| 6 |
import numpy as np
|
|
|
|
| 63 |
|
| 64 |
def extract_audio(self, video_path: str, timeline: List[Dict]) -> List[Dict]:
|
| 65 |
|
| 66 |
+
temp_audio = "/tmp/temp_audio.wav"
|
| 67 |
command = [
|
| 68 |
'ffmpeg', '-i', video_path,
|
| 69 |
'-vn', '-acodec', 'pcm_s16le',
|
|
|
|
| 72 |
]
|
| 73 |
|
| 74 |
try:
|
| 75 |
+
subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True, text=True)
|
| 76 |
has_audio = os.path.exists(temp_audio) and os.path.getsize(temp_audio) > 0
|
| 77 |
except subprocess.CalledProcessError:
|
| 78 |
+
print("No audio detected in video")
|
| 79 |
has_audio = False
|
| 80 |
|
| 81 |
if not has_audio:
|