"""Program to predict the car type from price, kmpl and bhp.
"""
import joblib
import sys

model = joblib.load("model.pkl")
scaleX = joblib.load("scalex.pkl")
labely = joblib.load("labely.pkl")

def predict(price, kmpl, bhp):
    """Predicts the model of a car using its price, milage and horse power.
    """
    print("predict", price, kmpl, bhp)
    X = scaleX.transform([[price, kmpl, bhp]])
    y = model.predict(X)
    return labely.inverse_transform(y)[0]

def square(x):
    return x*x

def main():
    price, kmpl, bhp = sys.argv[1:]
    price = float(price)
    kmpl = float(kmpl)
    bhp = float(bhp)
    print(predict(price, kmpl, bhp))
    
if __name__ == "__main__":
    main()