Update app.py

app.py

@@ -17,7 +17,7 @@ os.makedirs("/tmp/transformers_cache", exist_ok=True)
 os.makedirs("/tmp/hf_home", exist_ok=True)
 
 #############################################
-# 模型管理部分 (原 optimized_model_loading.py)
+# 模型管理部分
 #############################################
 
 class ModelManager:
@@ -165,8 +165,9 @@ class ModelManager:
             return float(probabilities[0][predicted_class_idx].item())
         else:
             return 1 - float(probabilities[0][predicted_class_idx].item())
+
 #############################################
-# 特征提取部分 (原 optimized_feature_extraction.py)
+# 特征提取部分
 #############################################
 
 class FeatureExtractor:
@@ -485,7 +486,7 @@ class FeatureExtractor:
             freq_blocks.append(np.mean(magnitude * mask))
         features["freq_anisotropy"] = float(np.std(freq_blocks))
 #############################################
-# 分析逻辑部分 (原 optimized_app.py)
+# 分析逻辑部分
 #############################################
 
 # 初始化管理器
@@ -599,6 +600,9 @@ def get_detailed_analysis(ai_probability, ps_score, beauty_score, ps_signs, ai_s
     elif valid_models_count == 1:
         confidence_prefix = "中等置信度："
     
+    # 计算编辑分数（在所有路径中都需要）
+    combined_edit_score = max(ps_score, beauty_score)
+    
     # 处理特征与模型判断不一致的情况
     if ai_feature_score > 0.8 and ai_probability < 0.6:
         ai_probability = max(0.8, ai_probability)
@@ -615,8 +619,6 @@ def get_detailed_analysis(ai_probability, ps_score, beauty_score, ps_signs, ai_s
         main_category = "AI生成"
     else:
         # 第二级分类：素人 vs 修图
-        combined_edit_score = max(ps_score, beauty_score)
-        
         if combined_edit_score > 0.5:
             category = confidence_prefix + "真人照片，修图痕迹明显"
             description = "图像基本是真人照片，但经过了明显的后期处理或美颜，修饰痕迹明显。"
@@ -638,7 +640,6 @@ def get_detailed_analysis(ai_probability, ps_score, beauty_score, ps_signs, ai_s
     beauty_details = "检测到的美颜特征：" + "、".join(beauty_signs) if beauty_signs else "未检测到明显的美颜特征。"
     
     return category, description, ps_details, ai_details, beauty_details, main_category
-
 def detect_ai_image(image):
     """Main detection function with optimizations"""
     if image is None:
@@ -768,5 +769,4 @@ iface = gr.Interface(
 )
 
 # 启动应用
-if __name__ == "__main__":
-    iface.launch(share=True)
+iface.launch(share=True)