Explorar o código

comment out more unused code

Karl hai 1 ano
pai
achega
b08966cd5f
Modificáronse 1 ficheiros con 8 adicións e 8 borrados
  1. 8 8
      auto_subtitle/main.py

+ 8 - 8
auto_subtitle/main.py

@@ -11,14 +11,14 @@ from utils.whisper import WhisperAI
 
 def process(args: dict):
     model_name: str = args.pop("model")
-    output_dir: str = args.pop("output_dir")
-    output_srt: bool = args.pop("output_srt")
-    srt_only: bool = args.pop("srt_only")
+    # output_dir: str = args.pop("output_dir")
+    # output_srt: bool = args.pop("output_srt")
+    # srt_only: bool = args.pop("srt_only")
     language: str = args.pop("language")
     sample_interval: str = args.pop("sample_interval")
     audio_channel: str = args.pop('audio_channel')
 
-    os.makedirs(output_dir, exist_ok=True)
+    # os.makedirs(output_dir, exist_ok=True)
 
     if model_name.endswith(".en"):
         warnings.warn(
@@ -39,11 +39,11 @@ def process(args: dict):
         print(f"Processing {episode['seriesTitle']} - {episode['episode_number']}")
         episode_data = get_episode_details(episode['sonarrEpisodeId'])
         audios = get_audio([episode_data['path']], audio_channel, sample_interval)
-        srt_output_dir = output_dir if output_srt or srt_only else tempfile.gettempdir()
-        subtitles = get_subtitles(audios, srt_output_dir, model_args, args)
+        # srt_output_dir = output_dir if output_srt or srt_only else tempfile.gettempdir()
+        subtitles = get_subtitles(audios, tempfile.gettempdir(), model_args, args)
 
-        if srt_only:
-            return
+        # if srt_only:
+        #     return
 
         add_subs_new(subtitles)
         update_show_in_soarr(episode['sonarrSeriesId'])