Add files via upload

test_my_prompt_script.py

@@ -0,0 +1,77 @@
+from modules.shared import opts, cmd_opts, state
+from modules.processing import Processed, StableDiffusionProcessingImg2Img, process_images, images
+from PIL import Image, ImageFont, ImageDraw, ImageOps
+from fonts.ttf import Roboto
+import modules.scripts as scripts
+import gradio as gr
+from random import randint
+
+def title(self):
+    return "Test my prompt!"
+
+def ui(self, is_img2img):
+    neg_pos = gr.Dropdown(label="Test negative or positive", choices=["Positive","Negative"], value="Positive")
+    font_size = gr.Slider(minimum=12, maximum=64, step=1, label='Font size', value=32)
+    return [neg_pos,font_size]
+
+def run(p,neg_pos,font_size):
+    def write_on_image(img, msg):
+        ix,iy = img.size
+        draw = ImageDraw.Draw(img)
+        margin=2
+        fontsize=font_size
+        draw = ImageDraw.Draw(img)
+        font = ImageFont.truetype(Roboto, fontsize)
+        text_height=iy-60
+        tx = draw.textbbox((0,0),msg,font)
+        draw.text((int((ix-tx[2])/2),text_height+margin),msg,(0,0,0),font=font)
+        draw.text((int((ix-tx[2])/2),text_height-margin),msg,(0,0,0),font=font)
+        draw.text((int((ix-tx[2])/2+margin),text_height),msg,(0,0,0),font=font)
+        draw.text((int((ix-tx[2])/2-margin),text_height),msg,(0,0,0),font=font)
+        draw.text((int((ix-tx[2])/2),text_height), msg,(255,255,255),font=font)
+        return img
+
+    output_images = []
+    p.do_not_save_samples = True
+    initial_seed = p.seed
+    if initial_seed == -1:
+        initial_seed = randint(1000000,9999999)
+    if neg_pos == "Positive":
+        initial_prompt =  p.prompt
+        prompt_array = p.prompt
+    else:
+        initial_prompt =  p.negative_prompt
+        prompt_array = p.negative_prompt
+
+    prompt_array = prompt_array.split(", ")
+    print("total images :", len(prompt_array))
+    for g in range(len(prompt_array)+1):
+        f = g-1
+        if f >= 0:
+            new_prompt =  ', '.join([prompt_array[x] for x in range(len(prompt_array)) if x is not f])
+        else:
+            new_prompt = initial_prompt
+
+        if neg_pos == "Positive":
+            p.prompt = new_prompt
+        else:
+            p.negative_prompt = new_prompt
+        p.seed = initial_seed
+
+        proc = process_images(p)
+
+        if f >= 0:
+            proc.images[0] = write_on_image(proc.images[0], prompt_array[f])
+        else:
+            proc.images[0] = write_on_image(proc.images[0], "full prompt")
+
+        output_images.append(proc.images[0])
+        images.save_image(proc.images[0], p.outpath_samples, "", proc.seed, proc.prompt, opts.samples_format, info= proc.info, p=p)
+
+    unwanted_grid_because_of_img_count = len(output_images) < 2 and opts.grid_only_if_multiple
+    if (opts.return_grid or opts.grid_save) and not p.do_not_save_grid and not unwanted_grid_because_of_img_count:
+        grid = images.image_grid(output_images)
+        if opts.grid_save:
+            images.save_image(grid, p.outpath_grids, "grid", initial_seed, initial_prompt, opts.grid_format, info=proc.info, short_filename=not opts.grid_extended_filename, p=p, grid=True)
+    return proc
+run(p)