Update app.py

app.py

@@ -56,24 +56,35 @@ def predict_new_j0_j45(age, aca_magnitude, aca_axis_deg):
 def main():
     st.title('Total Corneal Astigmatism Prediction')
 
-    # Input fields without default values
+    # Initialize session state for input values if not already present
+    if 'age' not in st.session_state:
+        st.session_state.age = None
+    if 'aca_magnitude' not in st.session_state:
+        st.session_state.aca_magnitude = None
+    if 'aca_axis' not in st.session_state:
+        st.session_state.aca_axis = None
+
+    # Input fields using session state
     age = st.number_input('Enter Patient Age (18-90 Years):', 
                           min_value=18.0, max_value=90.0, 
+                          value=st.session_state.age if st.session_state.age is not None else None,
                           step=0.1, 
                           key='age')
     
     aca_magnitude = st.number_input('Enter ACA Magnitude (0-10 Diopters):', 
                                     min_value=0.0, max_value=10.0, 
+                                    value=st.session_state.aca_magnitude if st.session_state.aca_magnitude is not None else None,
                                     step=0.01, 
                                     key='aca_magnitude')
     
     aca_axis = st.number_input('Enter ACA Axis (0-180 Degrees):', 
                                min_value=0.0, max_value=180.0, 
+                               value=st.session_state.aca_axis if st.session_state.aca_axis is not None else None,
                                step=0.1, 
                                key='aca_axis')
 
     if st.button('Predict!'):
-        if age and aca_magnitude and aca_axis:  # Check if all inputs have been provided
+        if age is not None and aca_magnitude is not None and aca_axis is not None:
             if 18 <= age <= 90 and 0 <= aca_magnitude <= 10 and 0 <= aca_axis <= 180:
                 # Calculate initial J0 and J45
                 initial_j0, initial_j45 = calculate_initial_j0_j45(aca_magnitude, aca_axis)