Update services/plantation/missing_patch_check.py

services/plantation/missing_patch_check.py

@@ -1,47 +1,46 @@
 import cv2
 import numpy as np
 import os
+import logging
+
+logging.basicConfig(
+    filename="app.log",
+    level=logging.INFO,
+    format="%(asctime)s - %(levelname)s - %(message)s"
+)
 
 def process_missing_patches(frame):
-    """
-    Detect missing patches of plants using contour analysis.
-    Args:
-        frame: Input frame (numpy array)
-    Returns:
-        list: List of missing patches
-        numpy array: Annotated frame with numbered labels
-    """
-    # Convert to grayscale
-    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-    # Apply thresholding to detect plant areas (simplified)
-    _, plant_mask = cv2.threshold(gray, 100, 255, cv2.THRESH_BINARY_INV)
-    contours, _ = cv2.findContours(plant_mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+    try:
+        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+        _, plant_mask = cv2.threshold(gray, 100, 255, cv2.THRESH_BINARY_INV)
+        contours, _ = cv2.findContours(plant_mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
 
-    detections = []
-    line_counter = 1  # Initialize counter for numbered labels
+        detections = []
+        line_counter = 1
 
-    # Identify gaps (missing patches) by analyzing contour areas
-    for contour in contours:
-        area = cv2.contourArea(contour)
-        if area < 500:  # Filter small gaps
-            continue
-        x, y, w, h = cv2.boundingRect(contour)
-        x_min, y_min, x_max, y_max = x, y, x + w, y + h
+        for contour in contours:
+            area = cv2.contourArea(contour)
+            if area < 500:
+                continue
+            x, y, w, h = cv2.boundingRect(contour)
+            x_min, y_min, x_max, y_max = x, y, x + w, y + h
 
-        # Add numbered label
-        detection_label = f"Line {line_counter} - Missing Patch"
-        detections.append({
-            "type": "missing_patch",
-            "label": detection_label,
-            "coordinates": [x_min, y_min, x_max, y_max]
-        })
+            detection_label = f"Line {line_counter} - Missing Patch"
+            detections.append({
+                "type": "missing_patch",
+                "label": detection_label,
+                "coordinates": [x_min, y_min, x_max, y_max]
+            })
 
-        # Draw bounding box and label
-        color = (139, 69, 19)  # Brown for missing patches
-        cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
-        cv2.putText(frame, detection_label, (x_min, y_min - 10), 
-                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
-        
-        line_counter += 1
+            color = (139, 69, 19)  # Brown as requested
+            cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
+            cv2.putText(frame, detection_label, (x_min, y_min - 10), 
+                        cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2)
+            
+            line_counter += 1
 
-    return detections, frame
+        logging.info(f"Detected {len(detections)} missing patches in plantation.")
+        return detections, frame
+    except Exception as e:
+        logging.error(f"Error detecting missing patches: {str(e)}")
+        return [], frame