Updated code

app.py

@@ -160,32 +160,55 @@ test_dict = {
 def calculate_z_score(test_score, mean, std_dev):
     return (test_score - mean) / std_dev
 
-def z_score_calculator(value, norm_mean, norm_sd):
+def z_score_calculator(value, norm_mean, norm_sd, invert=False):
     z_value = (value - norm_mean) / norm_sd
-    stanine_value = round(1.25 * z_value + 5.5)
+    if invert:
+        z_value = -z_value
+    stanine_value = zscore_to_stanine(z_value)
     z_score = round(z_value, 2)
     return z_score, stanine_value
 
-def test_calculator(age, education, values, test):
+def zscore_to_stanine(zscore):
+    if zscore <= -1.76:
+        return 1
+    elif zscore <= -1.26:
+        return 2
+    elif zscore <= -0.76:
+        return 3
+    elif zscore <= -0.26:
+        return 4
+    elif zscore <= 0.25:
+        return 5
+    elif zscore <= 0.75:
+        return 6
+    elif zscore <= 1.25:
+        return 7
+    elif zscore <= 1.75:
+        return 8
+    else:
+        return 9
+
+
+def test_calculator(age, education, values, test, invert=False):
         if age <= 60 and education <= 12:
             norm_mean = test_dict[test]["low_age_low_education"]["mean"]
             norm_sd = test_dict[test]["low_age_low_education"]["std"]
-            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd)
+            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd, invert=invert)
             return norm_mean, norm_sd, z_score, stanine_value
         elif age <= 60 and education > 12:
             norm_mean = test_dict[test]["low_age_high_education"]["mean"]
             norm_sd = test_dict[test]["low_age_high_education"]["std"]
-            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd)
+            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd, invert=invert)
             return norm_mean, norm_sd, z_score, stanine_value
         elif age > 60 and education <= 12:
             norm_mean = test_dict[test]["high_age_low_education"]["mean"]
             norm_sd = test_dict[test]["high_age_low_education"]["std"]
-            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd)
+            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd, invert=invert)
             return norm_mean, norm_sd, z_score, stanine_value
         elif age > 60 and education > 12:
             norm_mean = test_dict[test]["high_age_high_education"]["mean"]
             norm_sd = test_dict[test]["high_age_high_education"]["std"]
-            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd)
+            z_score, stanine_value = z_score_calculator(values, norm_mean, norm_sd, invert=invert)
             return norm_mean, norm_sd, z_score, stanine_value
         else:
             print("missing value/ wrong format")
@@ -267,6 +290,8 @@ def generate_graph(test_dict):
 
     for test in reversed(list(test_dict.keys())):
         #print("the test is", test)
+
+        # check if values are not empty
         ax.scatter(test_dict[test][4], i, s=50, color='black', label=test)
         i = i + 1
 
@@ -412,7 +437,9 @@ if st.button("Calculate Z-Score"):
     animal_mean, animal_std, animal_z, animal_stanine = test_calculator(age, education_level, animal, "animal")
     verb_mean, verb_std, verb_z, verb_stanine = test_calculator(age, education_level, verb, "verb")
     repetition_mean, repetition_std, repetition_z, repetition_stanine = test_calculator(age, education_level, repetition, "repetition")
-    months_backward_mean, months_backward_std, months_backward_z, months_backward_stanine = test_calculator(age, education_level, months_backward, "months_backward")
+    months_backward_mean, months_backward_std, months_backward_z, months_backward_stanine = test_calculator(age, education_level, months_backward, "months_backward", invert=True)
+    # months_backward_z = months_backward_z * -1
+    # print("the months backw z is: ", months_backward_z)
     logicogrammatic_mean, logicogrammatic_std, logicogrammatic_z, logicogrammatic_stanine = test_calculator(age, education_level, logicogrammatic, "logicogrammatic")
     inference_mean, inference_std, inference_z, inference_stanine = test_calculator(age, education_level, inference, "inference")
     reading_speed_mean, reading_speed_std, reading_speed_z, reading_speed_stanine = test_calculator(age, education_level, reading_speed, "reading_speed")