Extraltodeus/test_my_prompt
1
0
Fork 0
mirror of https://github.com/Extraltodeus/test_my_prompt.git synced 2026-04-30 19:21:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #25 from triviador/main

Browse Source 
Update test_my_prompt_custom_script.py
This commit is contained in:
Extraltodeus
2023-08-21 16:55:15 +02:00
committed by GitHub
parent 3fc7f795dd f0e7499b9e
commit 97073a0935
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/test_my_prompt_custom_script.py
View File

@@ -1,7 +1,7 @@
from modules.shared import opts, cmd_opts, state from modules.shared import opts, cmd_opts, state
from modules.processing import Processed, StableDiffusionProcessingImg2Img, process_images, images from modules.processing import Processed, StableDiffusionProcessingImg2Img, process_images, images
from PIL import Image, ImageFont, ImageDraw, ImageOps from PIL import Image, ImageFont, ImageDraw, ImageOps
from fonts.ttf import Roboto from modules.paths_internal import roboto_ttf_file
import modules.scripts as scripts import modules.scripts as scripts
import gradio as gr import gradio as gr
from collections import namedtuple from collections import namedtuple
@@ -22,7 +22,7 @@ class Script(scripts.Script):
def ui(self, is_img2img): def ui(self, is_img2img):
neg_pos = gr.Dropdown(label="Test negative or positive", choices=["Positive","Negative"], value="Positive") neg_pos = gr.Dropdown(label="Test negative or positive", choices=["Positive","Negative"], value="Positive")
skip_x_first = gr.Slider(minimum=0, maximum=32, step=1, label='Skip X first words', value=0) skip_x_first = gr.Slider(minimum=0, maximum=128, step=1, label='Skip X first words', value=0)
separator = gr.Textbox(label="Separator used", lines=1, value=", ") separator = gr.Textbox(label="Separator used", lines=1, value=", ")
grid_option = gr.Radio(choices=list(self.grid_options_mapping.keys()), label='Grid generation', value=self.default_grid_opt) grid_option = gr.Radio(choices=list(self.grid_options_mapping.keys()), label='Grid generation', value=self.default_grid_opt)
font_size = gr.Slider(minimum=12, maximum=64, step=1, label='Font size', value=32) font_size = gr.Slider(minimum=12, maximum=64, step=1, label='Font size', value=32)
@@ -35,7 +35,7 @@ class Script(scripts.Script):
margin=2 margin=2
fontsize=font_size fontsize=font_size
draw = ImageDraw.Draw(img) draw = ImageDraw.Draw(img)
font = ImageFont.truetype(Roboto, fontsize) font = ImageFont.truetype(roboto_ttf_file, fontsize)
text_height=iy-60 text_height=iy-60
tx = draw.textbbox((0,0),msg,font) 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)