zhangyong 6 mesi fa
parent
commit
6f75ab2b88
2 ha cambiato i file con 27 aggiunte e 2 eliminazioni
  1. 26 1
      analyze_video.py
  2. 1 1
      video_processing/video_processing.py

+ 26 - 1
analyze_video.py

@@ -12,7 +12,32 @@ class VideoRequest(BaseModel):
     mark: str
     sample_data: str
 
+@app.post("/process_video_test/")
+async def process_video_test(request: VideoRequest):
+    """处理视频请求"""
+    video_path = request.video_path
+    prompt = request.prompt
+    mark = request.mark
+    sample_data = request.sample_data
 
+    api_key = "AIzaSyB2kjF2-S2B5cJiosx_LpApd227w33CVvs"
+    try:
+        print("来一个请求,使用 API key:", api_key)
+        result, mark = await main(video_path, api_key, prompt, mark, sample_data)
+        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):
@@ -45,4 +70,4 @@ async def process_video(request: VideoRequest):
 
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8081)
+    uvicorn.run(app, host="0.0.0.0", port=8080)

+ 1 - 1
video_processing/video_processing.py

@@ -45,7 +45,7 @@ class VideoProcessing:
                 "二级分类": ["品类- 、分数-", "品类- 、分数-", "品类- 、分数-"]
             }
         }
-        url = "http://8.219.186.16:8081/process_video/"
+        url = "http://8.219.186.16:8080/process_video/"
         payload = json.dumps( {
             "video_path": video_path,
             "prompt": prompt,