add

models.py

@@ -5,7 +5,7 @@ class RequestModel(BaseModel):
     originId: int
     sequence: int
     assetCode: str
-    sourceUrl: str
+    source: str
 
 
 class ResponseModel(BaseModel):

similarity.py

@@ -1,3 +1,5 @@
+import base64
+import io
 from typing import List
 from skimage.metrics import structural_similarity as ssim
 import cv2
@@ -8,9 +10,13 @@ from PIL import Image
 from models import RequestModel, ResponseModel
 
 
-def load_image_url(url):
-    print(f'downloading.. {url}')
-    img = Image.open(requests.get(url, stream=True).raw)
+def load_image_url(source):
+    if source.startswith('http'):
+        img = Image.open(requests.get(source, stream=True).raw)
+    else:
+        img_data = base64.b64decode(source)
+        img = Image.open(io.BytesIO(img_data))
+
     img = np.array(img.convert('L'))
     return img
 
@@ -18,13 +24,13 @@ def load_image_url(url):
 def check_similarity(images: List[RequestModel]):
     print(f'checking similarity...')
 
-    original_image = load_image_url(images[0].sourceUrl)
+    original_image = load_image_url(images[0].source)
     original_image_shape = original_image.shape
 
     results = []
 
     for i in range(1, len(images)):
-        image = load_image_url(images[i].sourceUrl)
+        image = load_image_url(images[i].source)
         image = cv2.resize(image, original_image_shape[::-1])
         s, _ = ssim(original_image, image, full=True)
         similarity_score = (s + 1) * 50