add message for voiting; add probability in result of /predict_code

app.py

@@ -139,11 +139,14 @@ def predict_api():
     pred_groups = classify_group(text, top_n)
     pred_codes_top = get_top_result(pred_codes)
     pred_groups_top = get_top_result(pred_groups)
+
+    message_codes = 'models agree' if pred_codes_top is not None else 'models disagree'
+    message_group = 'models agree' if pred_groups_top is not None else 'models disagree'
     result = {
         "icd10": 
-            {'result': pred_codes_top, 'details': pred_codes},
+            {'result': pred_codes_top, 'details': pred_codes, 'message': message_codes},
         "dx_group":
-            {'result': pred_groups_top, 'details': pred_groups}
+            {'result': pred_groups_top, 'details': pred_groups, 'message': message_group},
     }
     return result
 
@@ -166,7 +169,12 @@ def predict_code_api():
     top_pred_code, pred_codes, all_codes_in_group = classify_code_by_group(text, group_name, top_n)
     result = {
         "icd10":
-            {'result': top_pred_code, 'details': pred_codes, 'all_codes': all_codes_in_group}
+            {
+                'result': top_pred_code, 
+                'probability': pred_codes[top_pred_code], 
+                'details': pred_codes, 
+                'all_codes': all_codes_in_group
+            }
     }
     return result