github-mirrors/bazarr
1
0
Fork 0
mirror of https://github.com/morpheus65535/bazarr.git synced 2025-04-24 14:47:16 -04:00
Code Issues Projects Releases Packages Wiki Activity

Handle empty responses from whisper to prevent JSONDecodeErrors

Browse source
This commit is contained in:
JayZed 2024-10-30 16:58:33 -04:00 • committed by GitHub
parent 4eb09c546d
commit 2c8712f891
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
custom_libs/subliminal_patch/providers/whisperai.py
View file

@ -5,6 +5,7 @@ from datetime import timedelta
from requests import Session
from requests.exceptions import JSONDecodeError
from subliminal_patch.subtitle import Subtitle
from subliminal_patch.providers import Provider
from subliminal import __short_version__
@ -269,10 +270,19 @@ class WhisperAIProvider(Provider):
params={'encode': 'false'},
files={'audio_file': out},
timeout=(self.response, self.timeout))
try:
results = r.json()
except JSONDecodeError:
results = {}
logger.debug(f"Whisper detected language of {path} as {r.json()['detected_language']}")
if len(results) == 0:
logger.info(f"Whisper returned empty response when detecting language")
return None
return whisper_get_language(r.json()["language_code"], r.json()["detected_language"])
logger.debug(f"Whisper detected language of {path} as {results['detected_language']}")
return whisper_get_language(results["language_code"], results["detected_language"])
def query(self, language, video):
if language not in self.languages: