Update periodontitis_detection.py

periodontitis_detection.py

@@ -226,7 +226,15 @@ class SimpleDentalSegmentationNoEnhance:
         abc_mask = self.resize_mask_to_original(abc_unet, (h_orig, w_orig))
         detections = self.detect_teeth(img_bgr)
 
-        combined = img_bgr.copy()
+        # Create a color overlay for CEJ (red) and ABC (blue)
+        overlay = img_bgr.copy()
+        overlay[cej_mask == 1] = (0, 0, 255)   # Red for CEJ
+        overlay[abc_mask == 1] = (255, 0, 0)   # Blue for ABC
+        
+        # Blend overlay with original
+        alpha = 0.4
+        combined = cv2.addWeighted(overlay, alpha, img_bgr, 1 - alpha, 0)
+
         all_results = []
 
         for det in detections: