Update app.py

app.py

@@ -30,8 +30,8 @@ model_j45.eval()
 def calculate_initial_j0_j45(magnitude, axis):
     """Calculate initial J0 and J45 from magnitude and axis."""
     axis_rad = math.radians(axis)
-    j0 = magnitude * math.cos(2 * axis_rad)
-    j45 = magnitude * math.sin(2 * axis_rad)
+    j0 = round(magnitude * math.cos(2 * axis_rad), 2)
+    j45 = round(magnitude * math.sin(2 * axis_rad), 2)
     return j0, j45
 
 def predict_new_j0_j45(age, aca_magnitude, aca_axis):
@@ -80,12 +80,12 @@ def main():
             # Calculate predicted magnitude and axis
             predicted_magnitude, predicted_axis = calculate_magnitude_and_axis(new_j0, new_j45)
             
-            st.success(f'Predicted Total Corneal Astigmatism Magnitude: {predicted_magnitude:.4f} D')
+            st.success(f'Predicted Total Corneal Astigmatism Magnitude: {predicted_magnitude:.2f} D')
             st.success(f'Predicted Total Corneal Astigmatism Axis: {predicted_axis:.1f}°')
             
             # Display intermediate values for verification
-            st.info(f'Initial J0: {initial_j0:.4f}, Initial J45: {initial_j45:.4f}')
-            st.info(f'Predicted J0: {new_j0:.4f}, Predicted J45: {new_j45:.4f}')
+            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}')
         else:
             st.error('Please ensure all inputs are within the specified ranges.')