Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 17:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Make the gallery interactive to load characters

Browse Source
This commit is contained in:
oobabooga 2023-02-26 17:19:36 -03:00
parent 8e3e8a070f
commit 7a776ccf87
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
extensions/gallery/script.py
View File

@ -29,6 +29,10 @@ def generate_html():
background-color: gray; background-color: gray;
} }
.character-gallery tr {
cursor: pointer;
}
.character-gallery .image-td { .character-gallery .image-td {
width: 150px; width: 150px;
} }
@ -45,7 +49,7 @@ def generate_html():
for file in Path("characters").glob("*"): for file in Path("characters").glob("*"):
if file.name.endswith(".json"): if file.name.endswith(".json"):
character = file.name.replace(".json", "") character = file.name.replace(".json", "")
table_html += "<tr>" table_html += f'<tr 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>" image_html = "<div class='placeholder'></div>"
for i in [ for i in [
@ -62,7 +66,7 @@ def generate_html():
except: except:
continue continue
table_html += f'<td class="image-td"=>{image_html}</td><td class="character-td">{character}</td>' table_html += f'<td class="image-td">{image_html}</td><td class="character-td">{character}</td>'
table_html += "</tr>" table_html += "</tr>"
table_html += "</table></div>" table_html += "</table></div>"

2
server.py
View File

@ -207,7 +207,7 @@ if shared.args.chat or shared.args.cai_chat:
shared.gradio['name2'] = gr.Textbox(value=shared.settings[f'name2{suffix}'], lines=1, label='Bot\'s name') shared.gradio['name2'] = gr.Textbox(value=shared.settings[f'name2{suffix}'], lines=1, label='Bot\'s name')
shared.gradio['context'] = gr.Textbox(value=shared.settings[f'context{suffix}'], lines=2, label='Context') shared.gradio['context'] = gr.Textbox(value=shared.settings[f'context{suffix}'], lines=2, label='Context')
with gr.Row(): with gr.Row():
shared.gradio['character_menu'] = gr.Dropdown(choices=available_characters, value='None', label='Character') shared.gradio['character_menu'] = gr.Dropdown(choices=available_characters, value='None', label='Character', elem_id='character-menu')
ui.create_refresh_button(shared.gradio['character_menu'], lambda : None, lambda : {'choices': get_available_characters()}, 'refresh-button') ui.create_refresh_button(shared.gradio['character_menu'], lambda : None, lambda : {'choices': get_available_characters()}, 'refresh-button')
with gr.Row(): with gr.Row():