add manual prompt filter

data/curation.json

@@ -0,0 +1,14 @@
+{
+  "abstract": [1,3],
+	"animal": [],
+	"architecture": [],
+	"art": [],
+	"artifact": [],
+	"food": [],
+	"illustration": [39],
+	"people": [49,50,51,62,54,56,48,60],
+	"produce & plant": [],
+	"scenery": [],
+	"vehicle": [],
+	"world knowledge": [84,83,85]
+}

pages/Gallery.py

@@ -1,3 +1,4 @@
+import itertools
 import json
 import os
 import requests
@@ -187,7 +188,7 @@ class GalleryApp:
 
         # set focus tag and prompt index if exists
         if st.session_state.gallery_focus['tag'] is None:
-             tag_focus_idx = 5
+             tag_focus_idx = 0
         else:
             tag_focus_idx = prompt_tags.index(st.session_state.gallery_focus['tag'])
 
@@ -591,6 +592,12 @@ def load_hf_dataset(show_NSFW=False):
     # add column to record current row index
     promptBook.loc[:, 'row_idx'] = promptBook.index
 
+    # apply curation filter
+    prompt_to_hide = json.load(open('./data/curation.json', 'r'))
+    prompt_to_hide = list(itertools.chain.from_iterable(prompt_to_hide.values()))
+    print('prompt to hide: ', prompt_to_hide)
+    promptBook = promptBook[~promptBook['prompt_id'].isin(prompt_to_hide)].reset_index(drop=True)
+
     # apply a nsfw filter
     if not show_NSFW:
         promptBook = promptBook[promptBook['norm_nsfw'] <= 0.8].reset_index(drop=True)