File size: 2,000 Bytes
0734ee1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
import gradio as gr
from modules import script_callbacks
import modules.generation_parameters_copypaste as parameters_copypaste
from modules import extensions
import os
def get_self_extension():
if '__file__' in globals():
filepath = __file__
else:
import inspect
filepath = inspect.getfile(lambda: None)
for ext in extensions.active():
if ext.path in filepath:
return ext
def on_ui_tabs():
ext = get_self_extension()
if ext is None:
return []
js_ = [f'{x.path}?{os.path.getmtime(x.path)}' for x in ext.list_files('js', '.js')]
js_.insert(0, ext.path)
with gr.Blocks(analytics_enabled=False) as fast_pnginfo:
with gr.Row():
with gr.Column(scale=2):
gr.HTML(value='\n'.join(js_), elem_id="fastpng_js_path", visible=False)
gr.Markdown(
f"""
<center>
</center>
""")
with gr.Row().style(equal_height=False):
with gr.Column(variant='panel'):
image = gr.Image(elem_id="fastpnginfo_image", label="Source", source="upload", interactive=True, type="pil")
# image.change(lambda x: x, _js='(e)=>{console.log(e);fastpngprocess()}')
with gr.Column(variant='panel'):
submit = gr.Button(label="Submit", elem_id="fastpnginfo_submit", interactive=True, variant="primary", visible=True)
fast_generation_info = gr.Textbox(label="Parameters", visible=True, elem_id="fastpnginfo_generation_info", interactive=False)
with gr.Row():
buttons = parameters_copypaste.create_buttons(["txt2img", "img2img", "inpaint", "extras"])
for tabname, button in buttons.items():
parameters_copypaste.register_paste_params_button(parameters_copypaste.ParamBinding(
paste_button=button, tabname=tabname, source_text_component=fast_generation_info, source_image_component=image,
))
return [(fast_pnginfo, "Fast PNG Info", "fast_pnginfo")];
script_callbacks.on_ui_tabs(on_ui_tabs) |