Update app.py

app.py

@@ -1137,6 +1137,7 @@ def detect_ai_image(image):
         adjusted_probability, ps_score, beauty_score, ps_signs, ai_signs, beauty_signs, valid_models
     )
     
+    # 构建最终结果
     # 构建最终结果
     final_result = {
         "ai_probability": adjusted_probability,
@@ -1145,7 +1146,7 @@ def detect_ai_image(image):
         "beauty_score": beauty_score,
         "ai_feature_score": ai_feature_score,
         "category": category,
-        "main_category": main_category,  # 添加主要分类
+        "main_category": main_category,
         "description": description,
         "ps_details": ps_details,
         "ai_details": ai_details,
@@ -1154,7 +1155,11 @@ def detect_ai_image(image):
         "features": image_features
     }
     
-    return final_result
+    # 为Label组件准备数据
+    label_data = {main_category: 1.0}
+    
+    # 返回两个值：JSON结果和Label数据
+    return final_result, label_data
 
 # 创建Gradio界面
 iface = gr.Interface(