Update app.py

app.py

@@ -119,14 +119,40 @@ def main():
         format="%.3f"
     )
     
+    # Medication Selection with Checkboxes
+    st.subheader("Select Medications to Display")
+    
+    # Define medications by rows
+    medication_rows = [
+        ["Alendronate", "Risedronate", "Ibandronate oral"],
+        ["Zoledronate", "Ibandronate IV (3mg)"],
+        ["Denosumab", "Denosumab + Teriparatide"],
+        ["Teriparatide", "Teriparatide + Denosumab"],
+        ["Romosozumab", "Romosozumab + Denosumab", "Romosozumab + Alendronate"],
+        ["Romosozumab + Ibandronate", "Romosozumab + Zoledronate"]
+    ]
+    
+    # Create checkboxes for each row
+    selected_medications = []
+    for row in medication_rows:
+        cols = st.columns(len(row))
+        for col, med in zip(cols, row):
+            if col.checkbox(med):
+                selected_medications.append(med)
+    
     # Load constants and medication data
     medication_data = load_medication_data()
     constants = REGRESSION_CONSTANTS.get(selected_site, {})
 
-    # Generate and display predictions
+    # Generate and display predictions for selected medications
     if st.button("Predict"):
-        predictions = generate_predictions(medication_data, site, bmd_patient, constants['mu'], constants['sigma'])
-        display_results(predictions, selected_site)
+        all_predictions = generate_predictions(medication_data, site, bmd_patient, constants['mu'], constants['sigma'])
+        filtered_predictions = [pred for pred in all_predictions if pred['Drug'] in selected_medications]
+        
+        if not filtered_predictions:
+            st.warning("No medications selected. Please select at least one medication to display predictions.")
+        else:
+            display_results(filtered_predictions, selected_site)
 
 if __name__ == "__main__":
     main()