zhangyong 6 hónapja
szülő
commit
fa2a92a772
3 módosított fájl, 10 hozzáadás és 7 törlés
  1. 3 3
      analyze_video.py
  2. 2 1
      google_ai/generativeai_video.py
  3. 5 3
      job_video_processing.py

+ 3 - 3
analyze_video.py

@@ -69,9 +69,9 @@ async def process_video(request: VideoRequest):
             "result": f"视频处理失败: {e}",
             "mark": f"视频处理失败: {e}"
         }
-    finally:
-        # 释放 API key,确保后续请求可以使用
-        await release_api_key(api_key)
+    # finally:
+    #     # 释放 API key,确保后续请求可以使用
+    #     await release_api_key(api_key)
 
 
 if __name__ == "__main__":

+ 2 - 1
google_ai/generativeai_video.py

@@ -90,6 +90,7 @@ class VideoAnalyzer:
                 ]
             }
             response = chat_session.send_message( message_content )
+            self.video_file.delete()
             return response
 
     def video_duration(self, filename):
@@ -113,7 +114,7 @@ async def main(video_path, api_key):
             # 初始化视频分析类
             analyzer = VideoAnalyzer(api_key )
             for file in genai.list_files():
-                file.delete()
+                print( "  ", file.name )
             duration = analyzer.video_duration( video_path )
             print( f"视频时长为{duration}秒" )
             if int( duration ) >= 600 or int( duration ) == 0:

+ 5 - 3
job_video_processing.py

@@ -10,9 +10,11 @@ def video_ai_task_start():
     with ThreadPoolExecutor( max_workers=max_workers) as executor:
         while True:
             try:
-                redis_task_list = ['task:video_ai']
-                # 提交所有任务并等待完成
-                futures = [executor.submit( process_video_ai, redis_task ) for redis_task in redis_task_list]
+                redis_task_list = ['task:video_ai'] * 2
+                futures = []
+                for redis_task in redis_task_list:
+                    futures.append( executor.submit( process_video_ai, redis_task ) )
+                    time.sleep( 1 )  # 每秒提交一个任务
                 wait( futures )  # 等待所有任务完成
             except Exception as e:
                 print(f"异常信息{e}")