mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-30 22:20:14 +01:00
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
from pathlib import Path
|
|
|
|
import gradio as gr
|
|
|
|
from modules.html_generator import image_to_base64
|
|
|
|
|
|
def generate_html():
|
|
css = """
|
|
.character-gallery {
|
|
margin: 1rem 0;
|
|
display: grid;
|
|
grid-template-columns: repeat(4, minmax(150px, 1fr));
|
|
grid-column-gap: 0.4rem;
|
|
grid-row-gap: 1.2rem;
|
|
}
|
|
|
|
.character-container {
|
|
cursor: pointer;
|
|
text-align: center;
|
|
position: relative;
|
|
opacity: 0.75;
|
|
}
|
|
|
|
.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;
|
|
box-shadow: 5px 5px 2px 0px rgb(0 0 0 / 50%);
|
|
}
|
|
|
|
.character-name {
|
|
margin-top: 0.2rem;
|
|
display: block;
|
|
font-size: 1.2rem;
|
|
font-weight: 600;
|
|
}
|
|
"""
|
|
|
|
container_html = f'<style>{css}</style><div class="character-gallery">'
|
|
|
|
# Iterate through files in image folder
|
|
for file in Path("characters").glob("*"):
|
|
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="data:image/png;base64,{image_to_base64(path)}">'
|
|
break
|
|
except:
|
|
continue
|
|
|
|
container_html += f'{image_html} <span class="character-name">{character}</span>'
|
|
container_html += "</div>"
|
|
|
|
container_html += "</div>"
|
|
return container_html
|
|
|
|
def ui():
|
|
with gr.Accordion("Character gallery"):
|
|
update = gr.Button("Refresh")
|
|
gallery = gr.HTML(value=generate_html())
|
|
update.click(generate_html, [], gallery)
|