Limit results and add download button

app.py

@@ -1,5 +1,6 @@
 from collections import Counter
 import streamlit as st
+import pandas as pd
 import srsly
 
 
@@ -17,9 +18,11 @@ st.sidebar.write(
     "A complete list of MeSH tags can be found here https://meshb.nlm.nih.gov/treeView"
 )
 st.sidebar.write("The grants data can be found https://www.threesixtygiving.org/")
+st.sidebar.header("Parameters")
 st.sidebar.write(
     "The model used to tag grants is https://huggingface.co/Wellcome/WellcomeBertMesh"
 )
+nb_results = st.sidebar.slider("Number of results to display", value=20, min_value=1, max_value=100)
 
 if "grants" not in st.session_state:
     st.session_state["grants"] = list(srsly.read_jsonl("tagged_grants.jsonl"))
@@ -43,4 +46,7 @@ if "results" in st.session_state:
                 if tag_i < len(most_common_tags):
                     tag = most_common_tags[tag_i]
                     st.button(tag, on_click=search, kwargs={"query": tag})
-    st.table(st.session_state["results"])
+    results = st.session_state["results"]
+    st.caption(f"Found {len(results)}. Displaying {nb_results}")
+    st.download_button("Download results", data=pd.DataFrame(results).to_csv(), file_name="results.csv", mime="text/csv")
+    st.table(results[:nb_results])