Update functions.py

functions.py

@@ -96,31 +96,43 @@ 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:
+        
+#         info = ydl.extract_info(url, download=False)
+#         title = info.get('title', None)
+#         ydl.download([url])
+
+#     #with open(temp_audio_file+'.mp3', 'rb') as file:
+#     audio_file = os.path.join('output', 'audio.mp3')
+
+#     return audio_file, title
+
+
 @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:
-        
-        info = ydl.extract_info(url, download=False)
-        title = info.get('title', None)
-        ydl.download([url])
+    yt = YouTube(url)
 
-    #with open(temp_audio_file+'.mp3', 'rb') as file:
-    audio_file = os.path.join('output', 'audio.mp3')
+    # Get the first available audio stream and download it
+    audio_stream =  yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download()
+    title = audio_stream.split('\\')[-1].split('.')[0]
 
-    return audio_file, title
+    return audio_stream, title
 
 
 @st.cache_resource