small refresh logic change

.gitignore

@@ -8,3 +8,4 @@ Thumbs.db
 secret.sh
 
 .idea/
+playgrounds/

pages/Gallery.py

@@ -362,12 +362,9 @@ class GalleryApp:
                         # else:
                         #     checked = False
 
-                    selection_control_button = st.empty()
-                    report_control_button = st.empty()
-
                     if image_status == 'report':
                         st.warning('You have reported this image')
-                        unreport = report_control_button.form_submit_button('Withdraw report', use_container_width=True, type='secondary', on_click=self.image_selection_control, args=(item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'deselect'))
+                        unreport = st.form_submit_button('Withdraw report', use_container_width=True, type='secondary', on_click=self.image_selection_control, args=(item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'deselect'))
 
                     else:
                         if image_status is None:
@@ -377,19 +374,21 @@ class GalleryApp:
 
                         if image_status == 'select' or image_status == 'reselect' or image_status is None:
                             # deselect = st.button('Deselect', key=f'select_{item["prompt_id"]}_{item["modelVersion_id"]}', use_container_width=True)
-                            deselect = selection_control_button.form_submit_button('Deselect', use_container_width=True, on_click=self.image_selection_control, args=(item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'deselect'))
+                            deselect = st.form_submit_button('Deselect', use_container_width=True, on_click=self.image_selection_control, args=(item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'deselect'))
 
-                            st.info(
-                                "Image selected. **Click 'Check out selections ➡️' on top to see all selected images**.")
 
                         elif image_status =='deselect':
                             # select = st.button('Select', key=f'select_{item["prompt_id"]}_{item["modelVersion_id"]}', use_container_width=True, type='primary')
-                            reselect = selection_control_button.form_submit_button('Reselect', use_container_width=True, type='primary', on_click=self.image_selection_control, args=(item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'reselect'))
+                            reselect = st.form_submit_button('Reselect', use_container_width=True, type='primary', on_click=self.image_selection_control, args=(item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'reselect'))
 
-                        report = report_control_button.form_submit_button('⚠️Report', use_container_width=True, type='secondary',
+                        report = st.form_submit_button('⚠️Report', use_container_width=True, type='secondary',
                                                        on_click=self.image_selection_control, args=(
                             item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'report'))
 
+                        if image_status == 'select' or image_status == 'reselect' or image_status is None:
+                            st.info(
+                                "Image selected. **Click 'Check out selections ➡️' on top to see all selected images**.")
+
                     # st.write(item)
                     # infos = ['model_name', 'modelVersion_name', 'model_download_count', 'clip_score', 'mcos_score',
                     #          'nsfw_score']