implement download error handler for yt downloads

main.py

@@ -1,4 +1,8 @@
+import time
+
 import click
+from loguru import logger
+from yt_dlp import DownloadError
 
 from downloader import VideoToAudioDownloader
 from transcriber import Transcriber
@@ -18,7 +22,12 @@ def download(urls, proxy):
         params = {"proxy": proxy}
     vad = VideoToAudioDownloader(params=params)
     for url in urls:
-        vad.download_audio_from_yt(url)
+        try:
+            logger.info(f"attempting to download: {url}")
+            vad.download_audio_from_yt(url)
+        except DownloadError as de:
+            logger.warning(f"failed to download {url} b/c: {de}")
+            time.sleep(5)
 
 @cli.command()
 def transcribe():
@@ -26,6 +35,7 @@ def transcribe():
     t.load_model()
     t.DataHandler.get_audio_files()
     for af in t.audio_files:
+        logger.info(f"transcribing: {af}")
         t.transcribe(af)
     t.persist()
     t.cleanup()

requirements.txt

@@ -8,6 +8,7 @@ future==0.18.3
 idna==3.4
 Jinja2==3.1.2
 llvmlite==0.39.1
+loguru==0.7.0
 MarkupSafe==2.1.2
 more-itertools==9.1.0
 mpmath==1.3.0