Update similarity.py

similarity.py

@@ -12,16 +12,23 @@ import logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 def orb_sim(img1, img2):
-    orb = cv2.ORB_create()
-    kp_a, desc_a = orb.detectAndCompute(img1, None)
-    kp_b, desc_b = orb.detectAndCompute(img2, None)
+    score = 0
+
+    try:
+        orb = cv2.ORB_create()
+        kp_a, desc_a = orb.detectAndCompute(img1, None)
+        kp_b, desc_b = orb.detectAndCompute(img2, None)
+    
+        bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
+        matches = bf.match(desc_a, desc_b)
+        similar_regions = [i for i in matches if i.distance < 20]
+        if len(matches) > 0:
+            score = len(similar_regions) / len(matches)
+    except Exception as e:
+        logging.error("Erro ao verificar similaridade ORB", exc_info=True)
+
+    return score
 
-    bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
-    matches = bf.match(desc_a, desc_b)
-    similar_regions = [i for i in matches if i.distance < 20]
-    if len(matches) == 0:
-        return 0
-    return len(similar_regions) / len(matches)
 
 def load_image_url(source):
     Image.MAX_IMAGE_PIXELS = None