Fixed infinite download retry loop in whisperai provider

morpheus65535 · Nov 19, 2023 · b25d416 · b25d416
1 parent d64d6b0
commit b25d416
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/bazarr/app/get_providers.py b/bazarr/app/get_providers.py
@@ -12,6 +12,7 @@
 import traceback
 import re
 
+from requests import ConnectionError
 from subzero.language import Language
 from subliminal_patch.exceptions import TooManyRequests, APIThrottled, ParseResponseError, IPAddressBlocked, \
     MustGetBlacklisted, SearchLimitReached

diff --git a/libs/subliminal_patch/core.py b/libs/subliminal_patch/core.py
@@ -38,7 +38,7 @@
 CUSTOM_PATHS = []
 INCLUDE_EXOTIC_SUBS = True
 
-DOWNLOAD_TRIES = 0
+DOWNLOAD_TRIES = 3
 DOWNLOAD_RETRY_SLEEP = 6
 
 # fixme: this may be overkill

diff --git a/libs/subliminal_patch/providers/whisperai.py b/libs/subliminal_patch/providers/whisperai.py
@@ -233,7 +233,7 @@ def detect_language(self, path) -> Language:
         r = self.session.post(f"{self.endpoint}/detect-language",
                               params={'encode': 'false'},
                               files={'audio_file': out},
-                              timeout=self.timeout)
+                              timeout=(5, self.timeout))
 
         logger.info(f"Whisper detected language of {path} as {r.json()['detected_language']}")
 
@@ -290,6 +290,6 @@ def download_subtitle(self, subtitle: WhisperAISubtitle):
         r = self.session.post(f"{self.endpoint}/asr",
                               params={'task': subtitle.task, 'language': whisper_get_language_reverse(subtitle.audio_language), 'output': 'srt', 'encode': 'false'},
                               files={'audio_file': out},
-                              timeout=self.timeout)
+                              timeout=(5, self.timeout))
 
         subtitle.content = r.content