Update app.py

app.py

@@ -98,9 +98,15 @@ def main():
     # Gender Selection
     gender = st.selectbox("Gender", ["Female"])
     
-    # Location (Site) Selection
-    site_options = ['Lumbar spine (L1-L4)', 'Femoral Neck', 'Total Hip']
-    site = st.selectbox("Select Region (Site)", site_options)
+    # Location (Site) Selection with Mapping
+    site_mapping = {
+        'Lumbar spine (L1-L4)': 'LS',
+        'Femoral Neck': 'FN',
+        'Total Hip': 'TH'
+    }
+    site_options = list(site_mapping.keys())
+    selected_site = st.selectbox("Select Region (Site)", site_options)
+    site = site_mapping[selected_site]  # Map to the actual value in the dataset
     
     # Input patient data
     bmd_patient = st.number_input(
@@ -115,16 +121,17 @@ def main():
     st.write("Loaded Medication Data:")
     st.write(medication_data.head())  # Display the loaded data for verification
     
-    constants = REGRESSION_CONSTANTS.get(site, {})
+    constants = REGRESSION_CONSTANTS.get(selected_site, {})
     if not constants:
-        st.warning(f"No constants available for the selected site: {site}")
+        st.warning(f"No constants available for the selected site: {selected_site}")
         return
 
     # Generate and display predictions
     if st.button("Predict"):
         st.write("Prediction started...")
         predictions = generate_predictions(medication_data, site, bmd_patient, constants['mu'], constants['sigma'])
-        display_results(predictions, site)
+        display_results(predictions, selected_site)
+
 
 if __name__ == "__main__":
     main()