Update app.py

app.py

@@ -960,8 +960,8 @@ def detect_photoshop_signs(image_features):
             ps_signs.append("边缘过渡过于平滑，可能有过度修饰")
     
     return min(ps_score, 1.0), ps_signs
-
-def get_detailed_analysis(ai_probability, ps_score, beauty_score, ps_signs, ai_signs, beauty_signs, valid_models_count, ai_feature_score):
+# 在这里添加get_detailed_analysis函数
+def get_detailed_analysis(ai_probability, ps_score, beauty_score, ps_signs, ai_signs, beauty_signs, valid_models_count, ai_feature_score, image_features=None):
     """提供更详细的分析结果，使用二级分类框架，优先考虑AI特征分析"""
     
     # 根据有效模型数量调整置信度描述
@@ -983,14 +983,34 @@ def get_detailed_analysis(ai_probability, ps_score, beauty_score, ps_signs, ai_s
         ai_probability = max(0.7, ai_probability)  # 当AI特征分数高时，提高AI概率
     
     # 特定关键特征的硬性覆盖
-    if "lbp_entropy" in image_features and image_features["lbp_entropy"] < 2.0:
-        if "freq_anisotropy" in image_features and image_features["freq_anisotropy"] < 0.05:
-            # 当微观纹理熵极低且频率分布各向异性极低时，几乎可以确定是AI生成
-            ai_probability = 0.95
-            category = confidence_prefix + "AI生成图像（确定）"
-            description = "检测到多个决定性AI生成特征，该图像几乎可以确定是AI生成的。"
-            main_category = "AI生成"
-            return category, description, ps_details, ai_details, beauty_details, main_category
+    if image_features is not None:
+        if "lbp_entropy" in image_features and image_features["lbp_entropy"] < 2.0:
+            if "freq_anisotropy" in image_features and image_features["freq_anisotropy"] < 0.05:
+                # 当微观纹理熵极低且频率分布各向异性极低时，几乎可以确定是AI生成
+                ai_probability = 0.95
+                category = confidence_prefix + "AI生成图像（确定）"
+                description = "检测到多个决定性AI生成特征，该图像几乎可以确定是AI生成的。"
+                main_category = "AI生成"
+                
+                # 添加具体的PS痕迹描述
+                if ps_signs:
+                    ps_details = "检测到的修图痕迹：" + "、".join(ps_signs)
+                else:
+                    ps_details = "未检测到明显的修图痕迹。"
+                
+                # 添加AI特征描述
+                if ai_signs:
+                    ai_details = "检测到的AI特征：" + "、".join(ai_signs)
+                else:
+                    ai_details = "未检测到明显的AI生成特征。"
+                
+                # 添加美颜特征描述
+                if beauty_signs:
+                    beauty_details = "检测到的美颜特征：" + "、".join(beauty_signs)
+                else:
+                    beauty_details = "未检测到明显的美颜特征。"
+                
+                return category, description, ps_details, ai_details, beauty_details, main_category
     
     # 第一级分类：AI生成 vs 真人照片
     if ai_probability > 0.6:  # 降低AI判定阈值，提高AI检出率