Upload app.py

app.py

@@ -104,11 +104,12 @@ def get_input_data():
             session.input_data.loc[0, 'age'] = st.slider('Age', 18, 75)
             # on_change=update_prediction(session.input_data)
         with col_sex:
-            sex = st.radio('Sex', ['Female', 'Male'])
+            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)
-            session.input_data.loc[0, 'gender'] = [1 if 'Female' in sex
-                                                   else 2][0]
+            session.input_data["gender"] = session.input_data["gender"].astype(
+                'category')
 
         # Row 2
         col_height, col_weight = st.columns(2)
@@ -135,11 +136,15 @@ 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]
-            session.input_data["cholesterol"] = session.input_data["cholesterol"].astype(
-                'category')
+            session.input_data["cholesterol"] = (session
+                                                 .input_data["cholesterol"]
+                                                 .astype('category')
+                                                 )
 
         with col_gluc:
             gluc = st.radio(
@@ -154,19 +159,19 @@ def get_input_data():
         col_alco, col_smk = st.columns(2)
         with col_alco:
             alco = st.radio('Alcohol intake', ['Yes', 'No'])
-            session.input_data.loc[0, 'alco'] = [1 if 'Yes' in alco else 2][0]
+            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'])
             session.input_data.loc[0, 'smoke'] = [1 if 'Yes' in smoke
-                                                  else 2][0]
+                                                  else 0][0]
             session.input_data["smoke"] = session.input_data["smoke"].astype(
                 bool)
 
         # Row 6
         active = st.radio('Physical activity', ['Yes', 'No'])
-        session.input_data.loc[0, 'active'] = [1 if 'Yes' in active else 2][0]
+        session.input_data.loc[0, 'active'] = [1 if 'Yes' in active else 0][0]
         session.input_data["active"] = session.input_data["active"].astype(
             bool)
 
@@ -241,7 +246,7 @@ if __name__ == "__main__":
     st.text("")
     st.text("")
 
-    # Generate poster
+    # Generate prediction
     if IS_CLICKED:
         st.write('Work in progress!')
         prediction = generate_prediction(session.input_data)