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

34 lines
1.3 KiB
JavaScript
Raw Normal View History

2023-08-13 06:12:15 +02:00
let gallery_element = document.getElementById('gallery-extension');
let chat_mode_element = document.getElementById('chat-mode');
2023-08-21 06:02:53 +02:00
let extensions_block = document.getElementById('extensions');
let extensions_block_size = extensions_block.childNodes.length;
let gallery_only = (extensions_block_size == 5);
2023-08-13 06:12:15 +02:00
2023-08-21 06:02:53 +02:00
document.querySelector('.header_bar').addEventListener('click', function(event) {
if (event.target.tagName === 'BUTTON') {
const buttonText = event.target.textContent.trim();
2023-08-13 06:12:15 +02:00
2023-08-21 06:02:53 +02:00
let chat_visible = (buttonText == 'Chat');
let default_visible = (buttonText == 'Default');
let notebook_visible = (buttonText == 'Notebook');
let chat_mode_visible = (chat_mode_element.offsetHeight > 0 && chat_mode_element.offsetWidth > 0);
// Only show this extension in the Chat tab
if (chat_visible) {
if (chat_mode_visible) {
gallery_element.style.display = 'block';
extensions_block.style.display = '';
} else {
gallery_element.style.display = 'none';
extensions_block.style.display = 'none';
}
} else {
gallery_element.style.display = 'none';
2023-08-21 06:02:53 +02:00
if (gallery_only) {
extensions_block.style.display = 'none';
}
}
2023-08-13 06:12:15 +02:00
}
});