Sfoglia il codice sorgente

fix relative imports

Karl Hudgell 1 anno fa
parent
commit
888c8b6920
2 ha cambiato i file con 14 aggiunte e 8 eliminazioni
  1. 3 3
      auto_subtitle/cli.py
  2. 11 5
      auto_subtitle/main.py

+ 3 - 3
auto_subtitle/cli.py

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

+ 11 - 5
auto_subtitle/main.py

@@ -1,9 +1,10 @@
 import os
 import warnings
 import tempfile
-from .utils.files import filename, write_srt
-from .utils.ffmpeg import get_audio, overlay_subtitles
-from .utils.whisper import WhisperAI
+from utils.files import filename, write_srt
+from utils.ffmpeg import get_audio, overlay_subtitles, add_subs_new
+from utils.bazarr import get_wanted_episodes, get_episode_details
+from utils.whisper import WhisperAI
 
 
 def process(args: dict):
@@ -23,7 +24,12 @@ def process(args: dict):
     # if translate task used and language argument is set, then use it
     elif language != "auto":
         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(
         'audio_channel'), sample_interval)
 
@@ -38,7 +44,7 @@ def process(args: dict):
     if srt_only:
         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,