explore dataframe filtering

appStore/target.py

@@ -140,11 +140,11 @@ def target_display():
 
 
             st.title("Auto Filter Dataframes in Streamlit")
-            filtered_df = filter_dataframe(hits)
-            filtered_df = filtered_df[filtered_df.keep == True]
+            # filtered_df = filter_dataframe(hits)
+            # filtered_df = filtered_df[filtered_df.keep == True]
             # st.write('Explore the data')
             # AgGrid(hits)
-            df_xlsx = to_excel(df,filtered_df)
+            df_xlsx = to_excel(df,st.session_state['target_hits'])
             
             with st.sidebar:
                 st.write('-------------')
@@ -173,7 +173,8 @@ def filter_dataframe(df: pd.DataFrame) -> pd.DataFrame:
     modify = st.checkbox("Add filters")
 
     if not modify:
-        return df
+        st.session_state['target_hits'] = df
+        return 
 
     df = df.copy()
 
@@ -234,7 +235,7 @@ def filter_dataframe(df: pd.DataFrame) -> pd.DataFrame:
                 if user_text_input:
                     df = df[df[column].str.contains(user_text_input)]
         df['keep'] = True
-        st.data_editor(
+        df = st.data_editor(
                         df,
                         column_config={
                             "keep": st.column_config.CheckboxColumn(
@@ -245,8 +246,9 @@ def filter_dataframe(df: pd.DataFrame) -> pd.DataFrame:
                         disabled=list(set(df.columns) - {'keep'}),
                         hide_index=True,
                     )
+        st.session_state['target_hits'] = df
         
-    return df
+    return
 
 
 # df = pd.read_csv(