Update Gallery.py

pages/Gallery.py

@@ -82,11 +82,11 @@ class GalleryApp:
                 sub_selecters = st.columns([1, 1, 1, 1])
 
                 with sub_selecters[0]:
-                    clip_weight = st.number_input('Clip Score Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[0], step=0.1, help='the weight for normalized clip score')
+                    clip_weight = st.number_input('Clip Score Weight', min_value=-100.0, max_value=100.0, value=1.0, step=0.1, help='the weight for normalized clip score')
                 with sub_selecters[1]:
-                    mcos_weight = st.number_input('Dissimilarity Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[1], step=0.1, help='the weight for m(eam) s(imilarity) q(antile) score for measuring distinctiveness')
+                    mcos_weight = st.number_input('Dissimilarity Weight', min_value=-100.0, max_value=100.0, value=0.8, step=0.1, help='the weight for m(eam) s(imilarity) q(antile) score for measuring distinctiveness')
                 with sub_selecters[2]:
-                    pop_weight = st.number_input('Popularity Weight', min_value=-100.0, max_value=100.0, value=st.session_state.score_weights[2], step=0.1, help='the weight for normalized popularity score')
+                    pop_weight = st.number_input('Popularity Weight', min_value=-100.0, max_value=100.0, value=0.2, step=0.1, help='the weight for normalized popularity score')
 
                 items.loc[:, 'weighted_score_sum'] = round(items[f'norm_clip'] * clip_weight + items[f'norm_mcos'] * mcos_weight + items[
                     'norm_pop'] * pop_weight, 4)
@@ -94,13 +94,13 @@ class GalleryApp:
                 continue_idx = 3
 
                 # save latest weights
-                st.session_state.score_weights[0] = clip_weight
-                st.session_state.score_weights[1] = mcos_weight
-                st.session_state.score_weights[2] = pop_weight
+                st.session_state.score_weights[0] = round(clip_weight, 2)
+                st.session_state.score_weights[1] = round(mcos_weight, 2)
+                st.session_state.score_weights[2] = round(pop_weight, 2)
 
             # select threshold
             with sub_selecters[continue_idx]:
-                nsfw_threshold = st.number_input('NSFW Score Threshold', min_value=0.0, max_value=1.0, value=st.session_state.score_weights[3], step=0.01, help='Only show models with nsfw score lower than this threshold, set 1.0 to show all images')
+                nsfw_threshold = st.number_input('NSFW Score Threshold', min_value=0.0, max_value=1.0, value=0.8, step=0.01, help='Only show models with nsfw score lower than this threshold, set 1.0 to show all images')
                 items = items[items['norm_nsfw'] <= nsfw_threshold].reset_index(drop=True)
 
             # save latest threshold
@@ -214,6 +214,7 @@ class GalleryApp:
         st.write('This is a gallery of images generated by the models')
 
         prompt_tags, tag, prompt_id, items = self.sidebar()
+        items, info, col_num = self.selection_panel(items)
 
         # add safety check for some prompts
         safety_check = True
@@ -223,16 +224,13 @@ class GalleryApp:
             unsafe_prompts[prompt_tag] = []
         # manually add unsafe prompts
         unsafe_prompts['world knowledge'] = [83]
-        # unsafe_prompts['art'] = [23]
         unsafe_prompts['abstract'] = [1, 3]
-        # unsafe_prompts['food'] = [34]
 
         if int(prompt_id.item()) in unsafe_prompts[tag]:
             st.warning('This prompt may contain unsafe content. They might be offensive, depressing, or sexual.')
-            safety_check = st.checkbox('I understand that this prompt may contain unsafe content. Show these images anyway.', key=f'{prompt_id}')
+            safety_check = st.checkbox('I understand that this prompt may contain unsafe content. Show these images anyway.', key=f'safety_{prompt_id}')
 
         if safety_check:
-            items, info, col_num = self.selection_panel(items)
 
             if 'selected_dict' in st.session_state:
                 # st.write('checked: ', str(st.session_state.selected_dict.get(prompt_id, [])))