Update app.py

app.py

@@ -10,10 +10,8 @@ import torch
 import librosa
 from transformers import Wav2Vec2ForCTC, AutoProcessor
 from pathlib import Path
-from moviepy.editor import VideoFileClip
 import magic  # For MIME type detection
-import tempfile
-import os
+from pydub import AudioSegment
 
 # Import functions from other modules
 from asr import transcribe, ASR_LANGUAGES
@@ -46,20 +44,11 @@ def extract_audio(input_bytes):
     
     if mime_type.startswith('audio/'):
         return sf.read(io.BytesIO(input_bytes))
-    elif mime_type.startswith('video/'):
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.webm') as temp_file:
-            temp_file.write(input_bytes)
-            temp_file_path = temp_file.name
-
-        try:
-            video = VideoFileClip(temp_file_path)
-            audio = video.audio
-            audio_array = audio.to_soundarray()
-            sample_rate = audio.fps
-            video.close()
-            return audio_array, sample_rate
-        finally:
-            os.unlink(temp_file_path)
+    elif mime_type.startswith('video/webm'):
+        audio = AudioSegment.from_file(io.BytesIO(input_bytes), format="webm")
+        audio_array = np.array(audio.get_array_of_samples())
+        sample_rate = audio.frame_rate
+        return audio_array, sample_rate
     else:
         raise ValueError(f"Unsupported MIME type: {mime_type}")