ahgsql/StyleSelectorXL
1
0
Fork 0
mirror of https://github.com/ahgsql/StyleSelectorXL.git synced 2026-04-30 11:11:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enhancement- Styles Can be Rendered as DropDown, Randomize Style Option added

Browse Source
This commit is contained in:
ahgsql
2023-08-05 18:40:04 +03:00
parent 9cbf833d5f
commit c683fc00ac
1 changed files with 33 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
scripts/StyleSelectorXL.py
View File

@@ -1,11 +1,11 @@
import contextlib import contextlib
import gradio as gr import gradio as gr
from modules import scripts from modules import scripts, shared, script_callbacks
from modules import script_callbacks from modules.ui_components import FormRow, FormColumn, FormGroup, ToolButton
import json import json
import os import os
import random
stylespath = "" stylespath = ""
@@ -121,21 +121,35 @@ class StyleSelectorXL(scripts.Script):
def ui(self, is_img2img): def ui(self, is_img2img):
with gr.Group(): with gr.Group():
with gr.Accordion("SDXL Styles", open=True): with gr.Accordion("SDXL Styles", open=True):
is_enabled = gr.Checkbox( with FormRow():
value=True, label="Enable Style Selector") with FormColumn(min_width=160):
style = gr.Radio( is_enabled = gr.Checkbox(
label='Codec', choices=self.styleNames, value='base') value=True, label="Enable Style Selector", info="Enable Or Disable Style Selector ")
with FormColumn(elem_id="Randomize Style"):
randomize = gr.Checkbox(
value=False, label="Randomize Style", info="This Will Override Selected Style")
style_ui_type = shared.opts.data.get(
"styles_ui", "radio-buttons")
if style_ui_type == "select-list":
style = gr.Dropdown(
self.styleNames, value='base', multiselect=False, label="Select Style")
else:
style = gr.Radio(
label='Style', choices=self.styleNames, value='base')
# Ignore the error if the attribute is not present # Ignore the error if the attribute is not present
return [is_enabled, style] return [is_enabled, randomize, style]
def process(self, p, is_enabled, style): def process(self, p, is_enabled, randomize, style):
if not is_enabled: if not is_enabled:
return return
p.extra_generation_params["Style Selector Enabled"] = True p.extra_generation_params["Style Selector Enabled"] = True
if randomize:
style = random.choice(self.styleNames)
for i, prompt in enumerate(p.all_prompts): # for each image in batch for i, prompt in enumerate(p.all_prompts): # for each image in batch
positivePrompt = createPositive(style, prompt) positivePrompt = createPositive(style, prompt)
@@ -162,3 +176,12 @@ class StyleSelectorXL(scripts.Script):
#self.neg_prompt_boxTXT = component #self.neg_prompt_boxTXT = component
# if kwargs.get("elem_id") == "img2img_neg_prompt": # if kwargs.get("elem_id") == "img2img_neg_prompt":
#self.neg_prompt_boxIMG = component #self.neg_prompt_boxIMG = component
def on_ui_settings():
section = ("styleselector", "Style Selector")
shared.opts.add_option("styles_ui", shared.OptionInfo(
"radio-buttons", "How should Style Names Rendered on UI", gr.Radio, {"choices": ["radio-buttons", "select-list"]}, section=section))
script_callbacks.on_ui_settings(on_ui_settings)