Update Gallery.py

pages/Gallery.py

@@ -203,6 +203,15 @@ class GalleryApp:
             prompt_id = items['prompt_id'].unique()[0]
             note = items['note'].unique()[0]
 
+            # show source
+            if isinstance(note, str):
+                if note.isdigit():
+                    st.caption(f"`Source: civitai`")
+                else:
+                    st.caption(f"`Source: {note}`")
+            else:
+                st.caption("`Source: Parti-prompts`")
+
             # show image metadata
             image_metadatas = ['prompt_id', 'prompt', 'negativePrompt', 'sampler', 'cfgScale', 'size', 'seed']
             for key in image_metadatas:
@@ -216,7 +225,7 @@ class GalleryApp:
             # for note as civitai image id, add civitai reference
             if isinstance(note, str) and note.isdigit():
                 try:
-                    st.write('**Civitai Reference**')
+                    st.write(f'**[Civitai Reference](https://civitai.com/images/{note})**')
                     res = requests.get(f'https://civitai.com/images/{note}')
                     # st.write(res.text)
                     soup = BeautifulSoup(res.text, 'html.parser')
@@ -241,11 +250,10 @@ class GalleryApp:
         for prompt_tag in prompt_tags:
             unsafe_prompts[prompt_tag] = []
         # manually add unsafe prompts
-        unsafe_prompts['civitai'] = [375790, 366222, 295008, 256477]
-        unsafe_prompts['people'] = [53]
-        unsafe_prompts['art'] = [23]
-        unsafe_prompts['abstract'] = [10, 12]
-        unsafe_prompts['food'] = [34]
+        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.')