from src.model import simlarity_model as model 
from src.util import image as image_util
from .model_implements.mobilenet_v3 import ModelnetV3

class Similarity:
    def get_models(self):
        return [model.SimilarityModel(name= 'mobilenet_v3', image_size= 224, model_cls = ModelnetV3())]        

    def check_similarity(self, img_urls, model):
        # model = self.get_models()[model_idx]
        imgs = []
        for url in img_urls:
            if url == "": continue
            imgs.append(image_util.load_image_url(url, required_size=(model.image_size, model.image_size)))
        model.model_cls.extract_feature(imgs[0])
        return 'oke'