jihuaqiang 3 hours ago
parent
commit
778f4baec6
1 changed files with 14 additions and 2 deletions
  1. 14 2
      content_indentify/indentify.py

+ 14 - 2
content_indentify/indentify.py

@@ -197,8 +197,20 @@ 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', ''):
+            # 判断识别是否成功:如果视频任一项的 asr_content 包含“视频上传失败”,则标记失败
+            video_analysis = recognition_result.get('video_analysis', {})
+            video_upload_failed = False
+            if isinstance(video_analysis, list):
+                for video_item in video_analysis:
+                    if isinstance(video_item, dict) and 'asr_content' in video_item:
+                        if '视频上传失败' in (video_item.get('asr_content') or ''):
+                            video_upload_failed = True
+                            break
+            elif isinstance(video_analysis, dict):
+                if 'asr_content' in video_analysis and '视频上传失败' in (video_analysis.get('asr_content') or ''):
+                    video_upload_failed = True
+
+            if video_upload_failed:
                 self.logger.info(f"记录 {record['id']} 识别失败,将 recognition_status 设置为 3")
                 return False