Karl/bazarr-ai-sub-generator
1
0
Fork 0
mirror of https://github.com/karl0ss/bazarr-ai-sub-generator.git synced 2025-04-26 14:59:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix relative imports

Browse Source
This commit is contained in:
Karl Hudgell 2024-01-07 11:26:05 +00:00
parent a9df37bf4f
commit 888c8b6920
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
auto_subtitle/cli.py
View File

@ -1,8 +1,8 @@
import argparse import argparse
from faster_whisper import available_models from faster_whisper import available_models
from .utils.constants import LANGUAGE_CODES from utils.constants import LANGUAGE_CODES
from .main import process from main import process
from .utils.convert import str2bool, str2timeinterval from utils.convert import str2bool, str2timeinterval
def main(): def main():

16
auto_subtitle/main.py
View File

@ -1,9 +1,10 @@
import os import os
import warnings import warnings
import tempfile import tempfile
from .utils.files import filename, write_srt from utils.files import filename, write_srt
from .utils.ffmpeg import get_audio, overlay_subtitles from utils.ffmpeg import get_audio, overlay_subtitles, add_subs_new
from .utils.whisper import WhisperAI from utils.bazarr import get_wanted_episodes, get_episode_details
from utils.whisper import WhisperAI
def process(args: dict): def process(args: dict):
@ -23,7 +24,12 @@ def process(args: dict):
# if translate task used and language argument is set, then use it # if translate task used and language argument is set, then use it
elif language != "auto": elif language != "auto":
args["language"] = language args["language"] = language
a = get_wanted_episodes()
print(f"Found {a['total']} episodes needing subtitles.")
for episode in a['data']:
episode_data = get_episode_details(episode['sonarrEpisodeId'])
print(episode_data)
audios = get_audio(args.pop("video"), args.pop( audios = get_audio(args.pop("video"), args.pop(
'audio_channel'), sample_interval) 'audio_channel'), sample_interval)
@ -38,7 +44,7 @@ def process(args: dict):
if srt_only: if srt_only:
return return
overlay_subtitles(subtitles, output_dir, sample_interval) add_subs_new(subtitles, output_dir, sample_interval)
def get_subtitles(audio_paths: list, output_dir: str, def get_subtitles(audio_paths: list, output_dir: str,