File size: 2,043 Bytes
db6367f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import random
import modules.scripts as scripts
import gradio as gr
import os
from modules import images
from modules.processing import process_images, Processed
from modules.shared import opts, cmd_opts, state
class RandomImageSizeScript(scripts.Script):
def title(self):
return "Image Size Randomize"
def ui(self, is_txt2img):
with gr.Tabs():
# Tab for direct text input
with gr.TabItem("Image Sizes"):
gr.HTML("<p>Enter image sizes in the format 'width,height', separated by new lines.</p>")
with gr.Row():
text_sizes = gr.Textbox(
label='Image Sizes',
placeholder='Enter sizes, e.g.:\n1024,512\n1024,768\n1024,1024',
elem_id=self.elem_id("text_sizes"),
lines=10 # Adjust line count for better usability
)
with gr.Row():
random_swap_text = gr.Checkbox(
label='Randomly swap width and height',
elem_id=self.elem_id("random_swap_text")
)
return [text_sizes, random_swap_text]
def run(self, p, text_sizes, random_swap_text):
sizes = []
# Read sizes from the textbox
if text_sizes:
try:
sizes = [tuple(map(int, line.strip().split(','))) for line in text_sizes.strip().split('\n') if line.strip()]
except ValueError:
print(f"Invalid size format in textbox: {text_sizes}")
if sizes:
width, height = random.choice(sizes)
if random_swap_text:
if random.choice([True, False]):
width, height = height, width
p.width, p.height = width, height
else:
print(f"No valid sizes found. Please check the textbox input.")
proc = process_images(p)
return proc
|