Update app.py

app.py

@@ -475,6 +475,43 @@ def segment_with_choice(use_box_choice, annot_value, mode="Overlay"):
     return Image.fromarray(overlay), result_text
 
 # ===== 计数功能 =====
+def count_cells_handler(image_path):
+    """计数处理函数"""
+    if image_path is None:
+        return None, "⚠️ 请先上传图像"
+    
+    if COUNT_MODEL is None:
+        return None, "❌ 计数模型未加载"
+    
+    try:
+        print(f"🔢 Counting - Image: {image_path}")
+        
+        result = run_count(
+            COUNT_MODEL,
+            image_path,
+            box=None,
+            device=COUNT_DEVICE,
+            visualize=True
+        )
+        
+        if 'error' in result:
+            return None, f"❌ 计数失败: {result['error']}"
+        
+        count = result['count']
+        viz_path = result['visualized_path']
+        result_text = f"✅ 检测到 {count:.1f} 个细胞"
+        
+        print(f"✅ Counting done - Count: {count:.1f}")
+        
+        return viz_path, result_text
+        
+    except Exception as e:
+        print(f"❌ Counting error: {e}")
+        import traceback
+        traceback.print_exc()
+        return None, f"❌ 计数失败: {str(e)}"
+
+# ===== 跟踪功能 =====
 import zipfile
 import tempfile
 import shutil