Upload app.py

app.py

@@ -106,8 +106,9 @@ def get_input_data():
         with col_sex:
             session.input_data.loc[0, 'gender'] = st.radio(
                 'Sex', ['Female', 'Male'])
-            st.write('<style>div.row-widget.stRadio > div{flex-direction:row;}\
-                     </style>', unsafe_allow_html=True)
+            st.write('<style> div[data-testid=column] > div > div > div > \
+                     div.stRadio > div{flex-direction: row;}</style>',
+                     unsafe_allow_html=True)
             session.input_data["gender"] = session.input_data["gender"].astype(
                 'category')
 
@@ -140,8 +141,6 @@ def get_input_data():
         with col_chole:
             cholest = st.radio(
                 'Cholesterol', ['Normal', 'Above normal', 'Well above normal'])
-            st.write('<style>div.row-widget.stRadio > div{flex-direction:\
-                     column;}</style>', unsafe_allow_html=True)
             session.input_data.loc[0, 'cholesterol'] = [
                 1 if 'Normal' in cholest else 2 if 'Above normal' in cholest
                 else 3][0]
@@ -165,12 +164,12 @@ def get_input_data():
         # Row 5
         col_alco, col_smk = st.columns(2)
         with col_alco:
-            alco = st.radio('Alcohol intake', ['Yes', 'No'])
+            alco = st.radio('Alcohol intake', ['Yes', 'No'], 'No')
             session.input_data.loc[0, 'alco'] = [1 if 'Yes' in alco else 0][0]
             session.input_data["alco"] = session.input_data["alco"].astype(
                 bool)
         with col_smk:
-            smoke = st.radio('Smoking', ['Yes', 'No'])
+            smoke = st.radio('Smoking', ['Yes', 'No'], 'No')
             session.input_data.loc[0, 'smoke'] = [1 if 'Yes' in smoke
                                                   else 0][0]
             session.input_data["smoke"] = session.input_data["smoke"].astype(
@@ -200,18 +199,17 @@ def generate_prediction(input_data):
 
     -param input_data: DataFrame with input data
 
-    -return: prediction of cardiovascular disease probability
+    -return: predicted probability of having a cardiovascular disease
     """
-
-    current_time = datetime.now().strftime("%H:%M:%S")
-    st.write("Current Time =", current_time)
-    st.write(input_data)
-    st.write(input_data.dtypes)
-
-    prediction = MODEL.predict(input_data)
-    st.write(prediction)
-
-    return prediction
+    # prediction = MODEL.predict(input_data)
+    proba = MODEL.predict_proba(input_data)
+    positive_proba = round(proba[1]*100)
+    st.markdown(f"<h1 style='text-align: center; color: grey;'>\
+                {positive_proba}</h1>",
+                unsafe_allow_html=True)
+    st.progress(positive_proba,
+                text='predicted cardiovascular disease probability.')
+    return proba
 
 
 ###############################
@@ -245,7 +243,6 @@ if __name__ == "__main__":
 
     # Get inputs
     session.input_data = get_input_data()
-    st.write(session.input_data)
 
     # Create button to trigger poster generation
     buffer1, col1, buffer2 = st.columns([1.3, 1, 1])
@@ -258,9 +255,6 @@ if __name__ == "__main__":
     if IS_CLICKED:
         st.write('Work in progress!')
         prediction = generate_prediction(session.input_data)
-        # current_time = datetime.now().strftime("%H:%M:%S")
-        # st.write("Current Time =", current_time)
-        # st.write(session.input_data)
 
     st.text("")
     st.text("")