-
Notifications
You must be signed in to change notification settings - Fork 27.3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
d9499f4
commit 4d4a9e7
Showing
10 changed files
with
314 additions
and
168 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,141 @@ | ||
| 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 = "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 (press Ctrl+Enter or Alt+Enter to generate)", 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 (press Ctrl+Enter or Alt+Enter to generate)", 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=[], | ||
| ) | ||
|
|
||
| self.interrupt.click( | ||
| fn=lambda: shared.state.interrupt(), | ||
| 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="<span>0/75</span>", 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="<span>0/75</span>", 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) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters