import streamlit as st
import pickle
import numpy as np



  
try:
    with open('heart_disease_model.pkl', 'rb') as file:
        loaded_model = pickle.load(file)
    print("Model loaded successfully!")
except FileNotFoundError:
    print("Error: The file 'heart_disease_model.pkl' was not found.")
except Exception as e:
    print("An error occurred while loading the model:", e)



def predict_cancer(input_data):
    input_data_reshaped = np.asarray(input_data).reshape(1, -1)
    prediction = loaded_model.predict(input_data_reshaped)
    return prediction[0]


def main():
    st.set_page_config(layout="centered")
    st.title('Heart disease Prediction')
    st.markdown('Enter the values for the input features:')
    feature_names = [
        'age',
        'sex',
        'cp',
        'trestbps  ',
        'chol',
		'fbs',
        'restecg',
        'thalach',
        'exang',
        'oldpeak',
		'ca',
		'thal',
		'target',
		
    ]
    col1, col2 = st.columns([2, 1])
    with col1:
        input_data = []
        for feature_name in feature_names:
            input_value = st.number_input(feature_name, step=0.01,)
            input_data.append(float(input_value))

        if st.button('Predict'):
            prediction = predict_cancer(input_data)
            if prediction == 0:
                st.error('The Person has Heart Disease')
            else:
                st.success('The Person does not have Heart Disease')
    with col2:
        st.markdown('**IF YES**')
        st.write('the person has heart disease, and he sholuld want to take medical care immediately. '
                 'It requires prompt medical attention and treatment.')

        st.markdown('**NO**')
        st.write('the person do not have any heart related issues.  '
                 'and no need to worry.')

        st.markdown('**NOTE:** The prediction provided by this app is for informational purposes only. '
                    'It is not a substitute for professional medical advice or diagnosis.')

    st.markdown('<br>', unsafe_allow_html=True)


if __name__ == '__main__':
    main()