task_config & improvement

Former-commit-id: 497c2df17859e1c0d7fd9d17df7ef096741a7350

configs/task_config.yaml

@@ -0,0 +1,6 @@
+# configuration for each task
+model: "gpt-4"
+local_dump: ./local_dump
+output_type: srt
+target_lang: CN
+field: SC2

src/task.py

@@ -111,11 +111,12 @@ class Task:
         self.SRT_Script.form_whole_sentence()
         # self.SRT_Script.spell_check_term()
         self.SRT_Script.correct_with_force_term()
-        self.SRT_Script.write_srt_file_src(str(Path(self.task_local_dir).with_suffix(""))+"_processed.srt")
+        processed_srt_path_en = str(Path(self.srt_path).with_suffix('')) + '_processed.srt'
+        self.SRT_Script.write_srt_file_src(processed_srt_path_en)
 
-        if self.launch_info["output_type"] == "ass":
+        if self.output_type == "ass":
             logging.info("write English .srt file to .ass")
-            assSub_en = srt2ass(str(Path(self.task_local_dir).with_suffix(""))+"_processed.srt", "default", "No", "Modest")
+            assSub_en = srt2ass(processed_srt_path_en)
             logging.info('ASS subtitle saved as: ' + assSub_en)
         self.script_input = self.SRT_Script.get_source_only()
         pass