Update app.py

app.py

@@ -282,7 +282,11 @@ class Predictor:
         # print(general_res_dict)
 
         # Prepare tag names for prompt (without confidence, no escaping yet)
-        general_tag_names = [x[0].replace('_', ' ') for x in sorted(general_res_dict.items(), key=lambda x: x[1], reverse=True)]
+        # general_tag_names = [x[0].replace('_', ' ') for x in sorted(general_res_dict.items(), key=lambda x: x[1], reverse=True)]
+        general_tag_names = [
+            x[0] if x[0] in kaomojis else x[0].replace('_', ' ')
+            for x in sorted(general_res_dict.items(), key=lambda x: x[1], reverse=True)
+        ]
         character_tag_names = [x[0].replace('_', ' ') for x in sorted(character_res_dict.items(), key=lambda x: x[1], reverse=True)]
 
         return general_tag_names, character_tag_names
@@ -375,7 +379,11 @@ def update_allowed_tags_set(selected_categories: list):
             # Get tags where the category column has a '1'
             # Convert to lower case for case-insensitive matching
             tags_in_category = _danbooru_tag_classifier_df[_danbooru_tag_classifier_df[category] == 1].index.tolist()
-            allowed_tags_set.update(tag.replace('_', ' ').lower() for tag in tags_in_category)
+            # allowed_tags_set.update(tag.replace('_', ' ').lower() for tag in tags_in_category)
+            allowed_tags_set.update(
+                tag.lower() if tag in kaomojis else tag.replace('_', ' ').lower()
+                for tag in tags_in_category
+            )
 
     _preprocessed_allowed_tags_set = allowed_tags_set
     _last_selected_tag_categories = selected_categories