update

ai_tag_task.py

@@ -128,29 +128,32 @@ def ai_tags(project, table, dt):
             retry += 1
             time.sleep(60)
             continue
-        pool = multiprocessing.Pool(processes=5)
+        # pool = multiprocessing.Pool(processes=5)
+        # for video_id in video_folder_list:
+        #     if video_id not in video_id_list:
+        #         continue
+        #     pool.apply_async(
+        #         func=process,
+        #         args=(video_id, video_info, download_folder)
+        #     )
+        # pool.close()
+        # pool.join()
         for video_id in video_folder_list:
             if video_id not in video_id_list:
                 continue
-            pool.apply_async(
-                func=process,
-                args=(video_id, video_info, download_folder)
-            )
-        pool.close()
-        pool.join()
-            # if video_info.get(video_id, None) is None:
-            #     shutil.rmtree(os.path.join(download_folder, video_id))
-            # else:
-            #     video_folder = os.path.join(download_folder, video_id)
-            #     for filename in os.listdir(video_folder):
-            #         video_type = filename.split('.')[-1]
-            #         if video_type in ['mp4', 'm3u8']:
-            #             video_file = os.path.join(video_folder, filename)
-            #             get_video_ai_tags(video_id=video_id, video_file=video_file, video_info=video_info.get(video_id))
-            #             # 将处理过的视频进行删除
-            #             shutil.rmtree(os.path.join(download_folder, video_id))
-            #         else:
-            #             shutil.rmtree(os.path.join(download_folder, video_id))
+            if video_info.get(video_id, None) is None:
+                shutil.rmtree(os.path.join(download_folder, video_id))
+            else:
+                video_folder = os.path.join(download_folder, video_id)
+                for filename in os.listdir(video_folder):
+                    video_type = filename.split('.')[-1]
+                    if video_type in ['mp4', 'm3u8']:
+                        video_file = os.path.join(video_folder, filename)
+                        get_video_ai_tags(video_id=video_id, video_file=video_file, video_info=video_info.get(video_id))
+                        # 将处理过的视频进行删除
+                        shutil.rmtree(os.path.join(download_folder, video_id))
+                    else:
+                        shutil.rmtree(os.path.join(download_folder, video_id))
 
 
 def timer_check():

whisper_asr.py

@@ -5,7 +5,6 @@ def get_whisper_asr(video, model='medium'):
     try:
         model = whisper.load_model(model)
         result = model.transcribe(video)
-        print(result["text"])
         return result['text']
     except:
         return ''