Update app.py

app.py

@@ -39,7 +39,26 @@ model3 = joblib.load('model3_ensemble.pkl')
 model4 = joblib.load('model4_ensemble.pkl')
 final_model = joblib.load('final_model_ensemble.pkl')
 
-def download_audio_from_youtube(youtube_url, output_path='.'):
+# def download_audio_from_youtube(youtube_url, output_path='.'):
+#     ydl_opts = {
+#         'format': 'bestaudio/best',
+#         'outtmpl': f'{output_path}/%(title)s.%(ext)s',
+#         'postprocessors': [{
+#             'key': 'FFmpegExtractAudio',
+#             'preferredcodec': 'wav',
+#             'preferredquality': '192',
+#         }],
+#         'postprocessor_args': ['-ar', '16000'],
+#         'prefer_ffmpeg': True,
+#     }
+
+#     with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+#         info_dict = ydl.extract_info(youtube_url, download=True)
+#         #i have issues with the .webm extension, force replace with .wav
+#         audio_file = ydl.prepare_filename(info_dict).replace('.webm', '.wav')
+#     return audio_file
+
+def download_audio_from_youtube(youtube_url, output_path='.', cookies_file='cookies.txt'):
     ydl_opts = {
         'format': 'bestaudio/best',
         'outtmpl': f'{output_path}/%(title)s.%(ext)s',
@@ -52,9 +71,12 @@ def download_audio_from_youtube(youtube_url, output_path='.'):
         'prefer_ffmpeg': True,
     }
 
+    if cookies_file:
+        ydl_opts['cookiefile'] = cookies_file
+
     with youtube_dl.YoutubeDL(ydl_opts) as ydl:
         info_dict = ydl.extract_info(youtube_url, download=True)
-        #i have issues with the .webm extension, force replace with .wav
+        # Force replace with .wav
         audio_file = ydl.prepare_filename(info_dict).replace('.webm', '.wav')
     return audio_file