Upload 3 files

app.py

@@ -93,47 +93,47 @@ def main():
                     output_text_pony = gr.TextArea(label="Output tags (Pony e621 style)", interactive=False, show_copy_button=True)
                     copy_btn_pony = gr.Button(value="Copy to clipboard", size="sm", interactive=False)
 
-        translate_input_prompt_button.click(translate_prompt, inputs=[input_general], outputs=[input_general])
-        translate_input_prompt_button.click(translate_prompt, inputs=[input_character], outputs=[input_character])
-        translate_input_prompt_button.click(translate_prompt, inputs=[input_copyright], outputs=[input_copyright])
+        translate_input_prompt_button.click(translate_prompt, [input_general], [input_general])
+        translate_input_prompt_button.click(translate_prompt, [input_character], [input_character])
+        translate_input_prompt_button.click(translate_prompt, [input_copyright], [input_copyright])
 
         generate_from_image_btn.click(
             predict_tags_wd,
-            inputs=[input_image, input_general, image_algorithms, general_threshold, character_threshold],
-            outputs=[
+            [input_image, input_general, image_algorithms, general_threshold, character_threshold],
+            [
                 input_copyright,
                 input_character,
                 input_general,
                 copy_input_btn,
             ],
-        ).then(
+        ).success(
             predict_tags_fl2_sd3,
-            inputs=[input_image, input_general, image_algorithms],
-            outputs=[input_general],
-        ).then(
-            remove_specific_prompt, inputs=[input_general, keep_tags], outputs=[input_general],
-        ).then(
-            convert_danbooru_to_e621_prompt, inputs=[input_general, input_tag_type], outputs=[input_general],
-        ).then(
-            insert_recom_prompt, inputs=[input_general, dummy_np, recom_prompt], outputs=[input_general, dummy_np],
+            [input_image, input_general, image_algorithms],
+            [input_general],
+        ).success(
+            remove_specific_prompt, [input_general, keep_tags], [input_general],
+        ).success(
+            convert_danbooru_to_e621_prompt, [input_general, input_tag_type], [input_general],
+        ).success(
+            insert_recom_prompt, [input_general, dummy_np, recom_prompt], [input_general, dummy_np],
         )
-        copy_input_btn.click(compose_prompt_to_copy, inputs=[input_character, input_copyright, input_general], outputs=[input_tags_to_copy]).then(
-            gradio_copy_text, inputs=[input_tags_to_copy], js=COPY_ACTION_JS,
+        copy_input_btn.click(compose_prompt_to_copy, [input_character, input_copyright, input_general], [input_tags_to_copy]).success(
+            gradio_copy_text, [input_tags_to_copy], js=COPY_ACTION_JS,
         )
 
         generate_btn.click(
             prompt_enhancer,
-            inputs=[input_character, input_copyright, input_general, prompt_enhancer_model],
-            outputs=[output_text, copy_btn, copy_btn_pony],
-        ).then(
-            convert_danbooru_to_e621_prompt, inputs=[output_text, tag_type], outputs=[output_text_pony],
-        ).then(
-            insert_recom_prompt, inputs=[output_text, dummy_np, recom_animagine], outputs=[output_text, dummy_np],
-        ).then(
-            insert_recom_prompt, inputs=[output_text_pony, dummy_np, recom_pony], outputs=[output_text_pony, dummy_np],
+            [input_character, input_copyright, input_general, prompt_enhancer_model],
+            [output_text, copy_btn, copy_btn_pony],
+        ).success(
+            convert_danbooru_to_e621_prompt, [output_text, tag_type], [output_text_pony],
+        ).success(
+            insert_recom_prompt, [output_text, dummy_np, recom_animagine], [output_text, dummy_np],
+        ).success(
+            insert_recom_prompt, [output_text_pony, dummy_np, recom_pony], [output_text_pony, dummy_np],
         )
-        copy_btn.click(gradio_copy_text, inputs=[output_text], js=COPY_ACTION_JS)
-        copy_btn_pony.click(gradio_copy_text, inputs=[output_text_pony], js=COPY_ACTION_JS)
+        copy_btn.click(gradio_copy_text, [output_text], js=COPY_ACTION_JS)
+        copy_btn_pony.click(gradio_copy_text, [output_text_pony], js=COPY_ACTION_JS)
 
     ui.launch()
 

tagger.py

@@ -81,6 +81,14 @@ def character_list_to_series_list(character_list):
     return output_series_tag
 
 
+def select_random_character(series: str, character: str):
+    from random import randrange
+    character_list = list(anime_series_dict.keys())
+    character = character_list[randrange(len(character_list) - 1)]
+    series = anime_series_dict.get(character.split(",")[0].strip(), "")
+    return series, character
+
+
 def danbooru_to_e621(dtag, e621_dict):
     def d_to_e(match, e621_dict):
         dtag = match.group(0)
@@ -415,23 +423,19 @@ def predict_tags(image: Image.Image, general_threshold: float = 0.3, character_t
     results = {
         wd_model.config.id2label[i]: float(logit.float()) for i, logit in enumerate(logits)
     }
-
     # rating, character, general
     rating, character, general = postprocess_results(
         results, general_threshold, character_threshold
     )
-
     prompt = gen_prompt(
         list(rating.keys()), list(character.keys()), list(general.keys())
     )
-
     output_series_tag = ""
     output_series_list = character_list_to_series_list(character.keys())
     if output_series_list:
         output_series_tag = output_series_list[0]
     else:
         output_series_tag = ""
-
     return output_series_tag, ", ".join(character.keys()), prompt, gr.update(interactive=True),