Heart_Disease_P.py

# -*- coding: utf-8 -*-
"""
Spyder Editor

This is a temporary script file.
"""


import pickle
import numpy as np
import streamlit as st


loaded_model=pickle.load(open("heart_disease.sav","rb"))
def heart_disease_prediction(input_data):
    input_data_np_array=np.asarray(input_data)
    input_data_reshape=input_data_np_array.reshape(1,-1)
    
    prediction=loaded_model.predict(input_data_reshape)
    print(prediction)
    
    if(prediction[0]==0):
        return 'The preson does not have a heart disease'
    else:
        return 'The preson has heart disease'

def main():
    #giving the title
    st.title('heart disease prediction Web app')
    
    #getting the inputs from the user
    Age = st.text_input('Age')
    Sex = st.text_input('Sex (1->Male, 0->Female)')
    Cp = st.text_input('Chest Pain Type(0,1,2,3)')
    trestbps = st.text_input('Resting Blood Pressure (in mm Hg on admission to the hospital)')
    chol = st.text_input('Serum Cholestral in mg/dl')
    fbs = st.text_input('Fasting Blood Sugar (1->True, 0->False)')
    restecg = st.text_input('Resting Electrocardiographic Results')
    thalach = st.text_input('Maximum Heart Rate achieved')
    exang = st.text_input('Exercise Induced Angina (1 = yes; 0 = no)')
    oldpeak = st.text_input('ST depression induced by exercise relative to rest')
    slope = st.text_input('The Slope of the peak exercise ST segment')
    ca = st.text_input('Number of major vessels (0-3) colored by flourosopy')
    thal = st.text_input('1 = Normal; 2 = Fixed defect; 3 = Reversable defect')
    
    #code for prediction
    diagnosis=""
    if st.button('Test Result: '):
        diagnosis =heart_disease_prediction(Age,Sex,Cp,trestbps,chol,fbs,restecg,thalach,exang,oldpeak,slope,ca,thal)
    st.success(diagnosis)
    
if __name__=="__main__":
    main()