Update app.py

app.py

@@ -56,11 +56,11 @@ def calculate_magnitude_and_axis(j0, j45):
     if j0 != 0:
         Z = math.degrees(0.5 * math.atan(j45 / j0))
     else:
-        Z = 90  # This matches Excel's behavior when A2 (j0) is zero
+        Z = 90  # This matches Excel's behavior when j0 is zero
     
     # Apply the conditional logic
     if j0 > 0:
-        if j45 >= 0:  # Changed to >= to match Excel's behavior
+        if j45 >= 0:
             final_axis = Z
         else:
             final_axis = Z + 180
@@ -70,7 +70,7 @@ def calculate_magnitude_and_axis(j0, j45):
     # Ensure the final axis is between 0 and 180
     final_axis = final_axis % 180
     
-    return magnitude, final_axis
+    return magnitude, final_axis, Z
 
 def main():
     st.set_page_config(page_title='Astigmatism Prediction', page_icon=':eyeglasses:', layout='wide')
@@ -91,7 +91,7 @@ def main():
             new_j0, new_j45 = predict_new_j0_j45(age, aca_magnitude, aca_axis)
             
             # Calculate predicted magnitude and axis
-            predicted_magnitude, predicted_axis = calculate_magnitude_and_axis(new_j0, new_j45)
+            predicted_magnitude, predicted_axis, Z = calculate_magnitude_and_axis(new_j0, new_j45)
             
             st.success(f'Predicted Total Corneal Astigmatism Magnitude: {predicted_magnitude:.2f} D')
             st.success(f'Predicted Total Corneal Astigmatism Axis: {predicted_axis:.1f}°')
@@ -99,6 +99,7 @@ def main():
             # Display intermediate values for verification
             st.info(f'Initial J0: {initial_j0:.2f}, Initial J45: {initial_j45:.2f}')
             st.info(f'Predicted J0: {new_j0:.2f}, Predicted J45: {new_j45:.2f}')
+            st.info(f'Intermediate Z: {Z:.2f}°')
             st.info(f'ACA Axis (Degrees): {aca_axis:.2f}°, ACA Axis (Radians): {math.radians(aca_axis):.4f}')
         else:
             st.error('Please ensure all inputs are within the specified ranges.')