Update app.py

app.py

@@ -10,6 +10,9 @@ from rapidfuzz import fuzz, utils
 if 'current_var' not in st.session_state:
     st.session_state['current_var'] = None
 
+if 'search_ongoing' not in st.session_state:
+    st.session_state['search_ongoing'] = False
+
 data = pd.read_parquet('var_meta.parquet')
 subject = pd.read_parquet('subject_meta.parquet')
 count_meta = pd.read_parquet('count_meta.parquet')
@@ -32,6 +35,7 @@ def search_data(query, search_criteria, search_filter):
         data_sub = data.loc[data.var_name.isin(selected_vars)]
     if len(query) == 0:
         return data_sub
+    data_sub['original_var_name'] = data_sub['var_name']
     query = query.lower()
     d1 = data_sub.loc[data_sub['var_name'].str.lower().str.contains(query)]
     d2 = data_sub.loc[data_sub['var_description'].str.lower().str.contains(query)]
@@ -67,6 +71,9 @@ def show_data(var):
 def show_search():
     st.session_state['current_var'] = None
 
+def start_search():
+    st.session_state['search_ongoing'] = True
+
 # Main function to run the Streamlit app
 def main():
     st.set_page_config(page_title="AI-Augmented Surveys: Browse Retrodictions", layout="wide")
@@ -83,10 +90,10 @@ def main():
             search_criteria = st.selectbox("Criteria", ['All', 'Variable Names', 'Variable Descriptions', 'Survey Questions', 'GSS Tags'])
         with col3:
             search_filter = st.multiselect("Filter by GSS tags", count_meta['option'])
-        search_button = st.button("Search")
+        search_button = st.button("Search", on_click=start_search)
 
         # Perform search when the search button is clicked
-        if (search_query or search_button):
+        if st.session_state['search_ongoing']:
             results = search_data(search_query, search_criteria, search_filter).reset_index(drop=True)
             if results.empty:
                 st.warning('No variables found.')
@@ -113,7 +120,7 @@ def main():
                         st.markdown(f"*{row['var_description'].strip()}*", unsafe_allow_html=True)
                         st.markdown(f"Survey question: {row['question']}", unsafe_allow_html=True)
                         st.markdown(f"GSS tags: {row['subject']}", unsafe_allow_html=True)
-                        button_list.append(st.button('View this variable', key=row['var_name'], on_click=show_data, args=(row['var_name'], )))
+                        button_list.append(st.button('View this variable', key=row['var_name'], on_click=show_data, args=(row['original_var_name'], )))
                 for i in range(len(button_list)):
                     if button_list[i]:
                         current_variable = str(results.loc[i, 'var_name'])