Upload livro.py

livro.py

@@ -20,8 +20,13 @@ def detectar_pontos(imagem_array, mascara_array=None, idade=6):
     gray = cv2.cvtColor(imagem_array, cv2.COLOR_BGR2GRAY)
     _, thresh = cv2.threshold(gray, 200, 255, cv2.THRESH_BINARY_INV)
 
-    if mascara_array is not None:
-        mask = cv2.cvtColor(mascara_array, cv2.COLOR_BGR2GRAY)
+    if mascara_array is not None and isinstance(mascara_array, np.ndarray):
+        try:
+            mask = cv2.cvtColor(mascara_array, cv2.COLOR_BGR2GRAY)
+            _, mask_thresh = cv2.threshold(mask, 10, 255, cv2.THRESH_BINARY)
+            thresh = cv2.bitwise_and(thresh, mask_thresh)
+        except Exception as e:
+            print("Erro ao processar máscara:", e)
         _, mask_thresh = cv2.threshold(mask, 10, 255, cv2.THRESH_BINARY)
         thresh = cv2.bitwise_and(thresh, mask_thresh)