application.py

"""
comorbidity-ml: a webapplication to the users the chance of morbidity
"""
from flask import Flask, render_template, request,session
import joblib
import numpy as np
import pandas as pd
from checker import check_logged_in, home_logged_in, wrong_info


# Loads the machine learning model
df = pd.read_csv("data/conditions.csv")
lr_cv_model = joblib.load("data/comorbid-trained-model.pkl")

application = Flask(__name__)


application.secret_key = "notasecret"


@application.route("/")
@home_logged_in
def home():
    """
    Function: home
    Input: none
    Returns: The main page of the webapplication
    """
    return render_template("home.html")


@application.route("/login")
def login():
    """
    Function: login
    Input: none
    Returns: The login page of the webapplication
    """
    return render_template("login.html")


@application.route("/check", methods=['POST'])
@wrong_info
def check():
    """
    Function: check
    Input: none
    Returns: Check username and password
    """
    session['logged_in'] = True
    return render_template("home.html")


@application.route("/logout", methods=['POST'])
def logout():
    """
    Function: logout
    Input: none
    Returns: logout session
    """
    session.pop('logged_in')
    return render_template("login.html")


@application.route("/survey")
@check_logged_in
def survey():
    """
    Function: entry
    Input: none
    Returns: The survey form to predict morbidity
    """
    return render_template("survey.html")


@application.route('/predict', methods=['POST'])
def result():
    """
    Function: result
    Input: none
    Returns: The result page of the user's prediction
    """
    letters = ['age', 'sex', 'smoking', 'alcohol', 'hypertension',
               'diabetes', 'rheuma', 'dementia', 'cancer', 'copd',
               'asthma', 'chd', 'ccd', 'cnd', 'cld', 'ckd', 'aids']
    pred_date = [int(request.form[letter]) for letter in letters]
    arr = np.array([pred_date])
    pred = lr_cv_model.predict(arr)
    return render_template('results.html', data=pred)


@application.route('/charts')
@check_logged_in
def charts() -> render_template:
    """
    Function: charts
    Input: none
    Returns: Returns the charts page in the webapplication
    """
    return render_template('charts.html')


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