Update app.py

app.py

@@ -536,11 +536,13 @@ class ImageProcessor:
 # Create processor instance
 processor = ImageProcessor()
 
+# 修改 run_demo 函数
 def run_demo(input_image, threshold, enhance_contrast, denoise, show_contours, 
             brightness, contrast, filter_type, process_mode):
     """主处理函数"""
     if input_image is None:
-        return [None] * 9 + ["请上传图像"]
+        # 返回 9 个值（对应 9 个输出组件）
+        return [None] * 6 + ["请上传图像", None, None]
     
     # 处理图像
     results = processor.process_image(
@@ -554,18 +556,18 @@ def run_demo(input_image, threshold, enhance_contrast, denoise, show_contours,
         process_mode=process_mode
     )
     
+    # 解包结果（确保有 9 个值）
     original, saliency_map, heatmap, overlayed, segmented, comparison, time_info, stats, analysis_plot = results
     
-    # 如果需要显示轮廓
-    if show_contours and saliency_map is not None and overlayed is not None:
+    # 如果需要显示轮廓（仅限完整模式）
+    if show_contours and process_mode == "完整分析" and saliency_map is not None and overlayed is not None:
         _, binary = cv2.threshold(saliency_map, 127, 255, cv2.THRESH_BINARY)
         contours, _ = cv2.findContours(binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
         overlay_with_contours = overlayed.copy()
         cv2.drawContours(overlay_with_contours, contours, -1, (0,255,0), 2)
         overlayed = overlay_with_contours
     
-    # 生成统计信息HTML
-# 生成统计信息HTML
+    # 生成统计信息HTML（始终处理）
     if stats:
         stats_html = "<div class='statistics-container' style='background: white; color: black;'>"
         for key, value in stats.items():
@@ -578,6 +580,19 @@ def run_demo(input_image, threshold, enhance_contrast, denoise, show_contours,
         stats_html += "</div>"
     else:
         stats_html = "<p style='color: black; font-weight: bold;'>无可用统计信息</p>"
+    
+    # 返回 9 个值
+    return (
+        original, 
+        saliency_map, 
+        heatmap, 
+        overlayed, 
+        segmented, 
+        comparison, 
+        time_info, 
+        stats_html,  # 使用生成的 HTML
+        analysis_plot
+    )
 
 def create_comparison_view(original, result, slider_value):
     """创建滑块对比视图"""