import streamlit as st
import pickle
from PIL import Image


ok = Image.open('ok.jpg')
not_ok = Image.open("notok.jpg")


st.write ('''

## Heart Health Predictor
''')

age = st.number_input('Enter your age', min_value= 0, max_value= 100, step = 1)

gender = st.selectbox(
    'Gender',
    ('Male', 'Female'))

if gender == "Male":
    gender_value = 1
else:
    gender_value = 0

bp = st.number_input('Enter your BP', value= 120, step = 1)
cholestrol = st.number_input('Enter your Cholestrol', value= 100, step = 1)


data = [[age, gender_value, bp, cholestrol]]

#load models
# @st.cache_data(allow_output_mutation = True)
def cache_model(path):
    model = pickle.load(open(path, 'rb'))
    return (model)

heart_model = cache_model("model.pkl")

if st.button('Predict'):
    result = heart_model.predict(data)
    if result == 1:
        st.image(not_ok)
        st.write("## Please visit a cardiologist")

    else:
        st.image(ok)
        st.write("## Great!, You are at no risk")