app.py

import streamlit as st
import pandas as pd
import joblib


st.title("Heart Disease Prediction")

gender = st.radio("Select Your Gender",["Male","Female"])

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

age = st.slider("Select Your Age")

cigsPerDay = st.number_input("Enter No of Ciggarettes Per Day")

BPMeds = st.radio("Do you take BP Medicines",["True","False"])

if BPMeds == "True":
    BPMeds = 1.0
else:
    BPMeds = 0.0

prevStroke = st.radio("Do you have any prevalent stroke",["True","False"])

if prevStroke=="True":
    prevStroke=1
else:
    prevStroke=0

prevalentHyp = st.radio("Do you have any prevalent Hypertension",["True","False"])

if prevalentHyp=="True":
    prevalentHyp=1
else:
    prevalentHyp=0

diabetes = st.radio("Do you have diabetes",["True","False"])

if diabetes=="True":
    diabetes=1
else:
    diabetes=0

totchol = st.number_input("Enter your Cholestrol Level")

sysBP = st.number_input("Enter your systolic Blood Pressure Level")

diaBP = st.number_input("Enter your Diastolic Blood Pressure Level")

bmi = st.number_input("Enter your BMI")

heart_rate = st.number_input("Enter your Heart Beats per minute")

if st.button("Predict"):
    model = joblib.load("heart_model.h5")
    prediction = model.predict([[gender,age,cigsPerDay,BPMeds,prevStroke,prevalentHyp,diabetes,totchol,sysBP,diaBP,bmi,heart_rate]])
    if prediction[0] ==0:
        prediction = "You are Not At Risk of getting Heart Disease"
        st.success(prediction)
    else:
        prediction = "You are at Risk of getting Heart Disease"
        st.success(prediction)