import joblib
import os

MODEL_PATH = "app/ml/model.pkl"

LABELS = {
    0: "water",
    1: "vegetation",
    2: "urban",
    3: "barren"
}

model = None

if os.path.exists(MODEL_PATH):
    model = joblib.load(MODEL_PATH)



def predict_land_type(features):

    if model is None:
        return "Model not trained"

    prediction = model.predict([features])[0]

    return LABELS[prediction]