Update tools/download/video.py

tools/download/video.py

@@ -1,3 +1,4 @@
+# Code based on cutechicken/whisper-webui-translate
 import io
 from contextlib import redirect_stderr
 from tempfile import mkdtemp
@@ -24,9 +25,9 @@ def download_url(
     destinationDirectory: str = None,
     playlistItems: str = "1",
     cookies_from_browser: str = None,
-) -> List[str]:
+) -> tuple[bool, str, List[str]]:
     try:
-        return _perform_download(
+        video_paths = _perform_download(
             url,
             maxDuration=maxDuration,
             outputTemplate=None,
@@ -34,16 +35,29 @@ def download_url(
             playlistItems=playlistItems,
             cookies_from_browser=cookies_from_browser,
         )
+        return True, None, video_paths
     except yt_dlp.utils.DownloadError as e:
         # In case of an OS error, try again with a different output template
         if e.msg and e.msg.find("[Errno 36] File name too long") >= 0:
-            return _perform_download(
-                url,
-                maxDuration=maxDuration,
-                outputTemplate="%(title).10s %(id)s.%(ext)s",
-                cookies_from_browser=cookies_from_browser,
-            )
-        raise
+            try:
+                video_paths = _perform_download(
+                    url,
+                    maxDuration=maxDuration,
+                    outputTemplate="%(title).10s %(id)s.%(ext)s",
+                    cookies_from_browser=cookies_from_browser,
+                )
+                return True, None, video_paths
+            except Exception as retry_error:
+                return False, str(retry_error), []
+        return False, str(e), []
+    except ExceededMaximumDuration as e:
+        return (
+            False,
+            f"Video exceeds maximum duration: {e.videoDuration}s > {e.maxDuration}s",
+            [],
+        )
+    except Exception as e:
+        return False, str(e), []
 
 
 def _perform_download(