nishchandel commited on
Commit
091d569
·
1 Parent(s): abe2568

audio detection bug fixed

Browse files
Files changed (1) hide show
  1. extraction/media_extractor.py +4 -2
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.DEVNULL, stderr=subprocess.DEVNULL, check=True)
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: