modify logging

Former-commit-id: 58f686a3649affdfb51a17120dc6697cac93e637

src/task.py

@@ -11,8 +11,6 @@ import subprocess
 from src.srt_util.srt import SrtScript
 from src.srt_util.srt2ass import srt2ass
 
-
-
 """
 Youtube link
     - link
@@ -85,25 +83,25 @@ class Task:
         logging.info(f" {self.source_lang} -> {self.target_lang} task in {self.field}")
         logging.info(f" Model: {self.model}")
         logging.info(f" subtitle_type: {self.output_type['subtitle']}")
-        logging.info(f" video_ouput: \t{self.output_type['video']}")
-        logging.info(f" bilingal_ouput: \t{self.output_type['bilingal']}")
+        logging.info(f" video_ouput: {self.output_type['video']}")
+        logging.info(f" bilingal_ouput: {self.output_type['bilingal']}")
 
     @staticmethod
     def fromYoutubeLink(youtube_url, task_id, task_dir, task_cfg):
         # convert to audio
-        logging.info("Task Creation method: Youtube Link")
+        logging.info(" Task Creation method: Youtube Link")
         return YoutubeTask(task_id, task_dir, task_cfg, youtube_url)
 
     @staticmethod
     def fromAudioFile(audio_path, task_id, task_dir, task_cfg):
         # get audio path
-        logging.info("Task Creation method: Audio File")
+        logging.info(" Task Creation method: Audio File")
         return AudioTask(task_id, task_dir, task_cfg, audio_path)
     
     @staticmethod
     def fromVideoFile(video_path, task_id, task_dir, task_cfg):
         # get audio path
-        logging.info("Task Creation method: Video File")
+        logging.info(" Task Creation method: Video File")
         return VideoTask(task_id, task_dir, task_cfg, video_path)
     
     # Module 1 ASR: audio --> SRT_script
@@ -172,16 +170,15 @@ class YoutubeTask(Task):
         if video:
             video.download(str(self.task_local_dir), filename=f"task_{self.task_id}.mp4")
             logging.info(f'Video Name: {video.default_filename}')
-            logging.info(f'Video download completed to {self.task_local_dir}!')
         else:
-            raise FileNotFoundError(f"Video stream not found for link {self.youtube_url}")
+            raise FileNotFoundError(f" Video stream not found for link {self.youtube_url}")
 
         audio = yt.streams.filter(only_audio=True).first()
         if audio:
             audio.download(str(self.task_local_dir), filename=f"task_{self.task_id}.mp3")
-            logging.info(f'Audio download completed to {self.task_local_dir}!')
+            # logging.info(f'Audio download completed to {self.task_local_dir}!')
         else:
-            logging.info("download audio failed, using ffmpeg to extract audio")
+            logging.info(" download audio failed, using ffmpeg to extract audio")
             subprocess.run(
                 ['ffmpeg', '-i', self.task_local_dir.joinpath(f"task_{self.task_id}.mp4"), '-f', 'mp3',
                  '-ab', '192000', '-vn', self.task_local_dir.joinpath(f"task_{self.task_id}.mp3")])
@@ -190,7 +187,9 @@ class YoutubeTask(Task):
         self.video_path = self.task_local_dir.joinpath(f"task_{self.task_id}.mp4")
         self.audio_path = self.task_local_dir.joinpath(f"task_{self.task_id}.mp3")
 
-        logging.info("Data Prep Complete. Start pipeline")
+        logging.info(f" Video File Dir: {self.video_path}")
+        logging.info(f" Audio File Dir: {self.audio_path}")
+        logging.info(" Data Prep Complete. Start pipeline")
 
         super().run_pipeline()
 
@@ -202,6 +201,8 @@ class AudioTask(Task):
         self.video_path = None
 
     def run(self):
+        logging.info(f" Video File Dir: {self.video_path}")
+        logging.info(f" Audio File Dir: {self.audio_path}")
         logging.info("Data Prep Complete. Start pipeline")
         super().run_pipeline()
 
@@ -219,5 +220,7 @@ class VideoTask(Task):
         logging.info("audio extraction finished")
 
         self.audio_path = self.task_local_dir.joinpath(f"task_{self.task_id}.mp3")
+        logging.info(f" Video File Dir: {self.video_path}")
+        logging.info(f" Audio File Dir: {self.audio_path}")
         logging.info("Data Prep Complete. Start pipeline")
         super().run_pipeline()