split task_cfg and launch_info

Former-commit-id: 8901366529d43d9ffb6c520b38f9689929234058

configs/task_config.yaml

@@ -6,5 +6,5 @@ output_type:
   video: False
   bilingal: False
 source_lang: EN
-target_lang: CN
+target_lang: ZH
 field: SC2

src/task.py

@@ -64,15 +64,18 @@ class Task:
         with self.__status_lock:
             self.__status = new_status
 
-    def __init__(self, task_id, task_local_dir, launch_info):
+    def __init__(self, task_id, task_local_dir, task_config):
         self.__status_lock = threading.Lock()
         self.__status = TaskStatus.CREATED
         openai.api_key = getenv("OPENAI_API_KEY")
-        self.launch_info = launch_info
+        self.launch_info = task_config # do not use, just for fallback
         self.task_local_dir = task_local_dir
-        self.model = launch_info["model"]
+        self.model = task_config["model"]
         self.gpu_status = 0
-        self.output_type = launch_info["output_type"]
+        self.output_type = task_config["output_type"]
+        self.target_lang = task_config["target_lang"]
+        self.source_lang = task_config["source_lang"]
+        self.field = task_config["field"]
         self.task_id = task_id
         self.progress = NotImplemented
         self.SRT_Script = None
@@ -80,11 +83,11 @@ class Task:
 
 
     @staticmethod
-    def fromYoutubeLink(youtube_url, task_id, launch_info):
+    def fromYoutubeLink(youtube_url, task_id, launch_info, task_config):
         # convert to audio
         logging.info("Task Creation method: Youtube Link")
         local_dump = Path(launch_info['local_dump'])  # should get from launch config
-        return YoutubeTask(task_id, local_dump.joinpath(f"task_{task_id}"), launch_info, youtube_url)
+        return YoutubeTask(task_id, local_dump.joinpath(f"task_{task_id}"), task_config, youtube_url)
 
     @staticmethod
     def fromAudioFile():
@@ -114,7 +117,7 @@ class Task:
         processed_srt_path_en = str(Path(self.task_local_dir).with_suffix('')) + '_processed.srt'
         self.SRT_Script.write_srt_file_src(processed_srt_path_en)
 
-        if self.output_type == "ass":
+        if self.output_type["subtitle"] == "ass":
             logging.info("write English .srt file to .ass")
             assSub_en = srt2ass(processed_srt_path_en)
             logging.info('ASS subtitle saved as: ' + assSub_en)
@@ -151,8 +154,8 @@ class Task:
         self.result = self.output_render()
 
 class YoutubeTask(Task):
-    def __init__(self, task_id, task_local_dir, launch_info, youtube_url):
-        super().__init__(task_id, task_local_dir, launch_info)
+    def __init__(self, task_id, task_local_dir, task_config, youtube_url):
+        super().__init__(task_id, task_local_dir, task_config)
         self.youtube_url = youtube_url
 
     def run(self):