app.py

import streamlit as st
import time
import requests

'''
# StrokeModel front

This front queries the [stroke model API](https://stroke-t44zttdg3q-ew.a.run.app/predict)

'''

with st.form(key='params_for_api'):

    gender = st.selectbox('Gender',('Male', 'Female', 'Other'))
    age = st.number_input('Age', min_value=0, max_value=100, step=1, help='insert value between 0 and 100')
    hypertension = st.selectbox('Hypertension', (0,1))
    st.caption("0 - I do not have hypertension \n\n 1 - I have hypertension")
    heart_disease = st.selectbox('Heart Disease', (0,1))
    st.caption("0 - I do not have a heart disease \n\n 1 - I have a heart disease")
    ever_married = st.selectbox('Married',('Yes', 'No'))
    work_type = st.selectbox('Work Type',('Government', 'Private', 'Children', 'Self-employed', 'Never worked'))
    Residence_type = st.selectbox('Residence Type',('Urban', 'Rural'))
    avg_glucose_level = st.number_input('Average Glucose Level', value=80, min_value=70, max_value=240, step=1, help='insert value between 70 and 240')
    st.caption("If unsure, please move to next category.")
    bmi = st.number_input('BMI', value=22.0, min_value=15.0, max_value=45.0, step=0.01, help='insert value between 15.0 and 45.0')
    st.caption("If unsure, please move to next category.")
    smoking_status = st.selectbox('Smoking Status',('Unknown','Formerly Smoked','Never Smoked','Smokes'))

    submit = st.form_submit_button('Make Prediction')

    if submit:
        params = dict(
            id=1,
            gender=gender,
            age=age,
            hypertension=hypertension,
            heart_disease=heart_disease,
            ever_married=ever_married,
            work_type=work_type,
            Residence_type=Residence_type,
            avg_glucose_level=avg_glucose_level,
            bmi=bmi,
            smoking_status=smoking_status
        )

        progress_text = "Predicting..."
        my_bar = st.progress(0, text=progress_text)

        for percent_complete in range(100):
            time.sleep(0.02)
            my_bar.progress(percent_complete + 1, text=progress_text)


        stroke_api_url = 'https://stroke-t44zttdg3q-ew.a.run.app/predict'
        response = requests.get(stroke_api_url, params=params)

        prediction = response.json()

        pred = prediction['stroke']

        st.header(f'Chance of stroke: {pred}')
        st.subheader('If 0, your answers indicate that you do not have an increased risk of having a stroke.')
        st.subheader('If 1, your answers indicate that you have an increased risk of having a stroke.')
        st.subheader('Open the cell below for more information:')
        with st.expander("**Open for more info**"):
            st.write("""
                Whether or not you have an increased risk of having a stroke, it is
                important to know how you and others may be affected by your lifestyles.
                Check out the following page on the National Heart, Lung and Blood Institute
                for more information: https://www.nhlbi.nih.gov/health/stroke/causes
            """)