Add application file

app.py

@@ -84,6 +84,23 @@ def generate(
     upscaler_strength: float = 0.55,
     upscale_by: float = 1.5,
     add_quality_tags: bool = True,
+    add_danbooru_tags: bool = False,
+    rating_tags: str = "general",
+    copyright_tags_list: list[str] = [],
+    character_tags_list: list[str] = [],
+    general_tags: str = "",
+    ban_tags: str = "",
+    do_cfg: bool = False,
+    cfg_scale: float = 1.5,
+    negative_tags: str = "",
+    total_token_length: str = "long",
+    max_new_tokens: int = 128,
+    min_new_tokens: int = 0,
+    temperature: float = 1.0,
+    top_p: float = 1.0,
+    top_k: int = 20,
+    num_beams: int = 1,
+    # model_backend: str = "Default",
     progress=gr.Progress(track_tqdm=True),
 ):
     generator = utils.seed_everything(seed)
@@ -96,6 +113,11 @@ def generate(
 
     prompt = utils.add_wildcard(prompt, wildcard_files)
 
+    generated_tags_animagine = ""
+    if add_danbooru_tags:
+        generated_tags_animagine = tags.add_tags(prompt, rating_tags, copyright_tags_list, character_tags_list, general_tags, ban_tags, do_cfg, cfg_scale, negative_tags, total_token_length, max_new_tokens, min_new_tokens, temperature, top_p, top_k, num_beams)
+        prompt = generated_tags_animagine.strip()
+
     prompt, negative_prompt = utils.preprocess_prompt(
         quality_prompt, quality_selector, prompt, negative_prompt, add_quality_tags
     )
@@ -138,7 +160,7 @@ def generate(
             "Model": DESCRIPTION,
             "Model hash": "e3c47aedb0",
         }
-    
+
     logger.info(json.dumps(metadata, indent=4))
 
     try:
@@ -185,7 +207,7 @@ def generate(
             for image_path in image_paths:
                 logger.info(f"Image saved as {image_path} with metadata")
 
-        return image_paths, metadata
+        return image_paths, metadata, generated_tags_animagine
     except Exception as e:
         logger.exception(f"An error occurred: {e}")
         raise
@@ -256,7 +278,7 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
                             value="Standard v3.1",
                         )
                     add_danbooru_tags = gr.Checkbox(
-                        label="Add Quality Tags", value=True
+                        label="Add Generated Tags", value=False
                     )
             with gr.Tab("Advanced Settings"):
                 with gr.Group():
@@ -646,6 +668,11 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
                 preview=True, 
                 show_label=False
             )
+            generated_tags_animagine = gr.Textbox(
+                label="Generated tags (AnimagineXL v3 style order)",
+                # placeholder="tags will be here in Animagine v3 style order",
+                interactive=False,
+            )
             with gr.Accordion(label="Generation Parameters", open=False):
                 gr_metadata = gr.JSON(label="metadata", show_label=False)
             gr.Examples(
@@ -699,8 +726,25 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
             upscaler_strength,
             upscale_by,
             add_quality_tags,
+            add_danbooru_tags,
+            rating_dropdown,
+            copyright_tags_dropdown,
+            character_tags_dropdown,
+            general_tags_textbox,
+            ban_tags_textbox,
+            do_cfg_check,
+            cfg_scale_slider,
+            negative_tags_textbox,
+            total_token_length_radio,
+            max_new_tokens_slider,
+            min_new_tokens_slider,
+            temperature_slider,
+            top_p_slider,
+            top_k_slider,
+            num_beams_slider,
+            # model_backend_radio,
         ],
-        outputs=[result, gr_metadata],
+        outputs=[result, gr_metadata, generated_tags_animagine],
         api_name="run",
     )
 

tags.py

@@ -397,4 +397,8 @@ def get_copyright_tags_list():
     return COPYRIGHT_TAGS_LIST
 
 def get_character_tags_list():
-    return CHARACTER_TAGS_LIST
+    return CHARACTER_TAGS_LIST
+
+def add_tags(prompt, rating_tags, copyright_tags_list, character_tags_list, general_tags, ban_tags, do_cfg, cfg_scale, negative_tags, total_token_length, max_new_tokens, min_new_tokens, temperature, top_p, top_k, num_beams):
+    prompt = handle_inputs(rating_tags, copyright_tags_list, character_tags_list, general_tags, ban_tags, do_cfg, cfg_scale, negative_tags, total_token_length, max_new_tokens, min_new_tokens, temperature, top_p, top_k, num_beams)[2]
+    return  prompt