diff options
author | AUTOMATIC <16777216c@gmail.com> | 2023-01-21 06:14:27 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-21 06:14:27 +0000 |
commit | 6d805b669e86233432f56ee1892d062103abe501 (patch) | |
tree | e76ace1a20b959f3a647740bc08085d4c82898aa /extensions-builtin/roll-artist/scripts/roll-artist.py | |
parent | 40ff6db5325fc34ad4fa35e80cb1e7768d9f7e75 (diff) | |
download | stable-diffusion-webui-gfx803-6d805b669e86233432f56ee1892d062103abe501.tar.gz stable-diffusion-webui-gfx803-6d805b669e86233432f56ee1892d062103abe501.tar.bz2 stable-diffusion-webui-gfx803-6d805b669e86233432f56ee1892d062103abe501.zip |
make CLIP interrogator download original text files if the directory does not exist
remove random artist built-in extension (to re-added as a normal extension on demand)
remove artists.csv (but what does it mean????????????????????)
make interrogate buttons show Loading... when you click them
Diffstat (limited to 'extensions-builtin/roll-artist/scripts/roll-artist.py')
-rw-r--r-- | extensions-builtin/roll-artist/scripts/roll-artist.py | 50 |
1 files changed, 0 insertions, 50 deletions
diff --git a/extensions-builtin/roll-artist/scripts/roll-artist.py b/extensions-builtin/roll-artist/scripts/roll-artist.py deleted file mode 100644 index c3bc1fd0..00000000 --- a/extensions-builtin/roll-artist/scripts/roll-artist.py +++ /dev/null @@ -1,50 +0,0 @@ -import random
-
-from modules import script_callbacks, shared
-import gradio as gr
-
-art_symbol = '\U0001f3a8' # 🎨
-global_prompt = None
-related_ids = {"txt2img_prompt", "txt2img_clear_prompt", "img2img_prompt", "img2img_clear_prompt" }
-
-
-def roll_artist(prompt):
- allowed_cats = set([x for x in shared.artist_db.categories() if len(shared.opts.random_artist_categories)==0 or x in shared.opts.random_artist_categories])
- artist = random.choice([x for x in shared.artist_db.artists if x.category in allowed_cats])
-
- return prompt + ", " + artist.name if prompt != '' else artist.name
-
-
-def add_roll_button(prompt):
- roll = gr.Button(value=art_symbol, elem_id="roll", visible=len(shared.artist_db.artists) > 0)
-
- roll.click(
- fn=roll_artist,
- _js="update_txt2img_tokens",
- inputs=[
- prompt,
- ],
- outputs=[
- prompt,
- ]
- )
-
-
-def after_component(component, **kwargs):
- global global_prompt
-
- elem_id = kwargs.get('elem_id', None)
- if elem_id not in related_ids:
- return
-
- if elem_id == "txt2img_prompt":
- global_prompt = component
- elif elem_id == "txt2img_clear_prompt":
- add_roll_button(global_prompt)
- elif elem_id == "img2img_prompt":
- global_prompt = component
- elif elem_id == "img2img_clear_prompt":
- add_roll_button(global_prompt)
-
-
-script_callbacks.on_after_component(after_component)
|