Update obstruction_detector.py

obstruction_detector.py

@@ -1,6 +1,7 @@
 import cv2 as cv
 import numpy as np
 from scipy.signal import find_peaks
+from PIL import Image  # Import PIL
 
 class ObstructionDetector:
     def __init__(self, threshold=500):
@@ -33,9 +34,9 @@ class ObstructionDetector:
         peaks, _ = find_peaks(smoothed_histogram.flatten(), height=self.threshold)
         return peaks
 
-    def detect_obstruction(self, img_path):
-        # Load the image
-        img = cv.imread(img_path)
+    def detect_obstruction(self, pil_image):  # Accept PIL image directly
+        # Convert PIL image to NumPy array
+        img = np.array(pil_image)
 
         # Preprocess the image
         preprocessed_img = self.preprocess_image(img)
@@ -51,4 +52,4 @@ class ObstructionDetector:
         else:
             report = "A imagem contém obstrução significativa | possui múltiplas distribuições de densidade claramente distintas."
 
-        return report
+        return report