import datetime import html import json import os.path import gradio as gr from modules import generation_parameters_copypaste, images, sysinfo, errors class UserMetadataEditor: def __init__(self, ui, tabname, page): self.ui = ui self.tabname = tabname self.page = page self.id_part = f"{self.tabname}_{self.page.id_page}_edit_user_metadata" self.box = None self.edit_name_input = None self.button_edit = None self.edit_name = None self.edit_description = None self.html_filedata = None self.html_preview = None self.button_cancel = None self.button_replace_preview = None self.button_save = None def get_user_metadata(self, name): item = self.page.items.get(name, {}) user_metadata = item.get('user_metadata', None) if user_metadata is None: user_metadata = {} item['user_metadata'] = user_metadata return user_metadata def create_default_editor_elems(self): with gr.Row(): with gr.Column(scale=2): self.edit_name = gr.HTML(elem_classes="extra-network-name") self.edit_description = gr.Textbox(label="Description", lines=4) self.html_filedata = gr.HTML() with gr.Column(scale=1, min_width=0): self.html_preview = gr.HTML() def create_default_buttons(self): with gr.Row(): self.button_cancel = gr.Button('Cancel') self.button_replace_preview = gr.Button('Replace preview', variant='primary') self.button_save = gr.Button('Save', variant='primary') self.button_cancel.click(fn=None, _js="closePopup") def get_card_html(self, name): item = self.page.items.get(name, {}) preview_url = item.get("preview", None) if not preview_url: filename, _ = os.path.splitext(item["filename"]) preview_url = self.page.find_preview(filename) item["preview"] = preview_url if preview_url: preview = f'''
''' else: preview = "
" return preview def get_metadata_table(self, name): item = self.page.items.get(name, {}) try: filename = item["filename"] stats = os.stat(filename) params = [ ('File size: ', sysinfo.pretty_bytes(stats.st_size)), ('Created: ', datetime.datetime.fromtimestamp(stats.st_ctime).strftime('%Y-%m-%d %H:%M')), ('Last modified: ', datetime.datetime.fromtimestamp(stats.st_mtime).strftime('%Y-%m-%d %H:%M')), ] return params except Exception as e: errors.display(e, f"reading info for {name}") return [] def put_values_into_components(self, name): user_metadata = self.get_user_metadata(name) params = self.get_metadata_table(name) table = '' + "".join(f"" for name, value in params) + '
{name}{value}
' return html.escape(name), user_metadata.get('description', ''), table, self.get_card_html(name) def write_user_metadata(self, name, metadata): item = self.page.items.get(name, {}) filename = item.get("filename", None) basename, ext = os.path.splitext(filename) with open(basename + '.json', "w", encoding="utf8") as file: json.dump(metadata, file) def save_user_metadata(self, name, desc): user_metadata = self.get_user_metadata(name) user_metadata["description"] = desc self.write_user_metadata(name, user_metadata) def create_editor(self): self.create_default_editor_elems() self.create_default_buttons() self.button_edit\ .click(fn=self.put_values_into_components, inputs=[self.edit_name_input], outputs=[self.edit_name, self.edit_description, self.html_filedata, self.html_preview])\ .then(fn=lambda: gr.update(visible=True), inputs=[], outputs=[self.box]) self.button_save.click(fn=self.save_user_metadata, inputs=[self.edit_name_input, self.edit_description], outputs=[]).then(fn=None, _js="closePopup") def create_ui(self): with gr.Box(visible=False, elem_id=self.id_part, elem_classes="edit-user-metadata") as box: self.box = box self.edit_name_input = gr.Textbox("Edit user metadata card id", visible=False, elem_id=f"{self.id_part}_name") self.button_edit = gr.Button("Edit user metadata", visible=False, elem_id=f"{self.id_part}_button") self.create_editor() def save_preview(self, index, gallery, name): if len(gallery) == 0: print("There is no image in gallery to save as a preview.") return [self.get_card_html(name)] + [page.create_html(self.ui.tabname) for page in self.ui.stored_extra_pages] item = self.page.items.get(name, {}) index = int(index) index = 0 if index < 0 else index index = len(gallery) - 1 if index >= len(gallery) else index img_info = gallery[index if index >= 0 else 0] image = generation_parameters_copypaste.image_from_url_text(img_info) geninfo, items = images.read_info_from_image(image) images.save_image_with_geninfo(image, geninfo, item["local_preview"]) return [self.get_card_html(name)] + [page.create_html(self.tabname) for page in self.ui.stored_extra_pages] def setup_ui(self, gallery): self.button_replace_preview.click( fn=self.save_preview, _js="function(x, y, z){return [selected_gallery_index(), y, z]}", inputs=[self.edit_name_input, gallery, self.edit_name_input], outputs=[self.html_preview, *self.ui.pages] )