Update streamlit_app_LDA.py

streamlit_app_LDA.py

@@ -17,6 +17,8 @@ from MIND_utils import df_to_self_states_json, element_short_desc_map
 st.set_page_config(layout="wide")
 st.title("Prototypical Self-States via Topic Modeling")
 
+uploaded_file = st.file_uploader("Upload your own data file (CSV)", type="csv")
+
 st.header("Model Parameters")
 lda_document_is = st.radio("A 'Document' in the topic model will correspond to a:", ("self-state", "segment"))
 num_topics = st.slider("Number of Topics", min_value=2, max_value=20, value=5)
@@ -44,12 +46,11 @@ show_long_elements = st.checkbox("Show full element name")
 # ---------------------------
 # Load Data
 # ---------------------------
-# You can also allow users to upload their file via st.file_uploader.
-# @st.cache(allow_output_mutation=True)
-def load_data():
-    return pd.read_csv("clean_annotations_safe.csv")
+@st.cache_data
+def load_data(csv):
+    return pd.read_csv(csv)
 
-df = load_data()
+df = load_data(uploaded_file or "clean_annotations_safe.csv")
 
 # ---------------------------
 # Preprocess Data: Build Documents