Update app.py

app.py

@@ -62,7 +62,7 @@ class GalleryApp:
                             )
 
                             # show checkbox
-                            self.promptBook.loc[items.iloc[idx+j]['row_idx'].item(), 'checked'] = st.checkbox('Select', value=self.promptBook.loc[items.iloc[idx+j]['row_idx'].item(), 'checked'] , key=f'select_{idx+j}')
+                            self.promptBook.loc[items.iloc[idx+j]['row_idx'].item(), 'checked'] = st.checkbox('Select', value=self.promptBook.loc[items.iloc[idx+j]['row_idx'].item(), 'checked'], key=f'select_{idx+j}')
 
                             # show selected info
                             for key in info:
@@ -159,7 +159,7 @@ class GalleryApp:
             elif filter == 'Unchecked':
                 items = items[items['checked'] == False].reset_index(drop=True)
 
-        with st.form(key='my_form'):
+        with st.form(key='my_form', clear_on_submit=True):
             buttons = st.columns([1, 1, 1])
             with buttons[0]:
                 submit = st.form_submit_button('Save selections', on_click=self.save_checked, use_container_width=True, type='primary')
@@ -184,6 +184,10 @@ class GalleryApp:
         dataset = load_dataset('NYUSHPRP/ModelCofferMetadata', split='train')
         # get checked images
         checked_info = self.promptBook['checked']
+        # print('checked_info: ', checked_info)
+        # for d in checked_info:
+        #     if d is True:
+        #         print('checked')
 
         if 'checked' in dataset.column_names:
             dataset = dataset.remove_columns('checked')
@@ -209,7 +213,7 @@ if __name__ == '__main__':
         st.session_state.promptBook = pd.DataFrame(load_dataset('NYUSHPRP/ModelCofferMetadata', split='train'))
         # add 'checked' column to promptBook if not exist
         if 'checked' not in st.session_state.promptBook.columns:
-            st.session_state.promptBook['checked'] = False
+            st.session_state.promptBook.loc[:, 'checked'] = False
 
         st.session_state.images = load_from_disk(os.path.join(os.getcwd(), 'data', 'promptbook'))
         # st.session_state.images = load_dataset('NYUSHPRP/ModelCofferPromptBook', split='train', streaming=True)