add force refresh

app.py

@@ -374,25 +374,23 @@ class GalleryApp:
         if safety_check:
             items, info, col_num = self.selection_panel_2(items)
             # self.gallery_standard(items, col_num, info)
-            buttons_space = st.container()
-            gallery_space = st.empty()
-
-            with gallery_space.container():
-                self.gallery_standard(items, col_num, info)
-
-            with buttons_space:
-                st.button('Refresh', on_click=gallery_space.empty, use_container_width=True, type='primary')
-
-            # with st.form(key=f'{prompt_id}', 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')
-            #     with buttons[1]:
-            #         submit = st.form_submit_button('Reset current prompt', on_click=self.reset_current_prompt, kwargs={'prompt_id': prompt_id} , use_container_width=True)
-            #     with buttons[2]:
-            #         submit = st.form_submit_button('Reset all selections', on_click=self.reset_all, use_container_width=True)
-            #
-            #     self.gallery_standard(items, col_num, info)
+
+            with st.form(key=f'{prompt_id}', clear_on_submit=True):
+                # buttons = st.columns([1, 1, 1])
+                buttons_space = st.container()
+                gallery_space = st.empty()
+                # with buttons[0]:
+                #     submit = st.form_submit_button('Save selections', on_click=self.save_checked, use_container_width=True, type='primary')
+                # with buttons[1]:
+                #     submit = st.form_submit_button('Reset current prompt', on_click=self.reset_current_prompt, kwargs={'prompt_id': prompt_id} , use_container_width=True)
+                # with buttons[2]:
+                #     submit = st.form_submit_button('Reset all selections', on_click=self.reset_all, use_container_width=True)
+
+                with gallery_space.container():
+                    self.gallery_standard(items, col_num, info)
+
+                with buttons_space:
+                    st.form_submit_button('Refresh', on_click=gallery_space.empty, use_container_width=True, type='primary')
 
     def reset_current_prompt(self, prompt_id):
         # reset current prompt