utils/prediction.py

def predict_drug(age, sex, blood_pressure, cholesterol, na_to_k_ratio, pipe):
    """
        Predict drugs based on patient features.

    :param age:
    :param sex:
    :param blood_pressure:
    :param cholesterol:
    :param na_to_k_ratio:
    :return: str: Predicted drug label.

    """

    features = [age, sex, blood_pressure, cholesterol, na_to_k_ratio]
    predicted_drug = pipe.predict([features])[0]

    label = f"Predicted Drug: {predicted_drug}"
    return label