semi real solution

This commit is contained in:
Miao Xiang
2023-03-26 15:10:16 -07:00
parent cd4fcd2f0c
commit a6b30069f0
4 changed files with 46 additions and 21 deletions
+27
View File
@@ -163,6 +163,7 @@ class ExtraNetworksPage:
"name": item["name"],
"description": (item.get("description") or ""),
"card_clicked": onclick,
"save_card_description": '"' + html.escape(f"""return saveCardDescription(event, {json.dumps(tabname)}, {json.dumps(item["local_preview"])})""") + '"',
"save_card_preview": '"' + html.escape(f"""return saveCardPreview(event, {json.dumps(tabname)}, {json.dumps(item["local_preview"])})""") + '"',
"search_term": item.get("search_term", ""),
"metadata_button": metadata_button,
@@ -212,6 +213,11 @@ class ExtraNetworksUi:
self.button_save_preview = None
self.preview_target_filename = None
self.button_save_description = None
self.description_target_filename = None
self.description_input = None
self.tabname = None
@@ -250,6 +256,10 @@ def create_ui(container, button, tabname):
ui.button_save_preview = gr.Button('Save preview', elem_id=tabname+"_save_preview", visible=False)
ui.preview_target_filename = gr.Textbox('Preview save filename', elem_id=tabname+"_preview_filename", visible=False)
ui.button_save_description = gr.Button('Save description', elem_id=tabname+"_save_description", visible=False)
ui.description_target_filename = gr.Textbox('Description save filename', elem_id=tabname+"_description_filename", visible=False)
ui.description_input = gr.Text(elem_id=tabname+"_description_input")
def toggle_visibility(is_visible):
is_visible = not is_visible
return is_visible, gr.update(visible=is_visible)
@@ -310,3 +320,20 @@ def setup_ui(ui, gallery):
outputs=[*ui.pages]
)
def save_description(filenamex,images,filename,descrip):
filename = filename.split('.')[0]+".description.txt"
file1 = open(filename,'w')
print(file1)
file1.write(descrip)
file1.close()
return [page.create_html(ui.tabname) for page in ui.stored_extra_pages]
ui.button_save_description.click(
fn=save_description,
_js="function(x,y,z){return selected_gallery_index(), y, z]}",
inputs=[ui.description_target_filename, gallery, ui.description_target_filename, ui.description_input],
outputs=[*ui.pages]
)