Update app.py

app.py

@@ -1,10 +1,12 @@
 import streamlit as st
+from moviepy.editor import VideoFileClip 
 import time
 from transformers import pipeline
 from pytube import YouTube
 from pydub import AudioSegment
 from audio_extract import extract_audio
 import google.generativeai as google_genai
+
 import os
 from dotenv import load_dotenv
 
@@ -34,9 +36,20 @@ def youtube_video_downloader(url):
 
 
 def audio_extraction(video_file):
-    video_file_bytes = os.fsencode(video_file)
-    audio = extract_audio(input_path=video_file_bytes, output_path=f"{video_file}.mp3")
-    return audio
+    # video_file_bytes = os.fsencode(video_file)
+    # audio = extract_audio(input_path=video_file_bytes, output_path=f"{video_file}.mp3")
+    video = VideoFileClip(video_file)
+
+    # Extract the audio from the video
+    audio = video.audio
+
+    # Write the audio to a file
+    audio_path = 'audio.mp3'
+    audio.write_audiofile(audio_path)
+
+    # Close the video file
+    video.close()
+    return audio_path
   
 
 def audio_processing(mp3_audio):