import helper.optimizer_def
from keras.models import load_model
import numpy as np

CONVERT_CLASS_PRED_TO_NAME = ["Common Rust", "Gray Leaf Spot", "Leaf Blight"]

def fetch_model(opt_name: str):
    return load_model(f'models/{opt_name}-model-001.h5', safe_mode=False)

def preprocess_image(img):
    img = img.resize((224, 224))
    img_array = np.array(img)
    img_array = np.expand_dims(img_array, axis=0)
    return img_array

def classify_image(model, img):
    prediction = model.predict(img)
    predicted_class = np.argmax(prediction, axis=1)[0]

    return CONVERT_CLASS_PRED_TO_NAME[predicted_class]