mirror
/
bazarr
mirror of https://github.com/morpheus65535/bazarr
1
0
Fork 0
Code Issues Releases Wiki Activity

Fixed log message, updated timeout 

When translating, the wrong output language was written to the log, but whisper output was not affected.
Also, changed timeout from 5 to passed timeout value in self.timeout as per recent Discord discussion.
This commit is contained in:
JayZed 2023-12-27 15:55:23 -05:00
parent dafa553587
commit 4a82c4f118
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
libs/subliminal_patch/providers/whisperai.py
View File

@ -256,7 +256,7 @@ class WhisperAIProvider(Provider):
r = self.session.post(f"{self.endpoint}/detect-language",
params={'encode': 'false'},
files={'audio_file': out},
timeout=(5, self.timeout))
timeout=(self.timeout, self.timeout))
logger.debug(f"Whisper detected language of {path} as {r.json()['detected_language']}")
@ -321,16 +321,21 @@ class WhisperAIProvider(Provider):
subtitle.content = None
return
logger.info(f'Starting WhisperAI {subtitle.task} to {language_from_alpha3(subtitle.audio_language)} for {subtitle.video.original_path}')
if subtitle.task == "transcribe":
output_language = subtitle.audio_language
else:
output_language = "eng"
logger.info(f'Starting WhisperAI {subtitle.task} to {language_from_alpha3(output_language)} for {subtitle.video.original_path}')
startTime = time.time()
r = self.session.post(f"{self.endpoint}/asr",
params={'task': subtitle.task, 'language': whisper_get_language_reverse(subtitle.audio_language), 'output': 'srt', 'encode': 'false'},
params={'task': subtitle.task, 'language': whisper_get_language_reverse(output_language), 'output': 'srt', 'encode': 'false'},
files={'audio_file': out},
timeout=(5, self.timeout))
timeout=(self.timeout, self.timeout))
endTime = time.time()
elapsedTime = timedelta(seconds=round(endTime - startTime))
logger.info(f'Completed WhisperAI {subtitle.task} to {language_from_alpha3(subtitle.audio_language)} in {elapsedTime} for {subtitle.video.original_path}')
logger.info(f'Completed WhisperAI {subtitle.task} to {language_from_alpha3(output_language)} in {elapsedTime} for {subtitle.video.original_path}')
subtitle.content = r.content