Upload 10 files

app.py

@@ -1595,6 +1595,8 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", elem_id="main", css=CSS) as app:
         use_textual_inversion_gui.change(set_textual_inversion_prompt, [use_textual_inversion_gui, prompt_gui, neg_prompt_gui, prompt_syntax_gui], [prompt_gui, neg_prompt_gui])
 
         generate_from_image_btn_gui.click(
+            lambda: ("", "", ""), None, [series_dbt, character_dbt, prompt_gui], queue=False,
+        ).success(
             predict_tags_wd,
             [input_image_gui, prompt_gui, image_algorithms, general_threshold_gui, character_threshold_gui],
             [series_dbt, character_dbt, prompt_gui, copy_button_dbt],
@@ -1755,7 +1757,6 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", elem_id="main", css=CSS) as app:
                     image_algorithms = gr.CheckboxGroup(["Use WD Tagger"], label="Algorithms", value=["Use WD Tagger"], visible=False)
                     keep_tags = gr.Radio(label="Remove tags leaving only the following", choices=["body", "dress", "all"], value="all")
                     generate_from_image_btn = gr.Button(value="GENERATE TAGS FROM IMAGE", size="lg", variant="primary")
-                
                 with gr.Group():
                     with gr.Row():
                         input_character = gr.Textbox(label="Character tags", placeholder="hatsune miku")
@@ -1778,21 +1779,18 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", elem_id="main", css=CSS) as app:
                         dummy_np = gr.Textbox(label="Negative prompt", value="", visible=False)
                         recom_animagine = gr.Textbox(label="Animagine reccomended prompt", value="Animagine", visible=False)
                         recom_pony = gr.Textbox(label="Pony reccomended prompt", value="Pony", visible=False)
-
-                generate_btn = gr.Button(value="GENERATE TAGS", size="lg", variant="primary")
-
-                with gr.Group():
-                    output_text = gr.TextArea(label="Output tags", interactive=False, show_copy_button=True)
-                    with gr.Row():
-                        copy_btn = gr.Button(value="Copy to clipboard", size="sm", interactive=False)
-                        copy_prompt_btn = gr.Button(value="Copy to primary prompt", size="sm", interactive=False)
-
-                with gr.Group():
-                    output_text_pony = gr.TextArea(label="Output tags (Pony e621 style)", interactive=False, show_copy_button=True)
-                    with gr.Row():
-                        copy_btn_pony = gr.Button(value="Copy to clipboard", size="sm", interactive=False)
-                        copy_prompt_btn_pony = gr.Button(value="Copy to primary prompt", size="sm", interactive=False)
-
+                    generate_btn = gr.Button(value="GENERATE TAGS", size="lg", variant="primary")
+                with gr.Row():
+                    with gr.Group():
+                        output_text = gr.TextArea(label="Output tags", interactive=False, show_copy_button=True)
+                        with gr.Row():
+                            copy_btn = gr.Button(value="Copy to clipboard", size="sm", interactive=False)
+                            copy_prompt_btn = gr.Button(value="Copy to primary prompt", size="sm", interactive=False)
+                    with gr.Group():
+                        output_text_pony = gr.TextArea(label="Output tags (Pony e621 style)", interactive=False, show_copy_button=True)
+                        with gr.Row():
+                            copy_btn_pony = gr.Button(value="Copy to clipboard", size="sm", interactive=False)
+                            copy_prompt_btn_pony = gr.Button(value="Copy to primary prompt", size="sm", interactive=False)
                 description_ui()
 
         translate_input_prompt_button.click(translate_prompt, inputs=[input_general], outputs=[input_general], queue=False)
@@ -1800,6 +1798,8 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", elem_id="main", css=CSS) as app:
         translate_input_prompt_button.click(translate_prompt, inputs=[input_copyright], outputs=[input_copyright], queue=False)
 
         generate_from_image_btn.click(
+            lambda: ("", "", ""), None, [input_copyright, input_character, input_general], queue=False,
+        ).success(
             predict_tags_wd,
             [input_image, input_general, image_algorithms, general_threshold, character_threshold],
             [input_copyright, input_character, input_general, copy_input_btn],

lora_dict.json

@@ -132,6 +132,13 @@
     "https://civitai.com/models/565994",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/66feca06-6a87-4975-9c76-29b1cc6fcc23/width=450/19026855.jpeg"
   ],
+  "1990's_PC_style_SDXL_v1": [
+    "1990's PC / 1990s_(style) / oekaki / retro_artstyle / oldfilm / halftone / jaggy_lines / 1980's anime / 1980s_(style)",
+    "SDXL 1.0",
+    "1990's PC game style",
+    "https://civitai.com/models/105369",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/fcb5db04-51f8-4032-b041-2d892de65b45/width=450/22025810.jpeg"
+  ],
   "2FingersSDXL_v03": [
     "fingering pussy",
     "SDXL 1.0",
@@ -195,6 +202,13 @@
     "https://civitai.com/models/562782",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/6851f9dc-6830-4953-aacb-5bbad26de292/width=450/18831154.jpeg"
   ],
+  "7eleven_storefront_PONY_V1": [
+    "seveneleven, konbini, scenery, storefront, japan, scenery, convenience store, shop, building, outdoors, road, cityscape, city, sky, street, sign, lamppost, real world location",
+    "Pony",
+    "\u30b3\u30f3\u30d3\u30cb\u306e\u5916\u89b37 PONY",
+    "https://civitai.com/models/611755",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/2fd43b8d-cd52-4935-b015-06533825ecb0/width=450/22035002.jpeg"
+  ],
   "ALTXL_001_4": [
     "",
     "SDXL 1.0",
@@ -230,6 +244,13 @@
     "https://civitai.com/models/539225",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/1cd30680-93ac-4741-a86c-01e011ea216d/width=450/17279766.jpeg"
   ],
+  "Age_Regression_PonyXL_Test": [
+    "AgeRegression / Oversized Clothes",
+    "Pony",
+    "[PonyXL] Age Regression (Concept)",
+    "https://civitai.com/models/620987",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/f88e7690-c358-4f9c-962f-b31d4e6ea4c4/width=450/22633804.jpeg"
+  ],
   "Age_Slider_V2_alpha1_0_rank4_noxattn_last": [
     "",
     "Pony",
@@ -321,6 +342,13 @@
     "https://civitai.com/models/368121",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0134ca56-bada-42aa-8f5e-312f68239a45/width=450/8572769.jpeg"
   ],
+  "Birthing": [
+    "birth",
+    "Pony",
+    "Give Birth",
+    "https://civitai.com/models/621212",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/c9b50ab2-53e6-4a89-bf6d-929df3a3271d/width=450/22674633.jpeg"
+  ],
   "BishoujoXL_P6_lokr_V4302": [
     "",
     "Pony",
@@ -461,6 +489,13 @@
     "https://civitai.com/models/599199",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ed5f0917-fbaa-4b9c-8a17-8d34f558038d/width=450/21266476.jpeg"
   ],
+  "Comic_Manga": [
+    "comic_manga / comic / english text / speech bubble",
+    "Pony",
+    "Comic Manga PonyXL",
+    "https://civitai.com/models/610948",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/52002535-dcdf-4b72-8fae-c7a95dc252af/width=450/22212775.jpeg"
+  ],
   "Competitive_Swimsuit_XL_V1_0": [
     "competitive swimsuit",
     "SDXL 1.0",
@@ -937,6 +972,13 @@
     "https://civitai.com/models/459565",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/110e5c12-2cd7-42d8-bb15-24c7d00751d2/width=450/12716381.jpeg"
   ],
+  "Giant_Girl__KK": [
+    "kk_giantess",
+    "Pony",
+    "Giant Girl | KK_concept_01",
+    "https://civitai.com/models/607101",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/dcd20560-f3c1-407e-b966-7062ee3c9798/width=450/21993362.jpeg"
+  ],
   "Girl_Peeping": [
     "peeping / watching / female pervert",
     "Pony",
@@ -986,6 +1028,13 @@
     "https://civitai.com/models/535966",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a7f58842-13be-40e7-a65d-0abe77106382/width=450/17057493.jpeg"
   ],
+  "Hadouken": [
+    "Hadouken, Kikoken, Fireball, Energy Ball, fighting, shockwave",
+    "Pony",
+    "Hadouken",
+    "https://civitai.com/models/613938",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/255d2b55-7430-4fd5-94f7-476ea5d9246b/width=450/22207991.jpeg"
+  ],
   "Hankyu3000_SDXL_V1": [
     "hankyu3000",
     "SDXL 1.0",
@@ -1112,6 +1161,13 @@
     "https://civitai.com/models/428790",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/336d89fd-db17-483d-b408-8f99be19f9d1/width=450/11188990.jpeg"
   ],
+  "IKneel_pdxl_Incrs_v1": [
+    "on one knee, full body, looking down, clenched hands, hand on ground, arm on knee",
+    "Pony",
+    "I kneel (Kneeling Vegeta) | Concept / Pose LoRA XL",
+    "https://civitai.com/models/610071",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/3f6dcf2f-1abd-4d9e-9e5f-df370b79aeae/width=450/21922297.jpeg"
+  ],
   "Ichigo_Mashimaro_XL": [
     "Nobue Itou / Chika Itou / Miu Matsuoka / Matsuri Sakuragi / Ana Coppola",
     "SDXL 1.0",
@@ -1252,6 +1308,20 @@
     "https://civitai.com/models/579435",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/3089540a-d132-41e9-87d4-c5c8fce6d499/width=450/19912464.jpeg"
   ],
+  "Leaned_Back_Reverse_Cowgirl_pony-v2_0": [
+    "lbrc, reverse cowgirl position,hetero,sex from behind",
+    "Pony",
+    "Leaned Back Reverse Cowgirl pony",
+    "https://civitai.com/models/610465",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/8ef2c1eb-0fd3-4255-bf34-50eff3d49665/width=450/22023141.jpeg"
+  ],
+  "Leaned_Back_Reverse_Cowgirl_pony-v2_1": [
+    "lbrc, reverse cowgirl position,hetero,sex from behind",
+    "Pony",
+    "Leaned Back Reverse Cowgirl pony",
+    "https://civitai.com/models/610465",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/b4f189dc-e269-49e2-be6d-b6ce9a259353/width=450/22086403.jpeg"
+  ],
   "Lift_Up_Dress": [
     "LIFT UP DRESS / CURTSEY / FROM BELOW / FROM BACK",
     "SDXL 1.0",
@@ -1399,6 +1469,13 @@
     "https://civitai.com/models/562753",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/6f136e74-5105-4fc8-9970-8601566df7d1/width=450/18834618.jpeg"
   ],
+  "Milking_bucket_lactation-000006": [
+    "Bucket / Lactating into bucket / Breast milk",
+    "Pony",
+    "Milking bucket (lactation)",
+    "https://civitai.com/models/622072",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/bbb01058-15a6-4a72-af23-a4db99fbaedf/width=450/22720212.jpeg"
+  ],
   "MiniAXV9XLScreencap": [
     "Anime Screencap",
     "Pony",
@@ -1525,6 +1602,13 @@
     "https://civitai.com/models/540142",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/b43d779b-ccf2-4ce9-8298-2756cb13fba4/width=450/17343433.jpeg"
   ],
+  "PDXL-CheekBulge-20": [
+    "cheek bulge",
+    "Pony",
+    "PDXL-CheekBulge",
+    "https://civitai.com/models/612839",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/1ca66d39-e310-43b6-a25b-88dc076725e1/width=450/22121952.jpeg"
+  ],
   "PDXL-RetroV2": [
     "retro artstyle",
     "Pony",
@@ -1553,6 +1637,13 @@
     "https://civitai.com/models/569364",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9aeef4dd-804d-4895-baa5-43a4e60b8b86/width=450/19243523.jpeg"
   ],
+  "POV_Unaligned_Paizuri_XL": [
+    "pov unaligned paizuri",
+    "Pony",
+    "POV Asymetric Paizuri XL",
+    "https://civitai.com/models/579227",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/dbaab072-e3a3-4f61-9a93-6326887b2cec/width=450/22255802.jpeg"
+  ],
   "POV_holding_money": [
     "",
     "Pony",
@@ -1833,6 +1924,13 @@
     "https://civitai.com/models/137615",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/3879c6bc-58aa-46cb-add2-523ff21fb550/width=450/10999075.jpeg"
   ],
+  "RoboStyle": [
+    "Jed-rbt",
+    "SDXL 1.0",
+    "RoboStyle",
+    "https://civitai.com/models/612224",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/49e6eef8-a325-4543-812b-5ce77055974d/width=450/22063195.jpeg"
+  ],
   "SPF2s": [
     "Svper",
     "Pony",
@@ -1931,6 +2029,13 @@
     "https://civitai.com/models/517826",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9a382d16-ef66-493a-933c-4fcac09e8fff/width=450/15930188.jpeg"
   ],
+  "Slave_Submission_XL-000003": [
+    "submissive_female, slavesubmissionXL, BDSM / slave, humiliation, exhibitionism, public_indecency  / assisted_exposure, panty_pull, torn_clothes,  / leash_pull, leash, collar, chain, restrained / Bondage, leather black panties, black bra, crotchless, o-ring / naked_dogeza, all_fours, top_down_bottom_up, ass",
+    "Pony",
+    "Slave_Submission_XL",
+    "https://civitai.com/models/517826",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/852e29e1-e9bd-46c8-b11c-11e5c5e0448c/width=450/22226417.jpeg"
+  ],
   "Spooning_Positions": [
     " 5p00n1ng / boy and girl in spooning position, boy behind girl, 1boy, 1girl, male b1g5p00n, / l41d_0u7 / 1_l3gup, one leg up / f3741_p05, fetal position / br347_6r4b, breast grab / w41s7_6r4b / 7h16h_6r4b / 4nk13_6r4b ",
     "Pony",
@@ -1980,6 +2085,13 @@
     "https://civitai.com/models/543007",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/32bb611d-6843-4197-844e-d8ddc1aabf6e/width=450/18288434.jpeg"
   ],
+  "Standee_v0_1_test": [
+    "cardboard_cutout, white_outline, standee, photo, photo_background, / holding_tray, / mall, / cafe,",
+    "Pony",
+    "Standee / Cardboard cutout | Pony XL | Concept | Early Beta",
+    "https://civitai.com/models/611106",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/06e4cc96-0c11-4ba8-a8fc-96bb8638b81f/width=450/21999572.jpeg"
+  ],
   "Standing_DP_SDXL": [
     " 1woman, vaginal, anal, suspended congress, 2males, 2penises, standing sex, carrying, carrying partner, suspended, double penetration, mmf_threesome, standing_double_penetration / <lora:Standing_DP_SDXL-v1:0.7> / explicit female nudity, explicity male nudity,  / cum, semen, sweat, steam",
     "SDXL 1.0",
@@ -2176,6 +2288,13 @@
     "https://civitai.com/models/87049",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0faf8db6-2682-4a18-a92d-a6483b1bb289/width=450/11123630.jpeg"
   ],
+  "Urinal_Slave-000008": [
+    "Urinal slave",
+    "Pony",
+    "Urinal Slave",
+    "https://civitai.com/models/611771",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/3b21296e-0ca8-4800-b094-38b39b3ab312/width=450/22058271.jpeg"
+  ],
   "Uterus_prolapse_LoRA_for_Pony_3_0": [
     "uterus prolapse",
     "Pony",
@@ -2309,6 +2428,20 @@
     "https://civitai.com/models/548344",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/aba9189c-2986-4919-9843-ff03c2d606a8/width=450/17938715.jpeg"
   ],
+  "Yuri_Scissors_Pose": [
+    "2GIRLS, YURI SCISSORS POSE / TWINCEST / SWEAT, HEAVY BREATHING / FEMALE EJACULATION",
+    "SDXL 1.0",
+    "Yuri Scissor Tribadism",
+    "https://civitai.com/models/510404",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/db178470-6eac-408a-bc91-d58cfbc22011/width=450/15483053.jpeg"
+  ],
+  "[hikari]fellatio_invitation_pony_v1": [
+    "open mouth, tongue out, pov / fellatio gesture / palms, cum in mouth, cum on hands",
+    "Pony",
+    "[hikari]\u53e3\u4ea4\u9080\u8bf7\u59ff\u52bf/\u53e3\u4ea4\u4e8b\u540e // Fellatio invitation/After fellatio  Concept LoRA",
+    "https://civitai.com/models/308865",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0fd71659-09aa-4179-bfd2-879b623403d2/width=450/22504558.jpeg"
+  ],
   "[hikari]missionary_holding_wrist_pony_v1": [
     "missionary_holding_wrist",
     "Pony",
@@ -2484,6 +2617,13 @@
     "https://civitai.com/models/593717",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/945b3a2e-8c70-4068-b45e-1e117c3d50ca/width=450/20829978.jpeg"
   ],
+  "_sdxl-style-aclchibi": [
+    "aclchibi, chibi",
+    "SDXL 1.0",
+    "for SDXL, auxiliary for generating chibi characters (\u30c1\u30d3\u30ad\u30e3\u30e9\u751f\u6210\u88dc\u52a9\u7528) Style LoRA",
+    "https://civitai.com/models/614638",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0c6ed1ad-da10-4c9a-90ba-a1d12a8ea46c/width=450/22239895.jpeg"
+  ],
   "_sdxl-transchool-pony": [
     "transchool",
     "Pony",
@@ -2519,6 +2659,13 @@
     "https://civitai.com/models/516752",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/e48cb2fb-7aaf-4424-8ea7-acf2160290b9/width=450/17102038.jpeg"
   ],
+  "agepony8": [
+    "",
+    "Pony",
+    "Multiple XL Sliders - Age, weight, Hands, etc",
+    "https://civitai.com/models/238419",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/74053ca5-9486-44ee-94d1-9cebcbbe7250/width=450/22207105.jpeg"
+  ],
   "akifn_pony_v2": [
     "",
     "Pony",
@@ -2855,6 +3002,13 @@
     "https://civitai.com/models/576402",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/d33233e7-8705-4661-bb3b-3f27483e8f3c/width=450/19720953.jpeg"
   ],
+  "blowingbubbles_XL_v1": [
+    "blowing bubbles",
+    "SDXL 1.0",
+    "blowing bubbles / \u30b7\u30e3\u30dc\u30f3\u7389 / \u3057\u3083\u307c\u3093\u7389",
+    "https://civitai.com/models/620899",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/14e8dd8a-10ee-40af-a1a8-5c85d1be43fd/width=450/22627249.jpeg"
+  ],
   "blowjob_face": [
     "pov, fellatio, oral, half closed eyes, ",
     "Pony",
@@ -2988,6 +3142,13 @@
     "https://civitai.com/models/558208",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/8fe2a91b-96dc-4f55-8c1f-646ebf8fe050/width=450/18547446.jpeg"
   ],
+  "brightnesspony1": [
+    "",
+    "Pony",
+    "Multiple XL Sliders - Age, weight, Hands, etc",
+    "https://civitai.com/models/238419",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9ed32d61-d049-46b5-aa74-8049a8d1ece0/width=450/22207333.jpeg"
+  ],
   "broken_condom_internal_cumshot-PD-1_0-pruned": [
     "broken condom internal cumshot",
     "Pony",
@@ -3667,6 +3828,13 @@
     "https://civitai.com/models/562864",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/8b75129f-9af1-458d-8a88-0eb816df00dc/width=450/18835386.jpeg"
   ],
+  "famima_storefront_PONY_V1": [
+    "famima, konbini, scenery, storefront, convenience store, outdoors, shop, scenery, sky, cloud, day, road, blue sky, power lines, utility pole, real world location, sign, building, street, japan",
+    "Pony",
+    "\u30b3\u30f3\u30d3\u30cb\u306e\u5916\u89b3F PONY",
+    "https://civitai.com/models/611741",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a10394f4-d680-4b70-91bc-0ac807a263f6/width=450/22033610.jpeg"
+  ],
   "fanningself_XL_v1": [
     "fanning self",
     "SDXL 1.0",
@@ -3912,6 +4080,13 @@
     "https://civitai.com/models/32541",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a02818c5-568b-4da2-a6f3-0c54f3fd604d/width=450/16910321.jpeg"
   ],
+  "guided_penetration_pony_V1_0": [
+    "",
+    "Pony",
+    "\u30bb\u30eb\u30d5\u633f\u5165/guided penetration",
+    "https://civitai.com/models/619646",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/17b69aa6-a3c3-47e9-952b-f37bfacc3625/width=450/22548752.jpeg"
+  ],
   "gyaruvXL1": [
     "gyaruv",
     "SDXL 1.0",
@@ -4171,6 +4346,13 @@
     "https://civitai.com/models/579451",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ead4e9ab-9fa2-4413-860b-cfd94da88f03/width=450/19913105.jpeg"
   ],
+  "huangdongmohu_bouncing_breastslower_body_slamming": [
+    "bouncing breasts / lower body slamming",
+    "Pony",
+    "\u80f8\u90e8\u548c\u4e0b\u534a\u8eab\u52a8\u6001\u6a21\u7cca,\u63fa\u308c\u308b\u304a\u3063\u3071\u3044\u3068\u4e0b\u534a\u8eab\u306e\u885d\u6483,bouncing breasts and lower body slamming",
+    "https://civitai.com/models/611235",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/dbb8da41-c419-40f6-9a0f-d871c25acac4/width=450/22007843.jpeg"
+  ],
   "human_onahole": [
     "human onahole,torso grab / rough sex",
     "Pony",
@@ -4213,6 +4395,20 @@
     "https://civitai.com/models/121579",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/4d702d98-996e-478c-9e1d-0115852a3457/width=450/1899358.jpeg"
   ],
+  "jihanki_PONY_V1": [
+    "vending machine, outdoors, scenery, coca-cola, road, street, utility pole, power lines",
+    "Pony",
+    "\u81ea\u52d5\u8ca9\u58f2\u6a5f1 Vending Machines in Japan PONY",
+    "https://civitai.com/models/613480",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/87a4b147-eabd-4c5b-8beb-bf3c24caa99b/width=450/22167109.jpeg"
+  ],
+  "jihanki_SDXL_V1": [
+    "vending machine, outdoors, scenery, coca-cola, road, street, utility pole, power lines",
+    "SDXL 1.0",
+    "\u81ea\u52d5\u8ca9\u58f2\u6a5f1 Vending Machines in Japan SDXL",
+    "https://civitai.com/models/613468",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/4a0d7921-d8d3-4b43-9d58-14e386b5af09/width=450/22166271.jpeg"
+  ],
   "jirai_kei_XL_V1_0": [
     "jirai kei / pink shirt / black skirt / ribbon",
     "SDXL 1.0",
@@ -4304,6 +4500,13 @@
     "https://civitai.com/models/234887",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/4d94a5a1-b7d2-40b7-b864-f112b26e09a7/width=450/4642510.jpeg"
   ],
+  "kakigoori_pony_V1_0": [
+    "kakigoori / shaved ice / spoon",
+    "Pony",
+    "\u304b\u304d\u6c37/shaved ice",
+    "https://civitai.com/models/621013",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ff24f6c9-08d8-41af-a544-dccffe40aa58/width=450/22634422.jpeg"
+  ],
   "kakikoori_SDXL_V2": [
     "kakikoori_red, food, cup, shaved ice, still life",
     "SDXL 1.0",
@@ -4395,6 +4598,13 @@
     "https://civitai.com/models/236038",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/dc7c2a65-32ba-4137-b725-b5415530ee47/width=450/6441240.jpeg"
   ],
+  "kk_sexybody_02": [
+    "kk_sexybody_02, 1girl, solo, breasts, wet, nipples, cum on body, excessive pubic hair, nude",
+    "Pony",
+    "Cum Juice Shower | KK",
+    "https://civitai.com/models/613694",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/da30a983-7daa-44e2-b712-29722a43602f/width=450/22179466.jpeg"
+  ],
   "kuragecut_v1": [
     "kuragecut",
     "SDXL 1.0",
@@ -4500,6 +4710,13 @@
     "https://civitai.com/models/562800",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/7d8047bb-e2e3-4e61-bfd9-36e13ecef008/width=450/18832543.jpeg"
   ],
+  "lawson_storefront_PONY_V1": [
+    "lawson, konbini, scenery, storefront, outdoors, sky, power lines, road, car, cloud, day, shop, building, blue sky, utility pole, real world location",
+    "Pony",
+    "\u30b3\u30f3\u30d3\u30cb\u306e\u5916\u89b3L PONY",
+    "https://civitai.com/models/611720",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/e0eb5c80-2658-4c8b-a8de-0cddf22ef82f/width=450/22032802.jpeg"
+  ],
   "leafumbrella_XL_v1": [
     "leaf umbrella",
     "SDXL 1.0",
@@ -4773,6 +4990,13 @@
     "https://civitai.com/models/455382",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/f85b95e6-f83f-4de4-8289-0f825d305cdf/width=450/12484731.jpeg"
   ],
+  "mesuinu_pony_V1_0": [
+    "mesuinu / paw pose / squatting / all fours,leg lift / peeing / leash",
+    "Pony",
+    "\u96cc\u72ac\u30dd\u30fc\u30ba/paw pose",
+    "https://civitai.com/models/621240",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/bf9c5f8e-7fe7-4481-a32d-a07ac987c1fe/width=450/22655424.jpeg"
+  ],
   "microphone_stand": [
     "microphone stand,  holding microphone stand,  singing, open mouth",
     "Pony",
@@ -4927,6 +5151,13 @@
     "https://civitai.com/models/467118",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/11eced9f-d779-4138-a940-ad81a0b0454c/width=450/17635537.jpeg"
   ],
+  "nosepony1": [
+    "",
+    "Pony",
+    "Multiple XL Sliders - Age, weight, Hands, etc",
+    "https://civitai.com/models/238419",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/2a7bd500-3258-416d-956d-1cf423f82e37/width=450/22207943.jpeg"
+  ],
   "nsfw70": [
     "",
     "Pony",
@@ -5004,6 +5235,13 @@
     "https://civitai.com/models/339451",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/19736a32-8916-40f2-8163-54aca8563b22/width=450/7597428.jpeg"
   ],
+  "ok_Pony_v1": [
+    "ok sign",
+    "Pony",
+    "[SDXL&Pony] ok sign / OK\u30dd\u30fc\u30ba / \u30aa\u30c3\u30b1\u30fc\u30de\u30fc\u30af",
+    "https://civitai.com/models/304205",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/8bb2b842-d3d8-4719-8d14-fd989e39d227/width=450/22209185.jpeg"
+  ],
   "oksignXLv1": [
     "ok sign",
     "SDXL 1.0",
@@ -5221,6 +5459,13 @@
     "https://civitai.com/models/590441",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/06195e98-9e44-4d7a-b139-d1e5e4947e16/width=450/20630854.jpeg"
   ],
+  "pee_pony_V2_0": [
+    "pee,peeing",
+    "Pony",
+    "\u304a\u3057\u3063\u3053/pee",
+    "https://civitai.com/models/590441",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9b873e11-1066-41ac-94fc-da6cf7653345/width=450/22678594.jpeg"
+  ],
   "peeing_underwater-05": [
     "peeing underwater",
     "Pony",
@@ -6411,6 +6656,13 @@
     "https://civitai.com/models/537445",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ad654bb6-4edb-43a0-84ba-44e5cd96d12b/width=450/17342314.jpeg"
   ],
+  "tombstone_XL_v1": [
+    "tombstone",
+    "SDXL 1.0",
+    "tombstone / \u304a\u5893\u53c2\u308a / \u304a\u76c6 / \u5893\u77f3 / \u304a\u5bfa",
+    "https://civitai.com/models/613583",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/845dfe97-a96f-4433-8c46-e0c8cc8af437/width=450/22176279.jpeg"
+  ],
   "tongue_around_penis": [
     "tongue_around_penis, licking penis, ",
     "Pony",
@@ -6782,6 +7034,13 @@
     "https://civitai.com/models/593599",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/4813edb1-66cf-4482-9524-d03a4da30063/width=450/21514778.jpeg"
   ],
+  "watermelon_pony_V1_0": [
+    "watermelon, holding watermelon / eating",
+    "Pony",
+    "\u30b9\u30a4\u30ab/watermelon",
+    "https://civitai.com/models/621695",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/053840b5-37fb-4fee-b7f1-93ddfd1ba12f/width=450/22690182.jpeg"
+  ],
   "waterslide_XL_v1": [
     "water slide",
     "SDXL 1.0",
@@ -6789,6 +7048,13 @@
     "https://civitai.com/models/406297",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/36d7f15a-b10b-4b5b-8b27-0e87a52a1a6a/width=450/10128067.jpeg"
   ],
+  "wavy_hair_pony_V1_0": [
+    "wavy hair",
+    "Pony",
+    "\u5dfb\u304d\u9aea/wavy hair",
+    "https://civitai.com/models/621376",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/40e2acab-a2f1-41f2-9b6b-eb53ba2debad/width=450/22666093.jpeg"
+  ],
   "wedding_dress_XL_V1_0": [
     "wedding dress",
     "SDXL 1.0",

tagger.py

@@ -1,12 +1,13 @@
 from PIL import Image
 import torch
 import gradio as gr
-import spaces  # ZERO GPU
-
+import spaces
 from transformers import (
     AutoImageProcessor,
     AutoModelForImageClassification,
 )
+from pathlib import Path
+
 
 WD_MODEL_NAMES = ["p1atdev/wd-swinv2-tagger-v3-hf"]
 WD_MODEL_NAME = WD_MODEL_NAMES[0]
@@ -30,12 +31,15 @@ PEOPLE_TAGS = (
 
 
 RATING_MAP = {
+    "sfw": "safe",
     "general": "safe",
     "sensitive": "sensitive",
     "questionable": "nsfw",
     "explicit": "explicit, nsfw",
 }
 DANBOORU_TO_E621_RATING_MAP = {
+    "sfw": "rating_safe",
+    "general": "rating_safe",
     "safe": "rating_safe",
     "sensitive": "rating_safe",
     "nsfw": "rating_explicit",
@@ -49,6 +53,34 @@ DANBOORU_TO_E621_RATING_MAP = {
 }
 
 
+# https://github.com/toriato/stable-diffusion-webui-wd14-tagger/blob/a9eacb1eff904552d3012babfa28b57e1d3e295c/tagger/ui.py#L368
+kaomojis = [
+    "0_0",
+    "(o)_(o)",
+    "+_+",
+    "+_-",
+    "._.",
+    "<o>_<o>",
+    "<|>_<|>",
+    "=_=",
+    ">_<",
+    "3_3",
+    "6_9",
+    ">_o",
+    "@_@",
+    "^_^",
+    "o_o",
+    "u_u",
+    "x_x",
+    "|_|",
+    "||_||",
+]
+
+
+def replace_underline(x: str):
+    return x.strip().replace("_", " ") if x not in kaomojis else x.strip()
+
+
 def to_list(s):
     return [x.strip() for x in s.split(",") if not s == ""]
 
@@ -62,9 +94,10 @@ def list_uniq(l):
 
 
 def load_dict_from_csv(filename):
-    from pathlib import Path
     dict = {}
-    if not Path(filename).exists(): return dict
+    if not Path(filename).exists():
+        if Path('./tagger/', filename).exists(): filename = str(Path('./tagger/', filename))
+        else: return dict
     try:
         with open(filename, 'r', encoding="utf-8") as f:
             lines = f.readlines()
@@ -100,7 +133,8 @@ def character_list_to_series_list(character_list):
 
 
 def select_random_character(series: str, character: str):
-    from random import randrange
+    from random import seed, randrange
+    seed()
     character_list = list(anime_series_dict.keys())
     character = character_list[randrange(len(character_list) - 1)]
     series = anime_series_dict.get(character.split(",")[0].strip(), "")
@@ -110,7 +144,7 @@ def select_random_character(series: str, character: str):
 def danbooru_to_e621(dtag, e621_dict):
     def d_to_e(match, e621_dict):
         dtag = match.group(0)
-        etag = e621_dict.get(dtag.strip().replace("_", " "), "")
+        etag = e621_dict.get(replace_underline(dtag), "")
         if etag:
             return etag
         else:
@@ -118,7 +152,6 @@ def danbooru_to_e621(dtag, e621_dict):
     
     import re
     tag = re.sub(r'[\w ]+', lambda wrapper: d_to_e(wrapper, e621_dict), dtag, 2)
-
     return tag
 
 
@@ -134,7 +167,7 @@ def convert_danbooru_to_e621_prompt(input_prompt: str = "", prompt_type: str = "
 
     e621_dict = danbooru_to_e621_dict
     for tag in tags:
-        tag = tag.strip().replace("_", " ")
+        tag = replace_underline(tag)
         tag = danbooru_to_e621(tag, e621_dict)
         if tag in PEOPLE_TAGS:        
             people_tags.append(tag)
@@ -162,6 +195,7 @@ def translate_prompt(prompt: str = ""):
             translated_prompt = translator.translate(prompt, src='auto', dest='en').text
             return translated_prompt
         except Exception as e:
+            print(e)
             return prompt
 
     def is_japanese(s):
@@ -194,6 +228,7 @@ def translate_prompt_to_ja(prompt: str = ""):
             translated_prompt = translator.translate(prompt, src='en', dest='ja').text
             return translated_prompt
         except Exception as e:
+            print(e)
             return prompt
 
     def is_japanese(s):
@@ -219,7 +254,7 @@ def translate_prompt_to_ja(prompt: str = ""):
 def tags_to_ja(itag, dict):
     def t_to_j(match, dict):
         tag = match.group(0)
-        ja = dict.get(tag.strip().replace("_", " "), "")
+        ja = dict.get(replace_underline(tag), "")
         if ja:
             return ja
         else:
@@ -238,7 +273,7 @@ def convert_tags_to_ja(input_prompt: str = ""):
     tags_to_ja_dict = load_dict_from_csv('all_tags_ja_ext.csv')
     dict = tags_to_ja_dict
     for tag in tags:
-        tag = tag.strip().replace("_", " ")
+        tag = replace_underline(tag)
         tag = tags_to_ja(tag, dict)
         out_tags.append(tag)
     
@@ -287,6 +322,7 @@ def insert_recom_prompt(prompt: str = "", neg_prompt: str = "", type: str = "Non
 def load_model_prompt_dict():
     import json
     dict = {}
+    path = 'model_dict.json' if Path('model_dict.json').exists() else './tagger/model_dict.json'
     try:
         with open('model_dict.json', encoding='utf-8') as f:
             dict = json.load(f)
@@ -365,7 +401,7 @@ def remove_specific_prompt(input_prompt: str = "", keep_tags: str = "all"):
 
     group_dict = tag_group_dict
     for tag in tags:
-        tag = tag.strip().replace("_", " ")
+        tag = replace_underline(tag)
         if tag in PEOPLE_TAGS:
             people_tags.append(tag)
         elif is_necessary(tag, keep_tags, group_dict):
@@ -393,7 +429,7 @@ def sort_taglist(tags: list[str]):
     rating_set = set(DANBOORU_TO_E621_RATING_MAP.keys()) | set(DANBOORU_TO_E621_RATING_MAP.values())
 
     for tag in tags:
-        tag = tag.strip().replace("_", " ")
+        tag = replace_underline(tag)
         if tag in PEOPLE_TAGS:
             people_tags.append(tag)
         elif tag in rating_set:
@@ -494,12 +530,13 @@ def predict_tags(image: Image.Image, general_threshold: float = 0.3, character_t
         output_series_tag = output_series_list[0]
     else:
         output_series_tag = ""
-    return output_series_tag, ", ".join(character.keys()), prompt, gr.update(interactive=True),
+    return output_series_tag, ", ".join(character.keys()), prompt, gr.update(interactive=True)
 
 
-def predict_tags_wd(image: Image.Image, input_tags: str, algo: list[str], general_threshold: float = 0.3, character_threshold: float = 0.8):
+def predict_tags_wd(image: Image.Image, input_tags: str, algo: list[str], general_threshold: float = 0.3,
+                     character_threshold: float = 0.8, input_series: str = "", input_character: str = ""):
     if not "Use WD Tagger" in algo and len(algo) != 0:
-        return "", "", input_tags, gr.update(interactive=True),
+        return input_series, input_character, input_tags, gr.update(interactive=True)
     return predict_tags(image, general_threshold, character_threshold)