Browse Source

上传失败状态判断

jihuaqiang 3 hours ago
parent
commit
ed95a9e2c6
1 changed files with 7 additions and 0 deletions
  1. 7 0
      content_indentify/indentify.py

+ 7 - 0
content_indentify/indentify.py

@@ -124,6 +124,7 @@ class ContentIdentifier:
         # 视频识别
         video_result = self.video_identifier.process_videos(formatted_content)
         
+
         # 整合结果
         recognition_result = {
             'image_analysis': image_result,
@@ -195,6 +196,12 @@ class ContentIdentifier:
             
             # 调用内容识别处理
             recognition_result = self.process_content_recognition(formatted_content)
+
+            # 判断识别是否成功,如果识别失败, 视频的'asr_content': '视频上传失败', 则将 recognition_status 设置为 3
+            if 'asr_content' in recognition_result.get('video_analysis', {}) and '视频上传失败' in recognition_result.get('video_analysis', {}).get('asr_content', ''):
+                self.logger.info(f"记录 {record['id']} 识别失败,将 recognition_status 设置为 3")
+                return False
+
             
             # 构建完整的识别结果
             complete_result = {