| from modules.shared import opts | |
| from modules.processing import Processed, process_images | |
| import modules.scripts as scripts | |
| import gradio as gr | |
| class Script(scripts.Script): | |
| def title(self): | |
| return "Test Prompts" | |
| def ui(self, is_img2img): | |
| pose1 = gr.Checkbox(label="Fruits on a table", value=True) | |
| pose2 = gr.Checkbox(label="Majestic landscape", value=False) | |
| pose3 = gr.Checkbox(label="Modern cityscape at night", value=False) | |
| pose4 = gr.Checkbox(label="Fantasy Dragon", value=False) | |
| pose5 = gr.Checkbox(label="Posing woman", value=False) | |
| pose6 = gr.Checkbox(label="Posing man", value=False) | |
| prompt_group = [pose1, pose2, pose3, pose4, pose5, pose6] | |
| with gr.Box(): | |
| options_header = gr.Markdown("<br>Options") | |
| nsfw = gr.Checkbox(label="NSFW content", value=False) | |
| sfw = gr.Checkbox(label="SFW content", value=True) | |
| return prompt_group + [nsfw, sfw] | |
| def run(self, p, pose1, pose2, pose3, pose4, pose5, pose6, nsfw, sfw): | |
| prompts = [] | |
| negative_prompts = [] | |
| if pose1: | |
| prompts.append("photo of fruits on top of a table") | |
| negative_prompts.append("(drawing, painting, deformed, mutated, drawing, anime, ugly, worse photo, worse proportions)") | |
| if pose2: | |
| prompts.append("majestic landscape with mountains, rivers, and a sunset") | |
| negative_prompts.append("(overexposed, underexposed, low resolution, unnatural colors, digital art, painting, drawing)") | |
| if pose3: | |
| prompts.append("modern cityscape at night with bright lights and tall buildings") | |
| negative_prompts.append("(blurry, overexposed, underexposed, low resolution, painting, drawing, anime, empty streets)") | |
| if pose4: | |
| prompts.append("a realistic evil dragon is flying over a castle") | |
| negative_prompts.append("(low resolution, blurry, overexposed, underexposed, cartoonish, unrealistic, anime, painting)") | |
| if pose5: | |
| prompts.append("full-body photo of a posing beautiful woman in a photoshooting") | |
| negative_prompts.append("(cat, horns, deformations, mutations, painting, any worse face, worse human, anime, invisible head, invisible body)") | |
| if pose6: | |
| prompts.append("full-body photo with a posing male man in a photoshooting") | |
| negative_prompts.append("(worse bodyshape, deformations, mutations, painting, cartoon, any worse human-body, anime, invisible head, invisible body)") | |
| if nsfw: | |
| prompts.append("naughty, hot erotic, somehow sexual, explicit NSFW content") | |
| negative_prompts.append("bad faces, worse, blurry") | |
| if sfw: | |
| prompts.append("sfw") | |
| negative_prompts.append("porn, erotic, nudity, nsfw, naked") | |
| p.prompt = ", ".join(prompts) | |
| p.negative_prompt = ", ".join(negative_prompts) | |
| proc = process_images(p) | |
| return Processed(p, proc.images, p.seed, all_prompts=proc.all_prompts, infotexts=proc.infotexts) | |