text-generation-webui/extensions/gallery/script.py

83 lines
2.5 KiB
Python
Raw Normal View History

2023-02-26 16:17:57 +01:00
from pathlib import Path
import gradio as gr
from modules.html_generator import get_image_cache
2023-02-26 16:17:57 +01:00
def generate_html():
css = """
.character-gallery {
margin: 1rem 0;
display: grid;
2023-02-27 15:30:52 +01:00
grid-template-columns: repeat(auto-fit, minmax(150px, 1fr));
grid-column-gap: 0.4rem;
grid-row-gap: 1.2rem;
}
.character-container {
cursor: pointer;
text-align: center;
position: relative;
2023-02-27 15:55:36 +01:00
opacity: 0.85;
}
.character-container:hover {
opacity: 1;
}
.character-container .placeholder, .character-container img {
width: 150px;
height: 200px;
background-color: gray;
object-fit: cover;
margin: 0 auto;
border-radius: 1rem;
border: 3px solid white;
2023-02-27 17:49:55 +01:00
box-shadow: 3px 3px 6px 0px rgb(0 0 0 / 50%);
}
.character-name {
2023-02-27 17:49:55 +01:00
margin-top: 0.3rem;
display: block;
font-size: 1.2rem;
font-weight: 600;
2023-02-27 15:46:15 +01:00
overflow-wrap: anywhere;
}
2023-02-26 16:17:57 +01:00
"""
container_html = f'<style>{css}</style><div class="character-gallery">'
2023-02-26 16:17:57 +01:00
# Iterate through files in image folder
2023-02-27 17:04:06 +01:00
for file in sorted(Path("characters").glob("*")):
2023-02-26 16:17:57 +01:00
if file.name.endswith(".json"):
character = file.name.replace(".json", "")
container_html += f'<div class="character-container" onclick=\'document.getElementById("character-menu").children[1].children[1].value = "{character}"; document.getElementById("character-menu").children[1].children[1].dispatchEvent(new Event("change"));\'>'
image_html = "<div class='placeholder'></div>"
for i in [
f"characters/{character}.png",
f"characters/{character}.jpg",
f"characters/{character}.jpeg",
]:
path = Path(i)
if path.exists():
try:
image_html = f'<img src="file/{get_image_cache(path)}">'
break
except:
continue
container_html += f'{image_html} <span class="character-name">{character}</span>'
container_html += "</div>"
2023-02-26 16:17:57 +01:00
container_html += "</div>"
return container_html
2023-02-26 16:17:57 +01:00
def ui():
with gr.Accordion("Character gallery", open=False):
2023-02-26 16:17:57 +01:00
update = gr.Button("Refresh")
gallery = gr.HTML(value=generate_html())
update.click(generate_html, [], gallery)