Update app.py

app.py

@@ -7,6 +7,8 @@ from matplotlib.ticker import MultipleLocator
 from pygam import GAM
 from rapidfuzz import fuzz, utils
 
+if 'current_var' not in st.session_state:
+    st.session_state['current_var'] = None
 
 data = pd.read_parquet('var_meta.parquet')
 subject = pd.read_parquet('subject_meta.parquet')
@@ -14,7 +16,6 @@ count_meta = pd.read_parquet('count_meta.parquet')
 dt_summary1 = pd.read_parquet('dt_summary1_alpaca.parquet')
 
 def highlight(s, query):
-    # Find the index of the query substring in the string
     index = s.find(query)
     if index != -1:
         result = s[:index] + '<span style="background-color: #FFFF00">' + query + '</span>' + s[index + len(query):]
@@ -73,9 +74,6 @@ def main():
     st.markdown("Junsol Kim, Byungkyu Lee<br />Paper: https://arxiv.org/abs/2305.09620", unsafe_allow_html=True)    
     n_var_per_page = 25
     n_cards_per_row = 5
-    current_variable = None
-    if 'current_var' not in st.session_state:
-        st.session_state['current_var'] = None
 
     if st.session_state['current_var'] is None:
         col1, col2, col3 = st.columns([3, 1, 1])
@@ -123,6 +121,7 @@ def main():
     else:
         goback = st.button('Go back', on_click=show_search)
         var = st.session_state['current_var']
+        print(var)
         tab_pred = dt_summary1.loc[(dt_summary1.variable == var) & (dt_summary1.pred_type == 'rescale')]
         tab_obs = dt_summary1.loc[(dt_summary1.variable == var) & (dt_summary1.pred_type == 'obsbin')]
         tab_obs = tab_obs.merge(tab_pred[['year']], how='right').rename({'mean': 'obs_mean', 'lci': 'obs_lci', 'uci': 'obs_uci'}, axis=1)