modify youtube creation

Former-commit-id: b760c5b2eed40ab922ae7e161dc77be6199c1359

entries/run.py

@@ -50,7 +50,7 @@ if __name__ == "__main__":
             'w', encoding='utf-8')])
 
     # TODO: write task info into log
-    
+
     # Task create
     if args.link is not None:
         try:
@@ -60,7 +60,7 @@ if __name__ == "__main__":
             raise RuntimeError("failed to create task from youtube link")
 
     # add task to the status queue
-    task.run_pipeline()
+    task.run()
 
 
 

src/task.py

@@ -74,26 +74,25 @@ class Task:
         self.gpu_status = 0
         self.output_type = launch_info["output_type"]
         self.task_id = task_id
-        self.progress = NotImplemented
+        self.audio_path = None
         self.SRT_Script = None
         self.result = None
 
-
     @staticmethod
     def fromYoutubeLink(youtube_url, task_id, launch_info):
-        # convert to audio
+        # get audio path
         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)
 
     @staticmethod
     def fromAudioFile():
-        # 
+        # get audio path
         return Task(...)
     
     @staticmethod
     def fromVideoFile():
-        # convert to audio
+        # get audio path
         return Task(...)
     
     # Module 1 ASR: audio --> SRT_script
@@ -157,27 +156,46 @@ class YoutubeTask(Task):
 
     def run(self):
         yt = YouTube(self.youtube_url)
-        local_dump = self.task_local_dir
         video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
 
         if video:
-            video.download(str(local_dump.joinpath(f"task_{self.task_id}")), filename=f"task_{self.task_id}.mp4")
-            logging.info(f'Video download completed to {local_dump.joinpath(f"task_{self.task_id}")}!')
+            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}")
 
         audio = yt.streams.filter(only_audio=True).first()
         if audio:
-            audio.download(str(local_dump.joinpath(f"task_{self.task_id}")), filename=f"task_{self.task_id}.mp3")
-            logging.info(f'Audio download completed to {local_dump.joinpath(f"task_{self.task_id}")}!')
+            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}!')
         else:
             logging.info("download audio failed, using ffmpeg to extract audio")
             subprocess.run(
-                ['ffmpeg', '-i', local_dump.joinpath(f"task_{self.task_id}").joinpath(f"task_{self.task_id}.mp4"), '-f', 'mp3',
-                 '-ab', '192000', '-vn', local_dump.joinpath(f"task_{self.task_id}").joinpath(f"task_{self.task_id}.mp3")])
+                ['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")])
             logging.info("audio extraction finished")
+        
+        self.audio_path = self.task_local_dir.joinpath(f"task_{self.task_id}.mp3")
 
         logging.info("Task Creation Complete.")
-        logging.info("Task Creation method: Youtube Link")
+
+        super().run_pipeline()
+
+class AudioTask(Task):
+    def __init__(self, task_id, task_local_dir, launch_info, audio_path):
+        super().__init__(task_id, task_local_dir, launch_info)
+        self.audio_path = audio_path
+
+    def run(self):
+        super().run_pipeline()
+
+class VideoTask(Task):
+    def __init__(self, task_id, task_local_dir, launch_info, video_path):
+        super().__init__(task_id, task_local_dir, launch_info)
+        self.audio_path = video_path
+
+    def run(self):
+        
 
         super().run_pipeline()