from typing import Dict, List, Tuple, Union quality_prompt_list = [ { "name": "(None)", "prompt": "{prompt}", "negative_prompt": "nsfw, lowres", }, { "name": "Standard v3.0", "prompt": "{prompt}, masterpiece, best quality", "negative_prompt": "nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name", }, { "name": "Standard v3.1", "prompt": "{prompt}, masterpiece, best quality, very aesthetic, absurdres", "negative_prompt": "nsfw, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]", }, { "name": "Light v3.1", "prompt": "{prompt}, (masterpiece), best quality, very aesthetic, perfect face", "negative_prompt": "nsfw, (low quality, worst quality:1.2), very displeasing, 3d, watermark, signature, ugly, poorly drawn", }, { "name": "Heavy v3.1", "prompt": "{prompt}, (masterpiece), (best quality), (ultra-detailed), very aesthetic, illustration, disheveled hair, perfect composition, moist skin, intricate details", "negative_prompt": "nsfw, longbody, lowres, bad anatomy, bad hands, missing fingers, pubic hair, extra digit, fewer digits, cropped, worst quality, low quality, very displeasing", }, ] style_list = [ { "name": "(None)", "prompt": "{prompt}", "negative_prompt": "", }, { "name": "Cinematic", "prompt": "{prompt}, cinematic still, emotional, harmonious, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy", "negative_prompt": "nsfw, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured", }, { "name": "Photographic", "prompt": "{prompt}, cinematic photo, 35mm photograph, film, bokeh, professional, 4k, highly detailed", "negative_prompt": "nsfw, drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly", }, { "name": "Anime", "prompt": "{prompt}, anime artwork, anime style, key visual, vibrant, studio anime, highly detailed", "negative_prompt": "nsfw, photo, deformed, black and white, realism, disfigured, low contrast", }, { "name": "Manga", "prompt": "{prompt}, manga style, vibrant, high-energy, detailed, iconic, Japanese comic style", "negative_prompt": "nsfw, ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, Western comic style", }, { "name": "Digital Art", "prompt": "{prompt}, concept art, digital artwork, illustrative, painterly, matte painting, highly detailed", "negative_prompt": "nsfw, photo, photorealistic, realism, ugly", }, { "name": "Pixel art", "prompt": "{prompt}, pixel-art, low-res, blocky, pixel art style, 8-bit graphics", "negative_prompt": "nsfw, sloppy, messy, blurry, noisy, highly detailed, ultra textured, photo, realistic", }, { "name": "Fantasy art", "prompt": "{prompt}, ethereal fantasy concept art, magnificent, celestial, ethereal, painterly, epic, majestic, magical, fantasy art, cover art, dreamy", "negative_prompt": "nsfw, photographic, realistic, realism, 35mm film, dslr, cropped, frame, text, deformed, glitch, noise, noisy, off-center, deformed, cross-eyed, closed eyes, bad anatomy, ugly, disfigured, sloppy, duplicate, mutated, black and white", }, { "name": "Neonpunk", "prompt": "{prompt}, neonpunk style, cyberpunk, vaporwave, neon, vibes, vibrant, stunningly beautiful, crisp, detailed, sleek, ultramodern, magenta highlights, dark purple shadows, high contrast, cinematic, ultra detailed, intricate, professional", "negative_prompt": "nsfw, painting, drawing, illustration, glitch, deformed, mutated, cross-eyed, ugly, disfigured", }, { "name": "3D Model", "prompt": "{prompt}, professional 3d model, octane render, highly detailed, volumetric, dramatic lighting", "negative_prompt": "nsfw, ugly, deformed, noisy, low poly, blurry, painting", }, ] _style_dict = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list} _quality_dict = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in quality_prompt_list} def preprocess_prompt( positive: str, negative: str = "", style_dict: Dict[str, dict] = _quality_dict, style_name: str = "Standard v3.1", # "Heavy v3.1" add_style: bool = True, ) -> Tuple[str, str]: p, n = style_dict.get(style_name, style_dict["(None)"]) if add_style and positive.strip(): formatted_positive = p.format(prompt=positive) else: formatted_positive = positive combined_negative = n if negative.strip(): if combined_negative: combined_negative += ", " + negative else: combined_negative = negative return formatted_positive, combined_negative def preprocess_prompts( positives: List[str], negatives: List[str] = None, style_dict = _style_dict, style_name: str = "Manga", # "(None)" quality_dict = _quality_dict, quality_name: str = "Standard v3.1", # "Heavy v3.1" add_style: bool = True, add_quality_tags = True, ) -> Tuple[List[str], List[str]]: if negatives is None: negatives = ['' for _ in positives] positives_ = [] negatives_ = [] for pos, neg in zip(positives, negatives): pos, neg = preprocess_prompt(pos, neg, quality_dict, quality_name, add_quality_tags) pos, neg = preprocess_prompt(pos, neg, style_dict, style_name, add_style) positives_.append(pos) negatives_.append(neg) return positives_, negatives_ def print_prompts( positives: Union[str, List[str]], negatives: Union[str, List[str]], has_background: bool = False, ) -> None: if isinstance(positives, str): positives = [positives] if isinstance(negatives, str): negatives = [negatives] for i, prompt in enumerate(positives): prefix = ((f'Prompt{i}' if i > 0 else 'Background Prompt') if has_background else f'Prompt{i + 1}') print(prefix + ': ' + prompt) for i, prompt in enumerate(negatives): prefix = ((f'Negative Prompt{i}' if i > 0 else 'Background Negative Prompt') if has_background else f'Negative Prompt{i + 1}') print(prefix + ': ' + prompt)