Update app.py

app.py

@@ -169,24 +169,44 @@ def main():
         unsafe_allow_html=True
     )
 
-    age = st.number_input('Enter Patient Age:', min_value=0.0, max_value=120.0, step=0.1)
-    if age < 18 or age > 90:
+    # Use session state to store input values
+    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
+
+    # Age input
+    age = st.number_input('Enter Patient Age:', min_value=18.0, max_value=90.0, step=0.1, value=st.session_state.age)
+    if age != st.session_state.age:
+        st.session_state.age = age
+    if age is not None and (age < 18 or age > 90):
         st.markdown('<p class="error">Error: Age must be between 18 and 90.</p>', unsafe_allow_html=True)
 
-    aca_magnitude = st.number_input('Enter ACA Magnitude:', min_value=0.0, max_value=20.0, step=0.1)
-    if aca_magnitude < 0 or aca_magnitude > 10:
+    # ACA Magnitude input
+    aca_magnitude = st.number_input('Enter ACA Magnitude:', min_value=0.0, max_value=10.0, step=0.1, value=st.session_state.aca_magnitude)
+    if aca_magnitude != st.session_state.aca_magnitude:
+        st.session_state.aca_magnitude = aca_magnitude
+    if aca_magnitude is not None and (aca_magnitude < 0 or aca_magnitude > 10):
         st.markdown('<p class="error">Error: ACA Magnitude must be between 0 and 10.</p>', unsafe_allow_html=True)
 
-    aca_axis = st.number_input('Enter ACA Axis:', min_value=0.0, max_value=180.0, step=0.1)
-    if aca_axis < 0 or aca_axis > 180:
+    # ACA Axis input
+    aca_axis = st.number_input('Enter ACA Axis:', min_value=0.0, max_value=180.0, step=0.1, value=st.session_state.aca_axis)
+    if aca_axis != st.session_state.aca_axis:
+        st.session_state.aca_axis = aca_axis
+    if aca_axis is not None and (aca_axis < 0 or aca_axis > 180):
         st.markdown('<p class="error">Error: ACA Axis must be between 0 and 180.</p>', unsafe_allow_html=True)
 
     if st.button('Predict!'):
-        if 18 <= age <= 90 and 0 <= aca_magnitude <= 10 and 0 <= aca_axis <= 180:
-            astigmatism = predict_astigmatism(age, aca_axis, aca_magnitude)
-            st.success(f'Predicted Total Corneal Astigmatism: {astigmatism:.4f}')
+        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:
+                astigmatism = predict_astigmatism(age, aca_axis, aca_magnitude)
+                st.success(f'Predicted Total Corneal Astigmatism: {astigmatism:.4f}')
+            else:
+                st.error('Please correct the input errors before predicting.')
         else:
-            st.error('Please correct the input errors before predicting.')
+            st.error('Please fill in all fields before predicting.')
 
 if __name__ == '__main__':
     main()