Update similarity.py

similarity.py

@@ -32,7 +32,7 @@ def preprocess_image_for_mobilenet(image):
 
     return image
 
-def mobilenet_sim(img1, img2):
+def mobilenet_sim(img1, img2, img1AssetCode, img2AssetCode):
     try:
         img1_proc = preprocess_image_for_mobilenet(img1)
         img2_proc = preprocess_image_for_mobilenet(img2)
@@ -42,13 +42,13 @@ def mobilenet_sim(img1, img2):
 
         sim = cosine_similarity(feat1, feat2)[0][0] 
         sim_score = (sim + 1) * 50 
-        print(f"MobileNet similarity score is {sim_score}")
+        print(f"MobileNet similarity score from {img1AssetCode} and {img2AssetCode} is {sim_score}")
         return float(sim_score)
     except Exception as e:
         logging.error("Erro ao calcular similaridade com MobileNet", exc_info=True)
         return 0
 
-def orb_sim(img1, img2):
+def orb_sim(img1, img2, img1AssetCode, img2AssetCode):
     score = 0
 
     try:
@@ -62,7 +62,7 @@ def orb_sim(img1, img2):
         if len(matches) > 0:
             score = (len(similar_regions) / len(matches)) * 100
             if (score > 0):
-                logging.info(f"Orb score is {score}")
+                logging.info(f"Orb score from {img1AssetCode} and {img2AssetCode} is {score}")
     except Exception as e:
         logging.error("Erro ao verificar similaridade ORB", exc_info=True)