add force refresh

app.py

@@ -220,7 +220,7 @@ class GalleryApp:
                 sub_selecters = st.columns([1, 1, 1, 1])
 
                 if 'default_weights' not in st.session_state:
-                    st.session_state.default_weights = [1.0, 1.0, 1.0]
+                    st.session_state.default_weights = [1.0, 0.8, 0.2]
 
                 with sub_selecters[0]:
                     clip_weight = st.number_input('Clip Score Weight', min_value=-100.0, max_value=100.0, value=st.session_state.default_weights[0], step=0.1, help='the weight for normalized clip score')
@@ -374,12 +374,15 @@ 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()
 
-            gallery_place = st.empty()
-
-            with gallery_place.container():
+            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]: