Update app.py

app.py

@@ -127,6 +127,52 @@ def display_goal_summary(goal_summary):
         summary_table = pd.DataFrame(goal_summary)
         st.table(summary_table)
 
+# Medication Selection with Collapsible Categories
+def select_medications():
+    st.subheader("Select Medications to Display")
+    show_all = st.checkbox("Show All Medications")
+    
+    selected_medications = []
+    if not show_all:
+        # Define categories and medications
+        categories = {
+            "Bisphosphonates": [
+                "Alendronate", "Risedronate", "Ibandronate oral", 
+                "Zoledronate", "Ibandronate IV (3mg)"
+            ],
+            "RANK Ligand Inhibitors": [
+                "Denosumab", "Denosumab + Teriparatide"
+            ],
+            "Anabolic Agents": [
+                "Teriparatide", "Teriparatide + Denosumab"
+            ],
+            "Sclerostin Inhibitors": [
+                "Romosozumab", "Romosozumab + Denosumab", 
+                "Romosozumab + Alendronate", "Romosozumab + Ibandronate", 
+                "Romosozumab + Zoledronate"
+            ]
+        }
+        
+        # Create collapsible sections
+        for category, medications in categories.items():
+            with st.expander(category):
+                for med in medications:
+                    if st.checkbox(med):
+                        selected_medications.append(med)
+    else:
+        # Include all medications if "Show All" is selected
+        selected_medications = [
+            "Alendronate", "Risedronate", "Ibandronate oral", 
+            "Zoledronate", "Ibandronate IV (3mg)", "Denosumab", 
+            "Denosumab + Teriparatide", "Teriparatide", 
+            "Teriparatide + Denosumab", "Romosozumab", 
+            "Romosozumab + Denosumab", "Romosozumab + Alendronate", 
+            "Romosozumab + Ibandronate", "Romosozumab + Zoledronate"
+        ]
+    
+    return selected_medications
+
+
 # Streamlit UI
 # Main function
 def main():
@@ -173,16 +219,7 @@ def main():
     ]
     
     # 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]
+    selected_medications = select_medications()
     
     # Load constants and medication data
     medication_data = load_medication_data()