Update app.py

app.py

@@ -65,36 +65,36 @@ def prepare_media(source_path_or_url: os.PathLike,
     if source_type == 'audio_filepath':
         audio_file = source_path_or_url
     elif source_type == 'youtube_url':
-        proxy_handler = {"http": "http://108.61.175.7:31802", "https":"http://190.187.201.26:8080", "https":"http://38.54.71.67:80"}
-        yt = YouTube(source_path_or_url, proxies=proxy_handler) #use_oauth=True, allow_oauth_cache=False)
-        audio_stream = min(yt.streams.filter(only_audio=True), key=lambda s: s.bitrate)
-        mp4_file = audio_stream.download(output_path='downloaded') # ./downloaded
-        audio_file = mp4_file[:-3] + 'mp3'
-        subprocess.run(['ffmpeg', '-i', mp4_file, '-ac', '1', audio_file])
-        os.remove(mp4_file)
+        # proxy_handler = {"http": "http://108.61.175.7:31802", "https":"http://190.187.201.26:8080", "https":"http://38.54.71.67:80"}
+        # yt = YouTube(source_path_or_url, proxies=proxy_handler) #use_oauth=True, allow_oauth_cache=False)
+        # audio_stream = min(yt.streams.filter(only_audio=True), key=lambda s: s.bitrate)
+        # mp4_file = audio_stream.download(output_path='downloaded') # ./downloaded
+        # audio_file = mp4_file[:-3] + 'mp3'
+        # subprocess.run(['ffmpeg', '-i', mp4_file, '-ac', '1', audio_file])
+        # os.remove(mp4_file)
         # Download from youtube
-        # try:
-        #     # Try PyTube first
-        #     # proxy_handler = {"http": "http://127.0.0.1:1087", "https":"http://127.0.0.1:1087"}
-        #     # yt = YouTube(source_path_or_url, proxies=proxy_handler)
-        #     yt = YouTube(source_path_or_url)
-        #     audio_stream = min(yt.streams.filter(only_audio=True), key=lambda s: s.bitrate)
-        #     mp4_file = audio_stream.download(output_path='downloaded') # ./downloaded
-        #     audio_file = mp4_file[:-3] + 'mp3'
-        #     subprocess.run(['ffmpeg', '-i', mp4_file, '-ac', '1', audio_file])
-        #     os.remove(mp4_file)
-        # except Exception as e:
-        #     try:
-        #         # Try alternative
-        #         print(f"Failed with PyTube, error: {e}. Trying yt-dlp...")
-        #         audio_file = './downloaded/yt_audio'
-        #         subprocess.run(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
-        #             '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
-        #             '--force-overwrites', '--cookies', 'amt/src/extras/c.txt'])
-        #         audio_file += '.mp3'
-        #     except Exception as e:
-        #         print(f"Alternative downloader failed, error: {e}. Please try again later!")
-        #         return None
+        try:
+            # Try PyTube first
+            # proxy_handler = {"http": "http://127.0.0.1:1087", "https":"http://127.0.0.1:1087"}
+            # yt = YouTube(source_path_or_url, proxies=proxy_handler)
+            yt = YouTube(source_path_or_url)
+            audio_stream = min(yt.streams.filter(only_audio=True), key=lambda s: s.bitrate)
+            mp4_file = audio_stream.download(output_path='downloaded') # ./downloaded
+            audio_file = mp4_file[:-3] + 'mp3'
+            subprocess.run(['ffmpeg', '-i', mp4_file, '-ac', '1', audio_file])
+            os.remove(mp4_file)
+        except Exception as e:
+            try:
+                # Try alternative
+                print(f"Failed with PyTube, error: {e}. Trying yt-dlp...")
+                audio_file = './downloaded/yt_audio'
+                subprocess.run(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio', '-6',
+                    '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
+                    '--force-overwrites', '--cookies', 'amt/src/extras/c.txt'])
+                audio_file += '.mp3'
+            except Exception as e:
+                print(f"Alternative downloader failed, error: {e}. Please try again later!")
+                return None
     else:
         raise ValueError(source_type)