import gradio as gr from modules import shared, ui_prompt_styles import modules.images 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 button_interrogate = None button_deepbooru = None interrupt = 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=False, id_part=None): if id_part is None: id_part = "img2img" if is_img2img else "txt2img" self.id_part = id_part self.is_img2img = is_img2img self.is_compact = is_compact if not is_compact: with gr.Row(elem_id=f"{id_part}_toprow", variant="compact"): self.create_classic_toprow() else: self.create_submit_box() 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() def create_prompts(self): with gr.Column(elem_id=f"{self.id_part}_prompt_container", elem_classes=["prompt-container-compact"] if self.is_compact else [], scale=6): 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(Press Ctrl+Enter to generate, Alt+Enter to skip, Esc to interrupt)", elem_classes=["prompt"]) self.prompt_img = gr.File(label="", 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(Press Ctrl+Enter to generate, Alt+Enter to skip, Esc to interrupt)", elem_classes=["prompt"]) 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") self.skip = gr.Button('Skip', elem_id=f"{self.id_part}_skip", elem_classes="generate-box-skip") self.submit = gr.Button('Generate', elem_id=f"{self.id_part}_generate", variant='primary') self.skip.click( fn=lambda: shared.state.skip(), inputs=[], outputs=[], ) def interrupt_function(): if shared.state.job_count > 1 and shared.opts.interrupt_after_current: shared.state.stop_generating() else: shared.state.interrupt() self.interrupt.click( fn=interrupt_function, inputs=[], outputs=[], ) def create_tools_row(self): with gr.Row(elem_id=f"{self.id_part}_tools"): from modules.ui import paste_symbol, clear_prompt_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.") if self.is_img2img: self.button_interrogate = ToolButton('📎', tooltip='Interrogate CLIP - use CLIP neural network to create a text describing the image, and put it into the prompt field', elem_id="interrogate") self.button_deepbooru = ToolButton('📦', tooltip='Interrogate DeepBooru - use DeepBooru neural network to create a text describing the image, and put it into the prompt field', elem_id="deepbooru") 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="0/75", elem_id=f"{self.id_part}_token_counter", elem_classes=["token-counter"]) self.token_button = gr.Button(visible=False, elem_id=f"{self.id_part}_token_button") self.negative_token_counter = gr.HTML(value="0/75", elem_id=f"{self.id_part}_negative_token_counter", elem_classes=["token-counter"]) 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], ) 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)