zhangyong há 6 meses atrás
pai
commit
b938a2fea6
2 ficheiros alterados com 29 adições e 1 exclusões
  1. 28 0
      analyze_video.py
  2. 1 1
      job_video_processing.py

+ 28 - 0
analyze_video.py

@@ -10,6 +10,34 @@ class VideoRequest(BaseModel):
     prompt: str
     mark: str
 
+
+@app.post("/testprocess_test_video/")
+async def process_video_test(request: VideoRequest):
+    """处理视频请求"""
+    video_path = request.video_path
+    prompt = request.prompt
+    mark = request.mark
+    api_key = "AIzaSyAUvBSpjFcm7b8FsgRUTG6anzoalDp9gYg"
+    try:
+        print("来一个请求,使用 API key:", api_key)
+        result, mark = await main(video_path, api_key, prompt, mark)
+        return {
+            "code": 0,
+            "message": "视频处理成功",
+            "result": result,
+            "mark": mark
+        }
+    except Exception as e:
+        print(f"视频处理失败: {str(e)}")
+        return {
+            "code": 1,
+            "message": f"视频处理失败: {e}",
+            "result": f"视频处理失败: {e}",
+            "mark": f"视频处理失败: {e}"
+        }
+
+
+
 @app.post("/process_video/")
 async def process_video(request: VideoRequest):
     """处理视频请求"""

+ 1 - 1
job_video_processing.py

@@ -2,7 +2,7 @@ import time
 from concurrent.futures import ThreadPoolExecutor
 from video_processing.video_processing import VideoProcessing
 
-max_workers = 10  # 最大线程数
+max_workers = 5  # 最大线程数
 
 
 def video_ai_task_start():