Update appbackup.py

appbackup.py

@@ -119,14 +119,47 @@ def main():
         format="%.3f"
     )
     
+    # Medication Selection with Checkboxes
+    st.subheader("Select Medications to Display")
+    
+    # Add "Show All" Option
+    show_all = st.checkbox("Show All Medications")
+    
+    # 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 = []
+    if not show_all:
+        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)
+    else:
+        # If "Show All" is checked, include all medications
+        selected_medications = [med for row in medication_rows for med in row]
+    
     # 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 or use the 'Show All' option.")
+        else:
+            display_results(filtered_predictions, selected_site)
 
 if __name__ == "__main__":
     main()