add filter to summary page

css/style.css

@@ -10,7 +10,7 @@ div.row-widget.stRadio > div[role="radiogroup"] > label[data-baseweb="radio"]  {
     padding-bottom: 3px;
     margin: 4px 0px;
     border-radius: 0;
-    border-bottom: 2px solid whitesmoke;
+    border-bottom: 2px solid rgba(169, 169, 169, 0.3);
     transition: border-bottom-color 0.2s ease 0s;
 }
 

pages/Summary.py

@@ -1,3 +1,4 @@
+import json
 import os
 
 import datasets
@@ -35,6 +36,12 @@ class DashboardApp:
                 if back_to_ranking:
                     switch_page('ranking')
 
+            with st.form('overall_feedback'):
+                feedback = st.text_area('Please leave your comments here.', key='comment')
+                submit_feedback = st.form_submit_button('Submit Feedback')
+                if submit_feedback:
+                    print(feedback)
+
         return tag
 
     def leaderboard(self, tag, db_table):
@@ -129,9 +136,21 @@ class DashboardApp:
                 # with st.expander(f'**{icon} {model_name}, [{modelVersion_name}](https://civitai.com/models/{model_id}?modelVersionId={modelVersion_id})**, Ranking Score: {winning_times}'):
                 with st.expander(f'Show Images'):
                     images = self.promptBook[self.promptBook['modelVersion_id'] == modelVersion_id]['image_id'].values
+
+                    safety_check = st.checkbox('Include potentially unsafe or offensive images', value=False, key=modelVersion_id)
+                    unsafe_prompts = json.load(open('data/unsafe_prompts.json', 'r'))
+                    # merge dict values into one list
+                    unsafe_prompts = [item for sublist in unsafe_prompts.values() for item in sublist]
+                    unsafe_images = self.promptBook[self.promptBook['prompt_id'].isin(unsafe_prompts)]['image_id'].values
+
+                    if not safety_check:
+                        # exclude unsafe prompts from images
+                        images = [image for image in images if image not in unsafe_images]
+
                     # st.write(f'### Images generated with {icon} {model_name}, {modelVersion_name}')
                     col_num = 4
                     image_cols = st.columns(col_num)
+
                     for j in range(len(images)):
                         with image_cols[j % col_num]:
                             image = f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{images[j]}.png"