Update functions.py

functions.py

@@ -101,6 +101,40 @@ def load_prompt():
 
 ###################### Functions #######################################################################################
 
+@st.cache_data
+def get_yt_audio(url):
+    temp_audio_file = os.path.join('output', 'audio')
+
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192',
+        }],
+        'outtmpl': temp_audio_file,
+        'quiet': True,
+    }
+
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
+
+    with open(temp_audio_file+'.mp3', 'rb') as file:
+        audio_file = file.read()
+
+    return audio_file
+
+if __name__ == "__main__":
+    video_url = url
+    temp_dir = r"C:\Users\nickm\OneDrive\Python_Code\Machine_Learning_Code\HuggingFace\ASR\temp"
+    temp_audio_file = download_audio(video_url, temp_dir)
+    with open(temp_audio_file+'.mp3', 'rb') as file:
+        audio_file = file.read()
+
+        # Use the audio_file variable here for further processing
+        print(len(audio_file))
+
+
 @st.cache_resource
 def load_models():