Update app.py

app.py

@@ -1,10 +1,9 @@
 import streamlit as st
 import time
 from transformers import pipeline
-
+import ffmpeg
 from pytube import YouTube
 from pydub import AudioSegment
-
 from audio_extract import extract_audio
 import google.generativeai as google_genai
 import os
@@ -35,8 +34,14 @@ def youtube_video_downloader(url):
     return vid_dld, title
 
 
-def audio_extraction(video_file, output_format):
+def audio_extraction(video_file):
     audio = extract_audio(input_path=os.fspath(video_file), output_path=f"{video_file}.mp3")
+    # input_file = ffmpeg.input(video_file)
+
+    # Extract the audio and save it as an MP3 file
+    # mp3_path = 'audio.mp3'
+    # audio_extracted = input_file.output(mp3_path, acodec='mp3').run()
+    # return mp3_path
     return audio