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

Rewrite tab detection js

Browse Source
This commit is contained in:
oobabooga 2023-08-20 21:02:53 -07:00
parent 57036abc76
commit 6394fef1db
2 changed files with 40 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
extensions/gallery/script.js
View File

@ -1,15 +1,18 @@
let gallery_element = document.getElementById('gallery-extension'); let gallery_element = document.getElementById('gallery-extension');
let chat_mode_element = document.getElementById('chat-mode'); let chat_mode_element = document.getElementById('chat-mode');
let extensions_block = gallery_element.parentElement; let extensions_block = document.getElementById('extensions');
let extensions_block_size = extensions_block.childNodes.length; let extensions_block_size = extensions_block.childNodes.length;
let gallery_only = (extensions_block_size == 5); let gallery_only = (extensions_block_size == 5);
main_parent.addEventListener('click', function(e) { document.querySelector('.header_bar').addEventListener('click', function(event) {
let chat_visible = (chat_tab.offsetHeight > 0 && chat_tab.offsetWidth > 0); if (event.target.tagName === 'BUTTON') {
const buttonText = event.target.textContent.trim();
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); let chat_mode_visible = (chat_mode_element.offsetHeight > 0 && chat_mode_element.offsetWidth > 0);
let notebook_visible = (notebook_tab.offsetHeight > 0 && notebook_tab.offsetWidth > 0);
let default_visible = (default_tab.offsetHeight > 0 && default_tab.offsetWidth > 0);
// Only show this extension in the Chat tab // Only show this extension in the Chat tab
if (chat_visible) { if (chat_visible) {
@ -26,4 +29,5 @@ main_parent.addEventListener('click', function(e) {
extensions_block.style.display = 'none'; extensions_block.style.display = 'none';
} }
} }
}
}); });

19
js/main.js
View File

@ -1,19 +1,17 @@
let chat_tab = document.getElementById('chat-tab'); let main_parent = document.getElementById('chat-tab').parentNode;
let notebook_tab = document.getElementById('notebook-tab');
let default_tab = document.getElementById('default-tab');
let main_parent = chat_tab.parentNode;
let extensions = document.getElementById('extensions'); let extensions = document.getElementById('extensions');
main_parent.childNodes[0].classList.add("header_bar"); main_parent.childNodes[0].classList.add("header_bar");
main_parent.style = "padding: 0; margin: 0"; main_parent.style = "padding: 0; margin: 0";
main_parent.parentNode.parentNode.style = "padding: 0"; main_parent.parentNode.parentNode.style = "padding: 0";
// Add an event listener to the generation tabs document.querySelector('.header_bar').addEventListener('click', function(event) {
main_parent.addEventListener('click', function(e) { if (event.target.tagName === 'BUTTON') {
let chat_visible = (chat_tab.offsetHeight > 0 && chat_tab.offsetWidth > 0); const buttonText = event.target.textContent.trim();
let notebook_visible = (notebook_tab.offsetHeight > 0 && notebook_tab.offsetWidth > 0);
let default_visible = (default_tab.offsetHeight > 0 && default_tab.offsetWidth > 0); let chat_visible = (buttonText == 'Chat');
let default_visible = (buttonText == 'Default');
let notebook_visible = (buttonText == 'Notebook');
// Check if one of the generation tabs is visible // Check if one of the generation tabs is visible
if (chat_visible || notebook_visible || default_visible) { if (chat_visible || notebook_visible || default_visible) {
@ -28,6 +26,7 @@ main_parent.addEventListener('click', function(e) {
} else { } else {
extensions.style.display = 'none'; extensions.style.display = 'none';
} }
}
}); });
//------------------------------------------------ //------------------------------------------------