Commit c9c457ed authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

stylistic changes for #13118

parent 73c2a03d
Loading
Loading
Loading
Loading
+7 −3
Original line number Diff line number Diff line
import datetime
import functools
import mimetypes
import os
import sys
@@ -152,7 +151,7 @@ def connect_clear_prompt(button):
    )


def update_token_counter(text, steps, is_positive=True):
def update_token_counter(text, steps, *, is_positive=True):
    try:
        text, _ = extra_networks.parse_prompt(text)

@@ -160,6 +159,7 @@ def update_token_counter(text, steps, is_positive=True):
            _, prompt_flat_list, _ = prompt_parser.get_multicond_prompt_list([text])
        else:
            prompt_flat_list = [text]

        prompt_schedules = prompt_parser.get_learned_conditioning_prompt_schedules(prompt_flat_list, steps)

    except Exception:
@@ -173,6 +173,10 @@ def update_token_counter(text, steps, is_positive=True):
    return f"<span class='gr-box gr-text-input'>{token_count}/{max_length}</span>"


def update_negative_prompt_token_counter(text, steps):
    return update_token_counter(text, steps, is_positive=False)


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"""

@@ -539,7 +543,7 @@ def create_ui():
            ]

            toprow.token_button.click(fn=wrap_queued_call(update_token_counter), inputs=[toprow.prompt, steps], outputs=[toprow.token_counter])
            toprow.negative_token_button.click(fn=wrap_queued_call(functools.partial(update_token_counter, is_positive=False)), inputs=[toprow.negative_prompt, steps], outputs=[toprow.negative_token_counter])
            toprow.negative_token_button.click(fn=wrap_queued_call(update_negative_prompt_token_counter), inputs=[toprow.negative_prompt, steps], outputs=[toprow.negative_token_counter])

        extra_networks_ui = ui_extra_networks.create_ui(txt2img_interface, [txt2img_generation_tab], 'txt2img')
        ui_extra_networks.setup_ui(extra_networks_ui, txt2img_gallery)