1580158
/
danbooru-prompt
mirror of https://github.com/EnsignMK/danbooru-prompt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add files via upload 

Tags can go immediately to the textprompt texbox
debug_branch
EnsignMK 2023-07-07 23:10:19 +02:00 committed by GitHub
parent df93ad96f1
commit 8cd21621f9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
scripts/fetch.py
View File

@ -1,3 +1,4 @@
import contextlib
import gradio as gr
from modules import scripts
@ -40,10 +41,24 @@ class BooruScript(scripts.Script):
with gr.Accordion("DanBooru Link", open=False):
fetch_tags = gr.Button(value='Get Tags', variant='primary')
link= gr.Textbox(label="insert link")
tags = gr.Textbox(value="", label="Tags", lines=5)
fetch_tags.click(fn=fetchTags, inputs=[link], outputs=[tags])
return [link, tags, fetch_tags]
with contextlib.suppress(AttributeError):
if is_img2img:
fetch_tags.click(fn=fetchTags, inputs=[link], outputs=[self.boxxIMG])
else:
fetch_tags.click(fn=fetchTags, inputs=[link], outputs=[self.boxx])
return [link, fetch_tags]
def after_component(self, component, **kwargs):
if kwargs.get("elem_id") == "txt2img_prompt":
self.boxx=component
if kwargs.get("elem_id") == "img2img_prompt":
self.boxxIMG=component
script_callbacks.on_ui_settings(on_ui_settings)