fix checkbox

app.py

@@ -97,7 +97,7 @@ class GalleryApp:
                 idx = prompts.index(prompt)
                 prompt_full = ', '.join(original_prompts[idx].split(', ')[:4]) + ', ' + prompt
             else:
-                prompt_full = st.selectbox('Select prompt', original_prompts)
+                prompt_full = st.selectbox('Select prompt', original_prompts, on_change=self.refresh)
 
             prompt_id = items[items['prompt'] == prompt_full]['prompt_id'].unique()[0]
             items = items[items['prompt_id'] == prompt_id].reset_index(drop=True)
@@ -159,7 +159,7 @@ class GalleryApp:
             elif filter == 'Unchecked':
                 items = items[items['checked'] == False].reset_index(drop=True)
 
-        with st.form(key='my_form', clear_on_submit=True):
+        with st.form(key=f'{prompt_id}', clear_on_submit=False):
             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')
@@ -169,6 +169,10 @@ class GalleryApp:
                 submit = st.form_submit_button('Reset all selections', on_click=self.reset_all, use_container_width=True)
             self.gallery_standard(items, col_num, info)
 
+    def refresh(self):
+        st.experimental_rerun()
+        st.cache_resource.clear()
+
     def reset_current_prompt(self, prompt_id):
         # reset current prompt
         self.promptBook.loc[self.promptBook['prompt_id'] == prompt_id, 'checked'] = False