mirror/bazarr
1
0
Fork 0
mirror of https://github.com/morpheus65535/bazarr synced 2025-02-23 14:31:06 +00:00
Code Issues Releases Wiki Activity

Added switch to control passing of video filename in whisper provider modal

Browse source
This commit is contained in:
JayZed 2024-11-18 23:37:41 -05:00 committed by GitHub
parent 48cdc8bfc8
commit e298d15724
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
bazarr/app/config.py
View file

@ -274,6 +274,7 @@ validators = [
Validator('whisperai.endpoint', must_exist=True, default='http://127.0.0.1:9000', is_type_of=str), Validator('whisperai.endpoint', must_exist=True, default='http://127.0.0.1:9000', is_type_of=str),
Validator('whisperai.response', must_exist=True, default=5, is_type_of=int, gte=1), Validator('whisperai.response', must_exist=True, default=5, is_type_of=int, gte=1),
Validator('whisperai.timeout', must_exist=True, default=3600, is_type_of=int, gte=1), Validator('whisperai.timeout', must_exist=True, default=3600, is_type_of=int, gte=1),
Validator('whisperai.pass_video_name', must_exist=True, default=False, is_type_of=bool),
Validator('whisperai.loglevel', must_exist=True, default='INFO', is_type_of=str, Validator('whisperai.loglevel', must_exist=True, default='INFO', is_type_of=str,
is_in=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']), is_in=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']),

1
bazarr/app/get_providers.py
View file

@ -330,6 +330,7 @@ def get_providers_auth():
'timeout': settings.whisperai.timeout, 'timeout': settings.whisperai.timeout,
'ffmpeg_path': _FFMPEG_BINARY, 'ffmpeg_path': _FFMPEG_BINARY,
'loglevel': settings.whisperai.loglevel, 'loglevel': settings.whisperai.loglevel,
'pass_video_name': settings.whisperai.pass_video_name,
}, },
"animetosho": { "animetosho": {
'search_threshold': settings.animetosho.search_threshold, 'search_threshold': settings.animetosho.search_threshold,

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

@ -233,7 +233,7 @@ class WhisperAIProvider(Provider):
video_types = (Episode, Movie) video_types = (Episode, Movie)
def __init__(self, endpoint=None, response=None, timeout=None, ffmpeg_path=None, loglevel=None): def __init__(self, endpoint=None, response=None, timeout=None, ffmpeg_path=None, pass_video_name=None, loglevel=None):
set_log_level(loglevel) set_log_level(loglevel)
if not endpoint: if not endpoint:
raise ConfigurationError('Whisper Web Service Endpoint must be provided') raise ConfigurationError('Whisper Web Service Endpoint must be provided')
@ -246,12 +246,16 @@ class WhisperAIProvider(Provider):
if not ffmpeg_path: if not ffmpeg_path:
raise ConfigurationError("ffmpeg path must be provided") raise ConfigurationError("ffmpeg path must be provided")
if pass_video_name is None:
raise ConfigurationError('Whisper Web Service Pass Video Name option must be provided')
self.endpoint = endpoint.rstrip("/") self.endpoint = endpoint.rstrip("/")
self.response = int(response) self.response = int(response)
self.timeout = int(timeout) self.timeout = int(timeout)
self.session = None self.session = None
self.ffmpeg_path = ffmpeg_path self.ffmpeg_path = ffmpeg_path
self.pass_video_name = pass_video_name
def initialize(self): def initialize(self):
self.session = Session() self.session = Session()
@ -369,9 +373,11 @@ class WhisperAIProvider(Provider):
logger.info(f'Starting WhisperAI {subtitle.task} to {language_from_alpha3(output_language)} for {subtitle.video.original_path}') logger.info(f'Starting WhisperAI {subtitle.task} to {language_from_alpha3(output_language)} for {subtitle.video.original_path}')
startTime = time.time() startTime = time.time()
video_name = subtitle.video.original_path if self.pass_video_name else None
r = self.session.post(f"{self.endpoint}/asr", r = self.session.post(f"{self.endpoint}/asr",
params={'task': subtitle.task, 'language': input_language, 'output': 'srt', 'encode': 'false'}, params={'task': subtitle.task, 'language': input_language, 'output': 'srt', 'encode': 'false',
'video_file': {video_name}},
files={'audio_file': out}, files={'audio_file': out},
timeout=(self.response, self.timeout)) timeout=(self.response, self.timeout))

6
frontend/src/pages/Settings/Providers/list.ts
View file

@ -561,6 +561,12 @@ export const ProviderList: Readonly<ProviderInfo[]> = [
name: "Logging level", name: "Logging level",
options: logLevelOptions, options: logLevelOptions,
}, },
{
type: "switch",
key: "pass_video_name",
name: "Pass video filename to Whisper (for logging)",
defaultValue: false,
},
{ {
type: "testbutton", type: "testbutton",
key: "whisperai", key: "whisperai",