from funcy import rcompose

from image_prediction.utils import get_logger

logger = get_logger()


class PredictionModelHandle:
    """Simplifies usage of ModelHandle instances for prediction purposes."""

    def __init__(self, model_handle):
        self.__predict = rcompose(model_handle.prep_images, model_handle.model.predict)

    def predict(self, *args, **kwargs):
        return self.__predict(*args, **kwargs)

    def __call__(self, *args, **kwargs):
        logger.debug("PredictionModelHandle.predict")
        return self.predict(*args, **kwargs)