from image_prediction.classifier.classifier import Classifier
from image_prediction.config import CONFIG
from image_prediction.estimator.adapter.adapter import EstimatorAdapter
from image_prediction.locations import MLRUNS_DIR
from image_prediction.model_loader.loader import ModelLoader
from image_prediction.model_loader.loaders.mlflow import MlflowConnector
from image_prediction.redai_adapter.mlflow import MlflowModelReader


class Pipeline:

    def __init__(self):
        model_loader = ModelLoader(MlflowConnector(MlflowModelReader(MLRUNS_DIR)))
        model = model_loader.load_model(CONFIG.service.run_id)
        classes = model_loader.load_classes(CONFIG.service.run_id)
        classifier = Classifier(EstimatorAdapter(model), classes)