Update app.py

app.py

@@ -56,67 +56,59 @@ def predict_new_j0_j45(age, aca_magnitude, aca_axis_deg):
 def main():
     st.title('Total Corneal Astigmatism Prediction')
 
-    # Initialize session state for input values if not already present
-    if 'age' not in st.session_state:
-        st.session_state.age = ''
-    if 'aca_magnitude' not in st.session_state:
-        st.session_state.aca_magnitude = ''
-    if 'aca_axis' not in st.session_state:
-        st.session_state.aca_axis = ''
-
-    # Input fields with session state
+    # Input fields without default values
     age = st.number_input('Enter Patient Age (18-90 Years):', 
                           min_value=18.0, max_value=90.0, 
-                          value=st.session_state.age, 
                           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, 
                                     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, 
                                step=0.1, 
                                key='aca_axis')
 
     if st.button('Predict!'):
-        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)
-            
-            # Predict new J0 and J45 using the models
-            new_j0, new_j45 = predict_new_j0_j45(age, aca_magnitude, aca_axis)
-            
-            # Calculate predicted magnitude and axis
-            predicted_magnitude = math.sqrt(new_j0**2 + new_j45**2)
-            predicted_axis = 0.5 * math.degrees(math.atan2(new_j45, new_j0))
-            if predicted_axis < 0:
-                predicted_axis += 180
-            
-            # Display results in green success boxes
-            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'''
-            #Input ACA - Magnitude: {aca_magnitude:.2f} D, Axis: {aca_axis:.1f}°
-            #Initial J0: {initial_j0:.2f}, Initial J45: {initial_j45:.2f}
-            #Predicted J0: {new_j0:.2f}, Predicted J45: {new_j45:.2f}
-            #''')
-            
-            # Additional debugging information (optional)
-            #if st.checkbox('Show detailed model inputs and outputs'):
-                #st.subheader('Debugging Information:')
-                #st.write(f"Input tensor for J0: {torch.tensor([[age, aca_axis, initial_j0]], dtype=torch.float32)}")
-                #st.write(f"Input tensor for J45: {torch.tensor([[age, aca_axis, initial_j45]], dtype=torch.float32)}")
-                #st.write(f"Raw J0 output: {new_j0}")
-                #st.write(f"Raw J45 output: {new_j45}")
+        if age and aca_magnitude and aca_axis:  # Check if all inputs have been provided
+            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)
+                
+                # Predict new J0 and J45 using the models
+                new_j0, new_j45 = predict_new_j0_j45(age, aca_magnitude, aca_axis)
+                
+                # Calculate predicted magnitude and axis
+                predicted_magnitude = math.sqrt(new_j0**2 + new_j45**2)
+                predicted_axis = 0.5 * math.degrees(math.atan2(new_j45, new_j0))
+                if predicted_axis < 0:
+                    predicted_axis += 180
+                
+                # Display results in green success boxes
+                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'''
+                #Input ACA - Magnitude: {aca_magnitude:.2f} D, Axis: {aca_axis:.1f}°
+                #Initial J0: {initial_j0:.2f}, Initial J45: {initial_j45:.2f}
+                #Predicted J0: {new_j0:.2f}, Predicted J45: {new_j45:.2f}
+                #''')
+                
+                # Additional debugging information (optional)
+                #if st.checkbox('Show detailed model inputs and outputs'):
+                    #st.subheader('Debugging Information:')
+                    #st.write(f"Input tensor for J0: {torch.tensor([[age, aca_axis, initial_j0]], dtype=torch.float32)}")
+                    #st.write(f"Input tensor for J45: {torch.tensor([[age, aca_axis, initial_j45]], dtype=torch.float32)}")
+                    #st.write(f"Raw J0 output: {new_j0}")
+                    #st.write(f"Raw J45 output: {new_j45}")
+            else:
+                st.error('Please ensure all inputs are within the specified ranges.')
         else:
-            st.error('Please ensure all inputs are within the specified ranges.')
+            st.error('Please fill in all input fields before predicting.')
 
 if __name__ == '__main__':
     main()