Update app.py

filter columns for download; allow selecting multiple agencies from settings menu

app.py

@@ -34,7 +34,7 @@ with ui.sidebar(title="Settings"):
     
     ui.input_switch("switch", "Show significant rules in plots", False)
     
-    ui.input_select("menu_agency", "Select agencies", choices=["all"] + AGENCIES, selected="all")
+    ui.input_select("menu_agency", "Select agencies", choices=["all"] + AGENCIES, selected="all", multiple=True)
     
     #ui.input_checkbox_group(
     #    "significant",
@@ -150,9 +150,30 @@ with ui.accordion(open=False):
             label="Download data as CSV",
             filename=f"rules_in_cra_window_accessed_{date.today()}.csv",
         )
-        async def download():
+        async def download(
+            output_cols: tuple | list = (
+                "document_number", 
+                "citation", 
+                "publication_date", 
+                "title", 
+                "type", 
+                "action", 
+                "abstract", 
+                "docket_ids", 
+                "json_url", 
+                "html_url", 
+                "agencies", 
+                "independent_reg_agency", 
+                "parent_slug", 
+                "subagency_slug", 
+                "president_id", 
+                "significant", 
+                "3f1_significant", 
+                "other_significant"
+                )
+            ):
             await asyncio.sleep(0.25)
-            yield filtered_df().to_csv(index=False)
+            yield filtered_df().loc[:, output_cols].to_csv(index=False)
 
 with ui.accordion(open=False):
     
@@ -187,8 +208,8 @@ def filtered_df():
         filt_df = filt_df.loc[filt_df["publication_date"] >= datetime.combine(input.start_date(), time(0, 0))]
     
     # filter agencies
-    if input.menu_agency() != "all":
-        bool_agency = [True if input.menu_agency() in agency else False for agency in filt_df["parent_slug"]]
+    if (input.menu_agency() is not None) and ("all" not in input.menu_agency()):
+        bool_agency = [True if sum(selected in agency for selected in input.menu_agency()) > 0 else False for agency in filt_df["parent_slug"]]
         filt_df = filt_df.loc[bool_agency]
     
     return filt_df