Update similarity.py

similarity.py

@@ -163,27 +163,40 @@ def load_image_url(source, assetCode, contentType=None, ffmpeg_path='ffmpeg', fr
 def check_similarity(images: List[RequestModel]):
     logging.info(f"Checking similarity for main source with resource id {images[0].originId}")
 
-    original_image = load_image_url(images[0].source, images[0].assetCode, images[0].contentType)
-    original_image_shape = original_image.shape
-
-    results = []
-
-    for i in range(1, len(images)):
-        try:
-            image = load_image_url(images[i].source, images[i].assetCode, images[i].contentType)
-            image = cv2.resize(image, original_image_shape[::-1])
-        
-            similarity_score = ssim_sim(original_image, image)
-            similarity_orb_score = orb_sim(original_image, image, images[0].assetCode, images[i].assetCode)
-            similarity_mobilenet_score = mobilenet_sim(original_image, image, images[0].assetCode, images[i].assetCode)
-        except Exception as e:
-            logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
+    try:
+        original_image = load_image_url(images[0].source, images[0].assetCode, images[0].contentType)
+        original_image_shape = original_image.shape
+    
+        results = []
+    
+        for i in range(1, len(images)):
+            try:
+                image = load_image_url(images[i].source, images[i].assetCode, images[i].contentType)
+                image = cv2.resize(image, original_image_shape[::-1])
+            
+                similarity_score = ssim_sim(original_image, image)
+                similarity_orb_score = orb_sim(original_image, image, images[0].assetCode, images[i].assetCode)
+                similarity_mobilenet_score = mobilenet_sim(original_image, image, images[0].assetCode, images[i].assetCode)
+            except Exception as e:
+                logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
+                similarity_score = 0
+                similarity_orb_score = 0
+                similarity_mobilenet_score = 0
+    
+            response = ResponseModel(originId=images[i].originId, source=images[i].source, sequence=images[i].sequence,
+                                     assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
+            results.append(response)
+            
+    except Exception as e:
+        logging.error(f"Error checking similarity for main source with resource id {images[i].originId} : {e}")
+
+        for i in range(1, len(images)):
             similarity_score = 0
             similarity_orb_score = 0
             similarity_mobilenet_score = 0
 
-        response = ResponseModel(originId=images[i].originId, source=images[i].source, sequence=images[i].sequence,
-                                 assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
-        results.append(response)
+            response = ResponseModel(originId=images[i].originId, source=images[i].source, sequence=images[i].sequence,
+                                     assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
+            results.append(response)
 
     return results