Update services/metrics_service.py

services/metrics_service.py

@@ -1,4 +1,5 @@
 import logging
+from typing import List, Dict, Any
 
 # Setup logging
 logging.basicConfig(
@@ -7,7 +8,7 @@ logging.basicConfig(
     format="%(asctime)s - %(levelname)s - %(message)s"
 )
 
-def update_metrics(detected_items):
+def update_metrics(detected_items: List[Dict[str, Any]]) -> Dict[str, Any]:
     """
     Update metrics based on detected items.
     Args:
@@ -16,23 +17,20 @@ def update_metrics(detected_items):
         dict: Updated metrics
     """
     try:
-        total_detections = len(detected_items)
-        avg_confidence = sum(item["confidence"] for item in detected_items if "confidence" in item) / total_detections if total_detections else 0
-        types = [item["type"] for item in detected_items if "type" in item]
-        type_counts = {t: types.count(t) for t in set(types)}
-
+        items = detected_items if detected_items else []
         metrics = {
-            "total_detections": total_detections,
-            "avg_confidence": avg_confidence,
-            "type_counts": type_counts
+            "items": items,
+            "total_detected": len(items),
+            "cracks": len([item for item in items if item.get("type") == "crack"]),
+            "holes": len([item for item in items if item.get("type") == "hole"])
         }
-
         logging.info(f"Updated metrics: {metrics}")
         return metrics
     except Exception as e:
         logging.error(f"Error updating metrics: {str(e)}")
         return {
-            "total_detections": 0,
-            "avg_confidence": 0,
-            "type_counts": {}
+            "items": [],
+            "total_detected": 0,
+            "cracks": 0,
+            "holes": 0
         }