mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-07-21 21:01:24 +08:00
174 lines
7.8 KiB
Python
174 lines
7.8 KiB
Python
from contextlib import nullcontext
|
|
|
|
import gradio as gr
|
|
|
|
import modules.images
|
|
from modules import shared, ui_prompt_styles
|
|
from modules.ui_components import ToolButton
|
|
|
|
|
|
class Toprow:
|
|
"""Creates a top row UI with prompts, generate button, styles, extra little buttons for things, and enables some functionality related to their operation"""
|
|
|
|
prompt = None
|
|
prompt_img = None
|
|
negative_prompt = None
|
|
|
|
interrupt = None
|
|
interrupting = None
|
|
skip = None
|
|
submit = None
|
|
|
|
paste = None
|
|
clear_prompt_button = None
|
|
apply_styles = None
|
|
restore_progress_button = None
|
|
|
|
token_counter = None
|
|
token_button = None
|
|
negative_token_counter = None
|
|
negative_token_button = None
|
|
|
|
ui_styles = None
|
|
|
|
submit_box = None
|
|
|
|
def __init__(self, is_img2img, *, is_compact=None, id_part=None):
|
|
self.is_img2img = is_img2img
|
|
|
|
if id_part is None:
|
|
id_part = "img2img" if is_img2img else "txt2img"
|
|
self.id_part = id_part
|
|
|
|
if is_compact is None:
|
|
is_compact = shared.opts.prompt_box_style == "Compact"
|
|
self.is_compact = is_compact
|
|
|
|
if is_compact:
|
|
self.create_submit_box()
|
|
else:
|
|
with gr.Row(elem_id=f"{id_part}_toprow", variant="compact"):
|
|
self.create_classic_toprow()
|
|
|
|
def create_classic_toprow(self):
|
|
self.create_prompts()
|
|
|
|
with gr.Column(scale=1, elem_id=f"{self.id_part}_actions_column"):
|
|
self.create_submit_box()
|
|
self.create_tools_row()
|
|
self.create_styles_ui()
|
|
|
|
def create_inline_toprow_prompts(self):
|
|
if not self.is_compact:
|
|
return
|
|
|
|
self.create_prompts()
|
|
|
|
with gr.Row(elem_classes=["toprow-compact-stylerow"]):
|
|
with gr.Column(elem_classes=["toprow-compact-tools"]):
|
|
self.create_tools_row()
|
|
with gr.Column():
|
|
self.create_styles_ui()
|
|
|
|
def create_inline_toprow_image(self):
|
|
if not self.is_compact:
|
|
return
|
|
|
|
self.submit_box.render()
|
|
|
|
@staticmethod
|
|
def _container_class() -> str:
|
|
match shared.opts.prompt_box_style:
|
|
case "Compact":
|
|
return "prompt-container-compact"
|
|
case "Scrollable":
|
|
return "prompt-container-scroll"
|
|
case _:
|
|
return None
|
|
|
|
def create_prompts(self):
|
|
with gr.Column(elem_id=f"{self.id_part}_prompt_container", elem_classes=self._container_class(), scale=6):
|
|
_container = gr.Accordion(label="Prompts", open=False) if shared.opts.prompt_box_style == "Accordion" else nullcontext()
|
|
_container.__enter__()
|
|
|
|
with gr.Row(elem_id=f"{self.id_part}_prompt_row", elem_classes=["prompt-row"]):
|
|
self.prompt = gr.Textbox(label="Prompt", elem_id=f"{self.id_part}_prompt", show_label=False, lines=3, placeholder="Prompt\n(Ctrl+Enter to Generate ; Alt+Enter to Skip ; Esc to Interrupt)", elem_classes=["prompt"])
|
|
self.prompt_img = gr.File(elem_id=f"{self.id_part}_prompt_image", file_count="single", type="binary", visible=False)
|
|
|
|
with gr.Row(elem_id=f"{self.id_part}_neg_prompt_row", elem_classes=["prompt-row"]):
|
|
self.negative_prompt = gr.Textbox(label="Negative Prompt", elem_id=f"{self.id_part}_neg_prompt", show_label=False, lines=3, placeholder="Negative Prompt\n(Ctrl+Enter to Generate ; Alt+Enter to Skip ; Esc to Interrupt)", elem_classes=["prompt"])
|
|
|
|
_container.__exit__()
|
|
|
|
self.prompt_img.change(
|
|
fn=modules.images.image_data,
|
|
inputs=[self.prompt_img],
|
|
outputs=[self.prompt, self.prompt_img],
|
|
show_progress=False,
|
|
)
|
|
|
|
def create_submit_box(self):
|
|
with gr.Row(elem_id=f"{self.id_part}_generate_box", elem_classes=["generate-box"] + (["generate-box-compact"] if self.is_compact else []), render=not self.is_compact) as submit_box:
|
|
self.submit_box = submit_box
|
|
|
|
self.interrupt = gr.Button("Interrupt", elem_id=f"{self.id_part}_interrupt", elem_classes="generate-box-interrupt", tooltip="End generation immediately or after completing current batch")
|
|
self.skip = gr.Button("Skip", elem_id=f"{self.id_part}_skip", elem_classes="generate-box-skip", tooltip="Stop generation of current batch and continues onto next batch")
|
|
self.interrupting = gr.Button("Interrupting...", elem_id=f"{self.id_part}_interrupting", elem_classes="generate-box-interrupting", tooltip="Interrupting generation...")
|
|
self.submit = gr.Button("Generate", elem_id=f"{self.id_part}_generate", variant="primary", tooltip="Right click generate forever menu")
|
|
|
|
def interrupt_function():
|
|
if not shared.state.stopping_generation and shared.state.job_count > 1 and shared.opts.interrupt_after_current:
|
|
shared.state.stop_generating()
|
|
gr.Info("Generation will stop after finishing this image, click again to stop immediately.")
|
|
else:
|
|
shared.state.interrupt()
|
|
|
|
self.skip.click(fn=shared.state.skip)
|
|
self.interrupt.click(fn=interrupt_function, _js='function(){ showSubmitInterruptingPlaceholder("' + self.id_part + '"); }')
|
|
self.interrupting.click(fn=interrupt_function)
|
|
|
|
def create_tools_row(self):
|
|
with gr.Row(elem_id=f"{self.id_part}_tools"):
|
|
from modules.ui import (
|
|
clear_prompt_symbol,
|
|
paste_symbol,
|
|
restore_progress_symbol,
|
|
)
|
|
|
|
self.paste = ToolButton(value=paste_symbol, elem_id="paste", tooltip="Read generation parameters from prompt or last generation if prompt is empty into user interface.")
|
|
self.clear_prompt_button = ToolButton(value=clear_prompt_symbol, elem_id=f"{self.id_part}_clear_prompt", tooltip="Clear prompt")
|
|
self.apply_styles = ToolButton(value=ui_prompt_styles.styles_materialize_symbol, elem_id=f"{self.id_part}_style_apply", tooltip="Apply all selected styles to prompts. Strips comments, if enabled.")
|
|
|
|
self.restore_progress_button = ToolButton(value=restore_progress_symbol, elem_id=f"{self.id_part}_restore_progress", visible=False, tooltip="Restore progress")
|
|
|
|
self.token_counter = gr.HTML(value="<span>0/75</span>", elem_id=f"{self.id_part}_token_counter", elem_classes=["token-counter"], visible=False)
|
|
self.token_button = gr.Button(visible=False, elem_id=f"{self.id_part}_token_button")
|
|
self.negative_token_counter = gr.HTML(value="<span>0/75</span>", elem_id=f"{self.id_part}_negative_token_counter", elem_classes=["token-counter"], visible=False)
|
|
self.negative_token_button = gr.Button(visible=False, elem_id=f"{self.id_part}_negative_token_button")
|
|
|
|
self.clear_prompt_button.click(
|
|
fn=lambda *x: x,
|
|
_js="confirm_clear_prompt",
|
|
inputs=[self.prompt, self.negative_prompt],
|
|
outputs=[self.prompt, self.negative_prompt],
|
|
show_progress=False,
|
|
)
|
|
|
|
def create_styles_ui(self):
|
|
self.ui_styles = ui_prompt_styles.UiPromptStyles(self.id_part, self.prompt, self.negative_prompt)
|
|
self.ui_styles.setup_apply_button(self.apply_styles)
|
|
|
|
def hook_paste_guard(self):
|
|
assert self.negative_prompt is not None and self.paste is not None
|
|
|
|
def guard(prompt: str) -> bool:
|
|
return gr.update(interactive=(not bool(prompt.strip())))
|
|
|
|
self.negative_prompt.change(
|
|
fn=guard,
|
|
inputs=[self.negative_prompt],
|
|
outputs=[self.paste],
|
|
show_progress=False,
|
|
queue=False,
|
|
)
|