Extraltodeus/test_my_prompt
1
0
Fork 0
mirror of https://github.com/Extraltodeus/test_my_prompt.git synced 2026-04-23 07:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test_my_prompt_custom_script.py

Browse Source
This commit is contained in:
Extraltodeus
2022-11-07 13:54:31 +01:00
committed by GitHub
parent 246877b51c
commit e513b0084f
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test_my_prompt_custom_script.py
View File

@@ -12,10 +12,11 @@ class Script(scripts.Script):
def ui(self, is_img2img):
neg_pos = gr.Dropdown(label="Test negative or positive", choices=["Positive","Negative"], value="Positive")
separator = gr.Textbox(label="Separator used", lines=1, value=", ")
font_size = gr.Slider(minimum=12, maximum=64, step=1, label='Font size', value=32)
return [neg_pos,font_size]
return [neg_pos,separator,font_size]
def run(self, p,neg_pos, font_size):
def run(self, p,neg_pos,separator,font_size):
def write_on_image(img, msg):
ix,iy = img.size
draw = ImageDraw.Draw(img)
@@ -44,12 +45,12 @@ class Script(scripts.Script):
initial_prompt = p.negative_prompt
prompt_array = p.negative_prompt
prompt_array = prompt_array.split(", ")
prompt_array = prompt_array.split(separator)
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])
new_prompt = separator.join([prompt_array[x] for x in range(len(prompt_array)) if x is not f])
else:
new_prompt = initial_prompt