dev test

coldStartTasks/ai_pipeline/__init__.py

@@ -2,4 +2,4 @@
 @author: luojunhui
 """
 
-from .video_to_text import GenerateTextFromVideo
+from .extract_video_best_frame import ExtractVideoBestFrame

run_video_understanding_with_google.py

@@ -5,30 +5,30 @@
 import datetime
 
 from applications import log
-from coldStartTasks.ai_pipeline import GenerateTextFromVideo
+from coldStartTasks.ai_pipeline import ExtractVideoBestFrame
 
 
 def main():
-    task = GenerateTextFromVideo()
+    task = ExtractVideoBestFrame()
 
     # 查询有多少任务正在处理中
-    processing_tasks = task.get_processing_task_num()
+    processing_tasks = task.get_processing_task_pool_size()
 
     if processing_tasks:
         print(
             f"{datetime.datetime.now()} 当前有 {processing_tasks} 个任务正在等待 google 处理..."
         )
-        task.convert_video_to_text_with_google_ai_task()
+        task.extract_best_frame_with_gemini_ai()
     else:
         print(f"{datetime.datetime.now()} 没有任务正在处理中...")
         # upload video to google ai
-        task.upload_video_to_google_ai_task()
+        task.upload_video_to_gemini_ai()
         log(
             task="video_understanding_with_google",
             function="main",
             message="upload_video_to_google_ai_task",
         )
-        task.convert_video_to_text_with_google_ai_task()
+        task.extract_best_frame_with_gemini_ai()
 
 
 if __name__ == "__main__":