Update app.py

app.py

@@ -16,7 +16,21 @@ explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 def main_func(HighBP, HighChol, CholCheck, BMI, Smoker, Stroke, HeartDiseaseorAttack, PhysActivity, Fruits, Veggies, HvyAlcoholConsump, AnyHealthcare, NoDocbcCost, GenHlth, MentHlth, PhysHlth, DiffWalk, Sex, Age, Education, Income):
     new_row = pd.DataFrame.from_dict({'HighBP': HighBP, 'HighChol': HighChol, 'CholCheck': CholCheck, 'BMI': BMI, 'Smoker': Smoker, 'Stroke': Stroke, 'HeartDiseaseorAttack': HeartDiseaseorAttack, 'PhysActivity':PhysActivity, 'Fruits':Fruits, 'Veggies':Veggies, 'HvyAlcoholConsump': HvyAlcoholConsump, 'AnyHealthcare': AnyHealthcare, 'NoDocbcCost': NoDocbcCost, 'GenHlth': GenHlth, 'MentHlth': MentHlth, 'PhysHlth': PhysHlth, 'DiffWalk': DiffWalk, 'Sex': Sex, 'Age': Age, 'Education': Education, 'Income': Income}, 
                                      orient = 'index').transpose()
-    
+    HighBP = map_HighBP(HighBP_Checked)
+    HighChol = map_HighChol(HighChol_Checked)
+    CholCheck = map_CholCheck(CholCheck_Checked)
+    Smoker = map_Smoker(Smoker_Checked)
+    Stroke = map_Stroke(Stroke_Checked)
+    HeartDiseaseorAttack = map_HeartDiseaseorAttack(HeartDiseaseorAttack_Checked)
+    PhysActivity = map_PhysActivity(PhysActivity_Checked)
+    Fruits = map_Fruits(Fruits_Checked)
+    Veggies = map_VeggiesCheck(Veggies_Checked)
+    HvyAlcoholConsump = map_HvyAlcoholConsump(HvyAlcoholConsump_Checked)
+    AnyHealthcare = map_AnyHealthcare(AnyHealthcare_Checked)
+    NoDocbcCost = map_NoDocbcCost(NoDocbcCost_Checked)
+    DiffWalk = map_DiffWalk(DiffWalk_Checked)
+
+
     prob = loaded_model.predict_proba(new_row)
     shap_values = explainer(new_row)
   #  plot = shap.force_plot(shap_values[0], matplotlib=True, figsize=(30,30), show=False)
@@ -160,6 +174,13 @@ with gr.Blocks(title=title) as demo:
     numeric_value = map_HighBP(HighBP_Checked)
     print(numeric_value)
 
+    def map_HeartDiseaseorAttack(value):
+        mapping = {'No': 0, 'Yes': 1}
+        return mapping[value]
+    HeartDiseaseorAttack_Checked = 'Yes' 
+    numeric_value = map_HeartDiseaseorAttack(HeartDiseaseorAttack_Checked)
+    print(numeric_value)
+
     def map_PhysActivity(value):
         mapping = {'No': 0, 'Yes': 1}
         return mapping[value]