# -*- coding: utf-8 -*-

import scipy.io as sio

from app.core.config import settings

q_learning_models = {}
ahu_supply_air_temp_set_dict = {}


async def load_q_learning_model():
    base_path = settings.PROJECT_DIR
    summer_model_path = (
        f"{base_path}/app/resources/ml_models/equipment/fcu/q_learning/net_1_summer.mat"
    )
    winter_model_path = (
        f"{base_path}/app/resources/ml_models/equipment/fcu/q_learning/net_1_winter.mat"
    )
    q_learning_models.update({"summer": sio.loadmat(summer_model_path)["net"][0, 0][0]})
    q_learning_models.update({"winter": sio.loadmat(winter_model_path)["net"][0, 0][0]})