HealthSupport/app.py at master · shruti175/HealthSupport · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
from flask import Flask, render_template, request, flash, redirect
import pickle
import numpy as np
import pandas as pd


app = Flask(__name__)

def predict(values, dic):
    if len(values) == 8:
        model = pickle.load(open('models/diabetes.pkl','rb'))
        values = np.asarray(values)
        return model.predict(values.reshape(1, -1))[0]
    elif len(values) == 30:
        model = pickle.load(open('models/cancer.pkl','rb'))
        values = np.asarray(values)
        return model.predict(values.reshape(1, -1))[0]
    elif len(values) == 13:
        model = pickle.load(open('models/heart.pkl','rb'))
        input_features = [float(x) for x in request.form.values()]
        features_value = [np.array(input_features)]

        features_name = ['sex', 'age', 'cp', 'trestbps',
                         'chol', 'fbs', 'restecg',
                         'thalach', 'exang', 'oldpeak',
                         'slope', 'ca', 'thal']

        df = pd.DataFrame(features_value, columns=features_name)
        output = model.predict(df)
        if output == 0:
            pred = 1
        else:
            pred = 0
        return pred
    elif len(values) == 18:
        model = pickle.load(open('models/kidney.pkl','rb'))
        values = np.asarray(values)
        return model.predict(values.reshape(1, -1))[0]
    elif len(values) == 10:
        model = pickle.load(open('models/liver.pkl','rb'))
        values = np.asarray(values)
        return model.predict(values.reshape(1, -1))[0]

@app.route('/')
@app.route('/home')
def home():
    return render_template('index.html')

@app.route('/heart')
def heart():
    return render_template('heart.html')

@app.route('/liver')
def liver():
    return render_template('liver.html')

@app.route('/kidney')
def kidney():
    return render_template('kidney.html')

@app.route('/cancer')
def cancer():
    return render_template('cancer.html')

@app.route('/diabetes')
def diabetes():
    return render_template('diabetes.html')

def name(values, dic):
    if len(values) == 8:
        disease="Diabetes"
        return disease
    elif len(values) == 30:
        disease="Breast Cancer"
        return disease
    elif len(values) == 13:
        disease = "Heart Disease"
        return disease
    elif len(values) == 18:
        disease = "Kidney Disease"
        return disease
    elif len(values) == 10:
        disease = "Liver Disease"
        return disease

@app.route("/predict", methods = ['POST', 'GET'])
def predictPage():
    try:
        if request.method == 'POST':
            to_predict_dict = request.form.to_dict()
            to_predict_list = list(map(float, list(to_predict_dict.values())))
            pred = predict(to_predict_list, to_predict_dict)
            disease = name(to_predict_list, to_predict_dict)
    except:
        message = "Please enter valid Data"
        return render_template("index.html", message = message)

    return render_template('result.html', pred = pred,name=disease)

if __name__ == "__main__":
    app.run(debug=True)