修改ai创作失败原因显示

video-gen-api/app/services/generation_ai_service.py

@@ -362,16 +362,26 @@ def _resolve_error_message(error_message: str | None) -> str | None:
         return error_message
 
     import re
-    # 匹配 code='InputVideoSensitiveContentDetected.PrivacyInformation' 格式
-    match = re.search(r"code='([^']+)'", error_message)
-    if match:
-        code = match.group(1)
-        if code in ARK_ERRORS:
-            return ARK_ERRORS[code]
-    else:
-        parts = error_message.split(":")
-        if len(parts) >= 2 and parts[1].strip() in ARK_ERRORS:
-            return ARK_ERRORS[parts[1].strip()]
+
+    # 匹配以下格式中的错误码：
+    # 1. {'error': {'code': 'XXX', ...}}  — str(error_obj) 的 Python dict 形式
+    # 2. {"error": {"code": "XXX", ...}}  — JSON 形式
+    # 3. code='XXX'                       — 旧格式
+    for pattern in [
+        r"'code'\s*:\s*'([^']+)'",          # 'code': 'XXX'
+        r'"code"\s*:\s*"([^"]+)"',          # "code": "XXX"
+        r"code='([^']+)'",                   # code='XXX'
+    ]:
+        match = re.search(pattern, error_message)
+        if match:
+            code = match.group(1)
+            if code in ARK_ERRORS:
+                return ARK_ERRORS[code]
+
+    # 兜底：按冒号分割，检查第二部分是否是已知错误码
+    parts = error_message.split(":")
+    if len(parts) >= 2 and parts[1].strip() in ARK_ERRORS:
+        return ARK_ERRORS[parts[1].strip()]
 
     # 没有匹配到已知错误码时，直接返回"生成失败"
     return "生成失败"