multiselect for congress num

app.py

@@ -18,6 +18,7 @@ st.set_page_config(layout="wide", page_title="LegisQA")
 SS = st.session_state
 
 SEED = 292764
+CONGRESS_NUMBERS = [113, 114, 115, 116, 117, 118]
 CONGRESS_GOV_TYPE_MAP = {
     "hconres": "house-concurrent-resolution",
     "hjres": "house-joint-resolution",
@@ -266,7 +267,14 @@ with st.sidebar:
         )
         st.text_input("Bill ID (e.g. 118-s-2293)", key="filter_legis_id")
         st.text_input("Bioguide ID (e.g. R000595)", key="filter_bioguide_id")
-        st.text_input("Congress (e.g. 118)", key="filter_congress_num")
+#        st.text_input("Congress (e.g. 118)", key="filter_congress_num")
+        st.multiselect(
+            "Congress Numbers",
+            CONGRESS_NUMBERS,
+            default=CONGRESS_NUMBERS,
+            key="filter_congress_nums",
+        )
+
 
     with st.expander("Prompt Config"):
         st.selectbox(
@@ -304,14 +312,18 @@ def get_vectorstore_filter():
         vs_filter["legis_id"] = SS["filter_legis_id"]
     if SS["filter_bioguide_id"] != "":
         vs_filter["sponsor_bioguide_id"] = SS["filter_bioguide_id"]
-    if SS["filter_congress_num"] != "":
-        vs_filter["congress_num"] = int(SS["filter_congress_num"])
+#    if SS["filter_congress_num"] != "":
+#        vs_filter["congress_num"] = int(SS["filter_congress_num"])
+    vs_filter = {"congress_num": {"$in": SS["filter_congress_nums"]}}
     return vs_filter
 
 
 if query_submitted:
 
     vs_filter = get_vectorstore_filter()
+    with st.sidebar:
+        with st.expander("Debug vs_filter"):
+            st.write(vs_filter)
     retriever = vectorstore.as_retriever(
         search_kwargs={"k": SS["n_ret_docs"], "filter": vs_filter},
     )
@@ -377,6 +389,5 @@ if out:
         write_doc_grp(legis_id, doc_grp)
 
     with st.expander("Debug doc format"):
-
         st.text_area("formatted docs", value=format_docs(out["context"]), height=600)
-        #    st.write(json.loads(format_docs(out["context"])))
+