Update app.py

app.py

@@ -1,6 +1,5 @@
 import whisper
 import yt_dlp
-import moviepy.editor as mp
 import os
 import torch
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
@@ -10,6 +9,7 @@ from datetime import timedelta
 import gradio as gr
 import torchaudio
 import whisper.tokenizer
+import ffmpeg
 
 # -----------------------------
 # Helper Functions
@@ -32,14 +32,23 @@ def download_youtube_audio(url):
     except Exception as e:
         raise RuntimeError(f"Error downloading audio: {str(e)}")
 
+
+
 def extract_audio_from_video(video_path):
+    output_audio = "audio.mp3"
     try:
-        clip = mp.VideoFileClip(video_path)
-        clip.audio.write_audiofile("audio.mp3")
-        return "audio.mp3"
+        ffmpeg.input(video_path).output(
+            output_audio,
+            vn=None,  # Disable video
+            acodec="libmp3lame",  # Use MP3 codec
+            ar="16000",  # 16kHz sample rate
+            ac="1"  # Mono channel
+        ).run(quiet=True, overwrite_output=True)
+        return output_audio
     except Exception as e:
-        raise RuntimeError(f"Error extracting audio: {str(e)}")
+        raise RuntimeError(f"FFmpeg error: {str(e)}")
 
+        
 def generate_srt(segments):
     subs = []
     for i, seg in enumerate(segments):