Update app.py

app.py

@@ -122,6 +122,9 @@ def main():
     # 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"],
@@ -134,11 +137,15 @@ def main():
     
     # 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)
+    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()
@@ -150,7 +157,7 @@ def main():
         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.")
+            st.warning("No medications selected. Please select at least one medication or use the 'Show All' option.")
         else:
             display_results(filtered_predictions, selected_site)