// various functions for interaction with ui.py not large enough to warrant putting them in separate files

function set_theme(theme){
    gradioURL = window.location.href
    if (!gradioURL.includes('?__theme=')) window.location.replace(gradioURL + '?__theme=' + theme);
}

function all_gallery_buttons() {
    var allGalleryButtons = gradioApp().querySelectorAll('[style="display: block;"].tabitem div[id$=_gallery].gradio-gallery .thumbnails > .thumbnail-item.thumbnail-small');
    var visibleGalleryButtons = [];
    allGalleryButtons.forEach(function(elem) {
        if (elem.parentElement.offsetParent) visibleGalleryButtons.push(elem);
    })
    return visibleGalleryButtons;
}

function selected_gallery_button() {
    var allCurrentButtons = gradioApp().querySelectorAll('[style="display: block;"].tabitem div[id$=_gallery].gradio-gallery .thumbnail-item.thumbnail-small.selected');
    var visibleCurrentButton = null;
    allCurrentButtons.forEach(function(elem) {
        if (elem.parentElement.offsetParent) visibleCurrentButton = elem;
    })
    return visibleCurrentButton;
}

function selected_gallery_index(){
    var buttons = all_gallery_buttons();
    var button = selected_gallery_button();
    var result = -1
    buttons.forEach(function(v, i){ if(v==button) { result = i } })
    return result
}

function extract_image_from_gallery(gallery){
    if (gallery.length == 0) return [null];
    if (gallery.length == 1) return [gallery[0]];
    index = selected_gallery_index()
    if (index < 0 || index >= gallery.length) index = 0;
    return [gallery[index]];
}

function args_to_array(args){
    res = []
    for(var i=0;i<args.length;i++) res.push(args[i])
    return res
}

function switch_to_txt2img(){
    gradioApp().querySelector('#tabs').querySelectorAll('button')[0].click();
    return args_to_array(arguments);
}

function switch_to_img2img_tab(no){
    gradioApp().querySelector('#tabs').querySelectorAll('button')[1].click();
    gradioApp().getElementById('mode_img2img').querySelectorAll('button')[no].click();
}

function switch_to_img2img(){
    switch_to_img2img_tab(0);
    return args_to_array(arguments);
}

function switch_to_sketch(){
    switch_to_img2img_tab(1);
    return args_to_array(arguments);
}

function switch_to_inpaint(){
    switch_to_img2img_tab(2);
    return args_to_array(arguments);
}

function switch_to_inpaint_sketch(){
    switch_to_img2img_tab(3);
    return args_to_array(arguments);
}

function switch_to_inpaint(){
    gradioApp().querySelector('#tabs').querySelectorAll('button')[1].click();
    gradioApp().getElementById('mode_img2img').querySelectorAll('button')[2].click();
    return args_to_array(arguments);
}

function switch_to_extras(){
    gradioApp().querySelector('#tabs').querySelectorAll('button')[2].click();
    return args_to_array(arguments);
}

function get_tab_index(tabId){
    var res = 0
    gradioApp().getElementById(tabId).querySelector('div').querySelectorAll('button').forEach(function(button, i){
        if(button.className.indexOf('selected') != -1) res = i
    })
    return res
}

function create_tab_index_args(tabId, args){
    var res = []
    for(var i=0; i<args.length; i++) res.push(args[i])
    res[0] = get_tab_index(tabId)
    return res
}

function get_img2img_tab_index() {
    let res = args_to_array(arguments)
    res.splice(-2)
    res[0] = get_tab_index('mode_img2img')
    return res
}

function create_submit_args(args){
    res = []
    for(var i=0;i<args.length;i++) res.push(args[i])
    // As it is currently, txt2img and img2img send back the previous output args (txt2img_gallery, generation_info, html_info) whenever you generate a new image.
    // This can lead to uploading a huge gallery of previously generated images, which leads to an unnecessary delay between submitting and beginning to generate.
    // I don't know why gradio is sending outputs along with inputs, but we can prevent sending the image gallery here, which seems to be an issue for some.
    // If gradio at some point stops sending outputs, this may break something
    if(Array.isArray(res[res.length - 3])) res[res.length - 3] = null
    return res
}

function showSubmitButtons(tabname, show){
    gradioApp().getElementById(tabname+'_interrupt').style.display = show ? "none" : "block"
    gradioApp().getElementById(tabname+'_skip').style.display = show ? "none" : "block"
    // gradioApp().getElementById(tabname+'_interrupt').style.display = "block"
    // gradioApp().getElementById(tabname+'_skip').style.display = "block"
}

function submit(){
    rememberGallerySelection('txt2img_gallery')
    showSubmitButtons('txt2img', false)
    const id = randomId()
    const atEnd = () => showSubmitButtons('txt2img', true)
    requestProgress(id, gradioApp().getElementById('txt2img_gallery_container'), gradioApp().getElementById('txt2img_gallery'), atEnd)
    var res = create_submit_args(arguments)
    res[0] = id
    return res
}

function submit_img2img(){
    rememberGallerySelection('img2img_gallery')
    showSubmitButtons('img2img', false)
    const id = randomId()
    const atEnd = () => showSubmitButtons('img2img', true)
    requestProgress(id, gradioApp().getElementById('img2img_gallery_container'), gradioApp().getElementById('img2img_gallery'), atEnd)
    var res = create_submit_args(arguments)
    res[0] = id
    res[1] = get_tab_index('mode_img2img')
    return res
}

function modelmerger(){
    const id = randomId()
    requestProgress(id, gradioApp().getElementById('modelmerger_results_panel'), null)
    var res = create_submit_args(arguments)
    res[0] = id
    return res
}

function ask_for_style_name(_, prompt_text, negative_prompt_text) {
    name_ = prompt('Style name:')
    return [name_, prompt_text, negative_prompt_text]
}

function confirm_clear_prompt(prompt, negative_prompt) {
    if(confirm("Delete prompt?")) {
        prompt = ""
        negative_prompt = ""
    }
    return [prompt, negative_prompt]
}

promptTokecountUpdateFuncs = {}

function recalculatePromptTokens(name){
    if(promptTokecountUpdateFuncs[name]){
        promptTokecountUpdateFuncs[name]()
    }
}

function recalculate_prompts_txt2img(){
    recalculatePromptTokens('txt2img_prompt')
    recalculatePromptTokens('txt2img_neg_prompt')
    return args_to_array(arguments);
}

function recalculate_prompts_img2img(){
    recalculatePromptTokens('img2img_prompt')
    recalculatePromptTokens('img2img_neg_prompt')
    return args_to_array(arguments);
}

opts = {}
stored_opts = {}
function parseOpts(json_string){
    let settings_data = JSON.parse(json_string)
    opts = settings_data.values
    stored_opts = new Set(settings_data.stored_values)
}

onUiUpdate(function(){
    if(Object.keys(opts).length != 0) return;
    json_elem = gradioApp().getElementById('settings_json')
    if(json_elem == null) return;
    var textarea = json_elem.querySelector('textarea')
    var jsdata = textarea.value
    parseOpts(jsdata)
    executeCallbacks(optionsChangedCallbacks);
    Object.defineProperty(textarea, 'value', {
        set: function(newValue) {
            var valueProp = Object.getOwnPropertyDescriptor(HTMLTextAreaElement.prototype, 'value');
            var oldValue = valueProp.get.call(textarea);
            valueProp.set.call(textarea, newValue);
            if (oldValue != newValue) parseOpts(textarea.value)
            executeCallbacks(optionsChangedCallbacks);
        },
        get: function() {
            var valueProp = Object.getOwnPropertyDescriptor(HTMLTextAreaElement.prototype, 'value');
            return valueProp.get.call(textarea);
        }
    });
    json_elem.parentElement.style.display="none"
    function registerTextarea(id, id_counter, id_button){
        var prompt = gradioApp().getElementById(id)
        var counter = gradioApp().getElementById(id_counter)
        var textarea = gradioApp().querySelector("#" + id + " > label > textarea");
        if(counter.parentElement == prompt.parentElement) return
        prompt.parentElement.insertBefore(counter, prompt)
        prompt.parentElement.style.position = "relative"
      promptTokecountUpdateFuncs[id] = function(){ update_token_counter(id_button); }
      textarea.addEventListener("input", promptTokecountUpdateFuncs[id]);
    }
    registerTextarea('txt2img_prompt', 'txt2img_token_counter', 'txt2img_token_button')
    registerTextarea('txt2img_neg_prompt', 'txt2img_negative_token_counter', 'txt2img_negative_token_button')
    registerTextarea('img2img_prompt', 'img2img_token_counter', 'img2img_token_button')
    registerTextarea('img2img_neg_prompt', 'img2img_negative_token_counter', 'img2img_negative_token_button')
})

onOptionsChanged(function(){
    elem = gradioApp().getElementById('sd_checkpoint_hash')
    sd_checkpoint_hash = opts.sd_checkpoint_hash || ""
    shorthash = sd_checkpoint_hash.substr(0,10)

	if(elem && elem.textContent != shorthash){
	    elem.textContent = shorthash
	    elem.title = sd_checkpoint_hash
	    elem.href = "https://google.com/search?q=" + sd_checkpoint_hash
	}
})

onOptionsChanged(function(){
    setting_elems = gradioApp().querySelectorAll('#settings [id^="setting_"]')
    setting_elems.forEach(function(elem){
        setting_name = elem.id.replace("setting_", "")
        markIfModified(setting_name, opts[setting_name])
    })
})

onUiLoaded(function(){
    tab_nav_element = gradioApp().querySelector('#settings > .tab-nav')
    tab_nav_buttons = gradioApp().querySelectorAll('#settings > .tab-nav > button')
    tab_elements = gradioApp().querySelectorAll('#settings > div:not(.tab-nav)')

    // Add mutation observer to to all tab items
    // HACK to keep gradio from closing when showing all pages
    const observer = new MutationObserver(function(mutations) {
        const show_all_pages_dummy = gradioApp().getElementById('settings_show_all_pages')
        mutations.forEach(function(mutation) {
            if (mutation.type === 'attributes' && mutation.attributeName === 'style') {
                if (show_all_pages_dummy.style.display != "none") {
                    gradioApp().querySelectorAll('#settings [id^="settings_"]').forEach(function(elem){
                        elem.style.display = "block";
                    })
                }
            }
        })
    })

    // Add a wrapper for the tab content (everything but the tab nav)
    const tab_content_wrapper = document.createElement('div')
    tab_content_wrapper.className = "tab-content"
    tab_nav_element.parentElement.insertBefore(tab_content_wrapper, tab_nav_element.nextSibling)

    tab_elements.forEach(function(elem, index){
        // Add a modification indicator to the toplevel tab button
        let new_indicator = document.createElement('button')
        new_indicator.id = "modification_indicator_"+elem.id
        new_indicator.className = "modification-indicator"
        new_indicator.disabled = true
        new_indicator.onclick = () => onSettingsModificationIndicatorClicked(elem.id)

        // Add a modification indicator to the toplevel tab button
        tab_nav_element.insertBefore(new_indicator, tab_nav_buttons[index])

        // Add the tab content to the wrapper
        tab_content_wrapper.appendChild(elem)

        // Add the mutation observer to the tab element
        observer.observe(elem, { attributes: true, attributeFilter: ['style'] })
    })


    // Add the dirtyable class to the tab nav element
    tab_nav_element.classList.add('dirtyable')
})


dirty_opts = {}
function markIfModified(setting_name, value) {
    elem = gradioApp().getElementById("modification_indicator_"+setting_name)
    if(elem == null) return;
    // Use JSON.stringify to compare nested objects (e.g. arrays for checkbox-groups)
    let previous_value_json = JSON.stringify(opts[setting_name])
    let changed_value = JSON.stringify(value) != previous_value_json
    if (changed_value) {
        elem.title = `Click to revert to previous value: ${previous_value_json}`
    }

    let unsaved_value = !stored_opts.has(setting_name)
    if (unsaved_value) {
        elem.title = 'Unsaved value, apply settings to save.';
    }
    // TODO properly propagate unsaved value message to tab indicator / color marker differntly?
    let is_dirty = unsaved_value || changed_value
    elem.disabled = !is_dirty

    // Get parent tab element
    let tab_element = elem.closest('[id^="settings_"]')
    if (tab_element == null) return;
    let tab_name = tab_element.id
    if (dirty_opts[tab_name] == undefined) {
        dirty_opts[tab_name] = new Set()
    }
    is_dirty ? dirty_opts[tab_name].add(setting_name) : dirty_opts[tab_name].delete(setting_name)
    let is_tab_dirty = dirty_opts[tab_name].size > 0

    // Set the indicator on the tab nav element
    let tab_nav_indicator = tab_nav_element.querySelector('#modification_indicator_'+tab_name)
    tab_nav_indicator.disabled = !is_tab_dirty
    if (dirty_opts[tab_name].size > 1) {
        tab_nav_indicator.title = `Click to reset ${dirty_opts[tab_name].size} unapplied changes in this tab.`
    } else if (dirty_opts[tab_name].size == 1) {
        tab_nav_indicator.title = `Click to reset 1 unapplied change in this tab.`
    }
}

function onSettingComponentChanged(setting_name, value) {
    markIfModified(setting_name, value)
    return []
}

function onSettingsModificationIndicatorClicked(tab_name) {
    dirty_opts[tab_name].forEach(function(setting_name){
        // Click each setting's modification indicator
        gradioApp().getElementById("modification_indicator_"+setting_name).click()
    })
}

function onModificationIndicatorClicked(setting_name) {
    markIfModified(setting_name, opts[setting_name])
    return JSON.parse(JSON.stringify(opts[setting_name]));
}

let txt2img_textarea, img2img_textarea = undefined;
let wait_time = 800
let token_timeouts = {};

function update_txt2img_tokens(...args) {
	update_token_counter("txt2img_token_button")
	if (args.length == 2)
		return args[0]
	return args;
}

function update_img2img_tokens(...args) {
	update_token_counter("img2img_token_button")
	if (args.length == 2)
		return args[0]
	return args;
}

function getTranslation(...args) {
  return null
}

function update_token_counter(button_id) {
	if (token_timeouts[button_id])
		clearTimeout(token_timeouts[button_id]);
	token_timeouts[button_id] = setTimeout(() => gradioApp().getElementById(button_id)?.click(), wait_time);
}

function restart_reload(){
    document.body.innerHTML='<h1 style="font-family:monospace;margin-top:20%;color:lightgray;text-align:center;">Reloading...</h1>';
    setTimeout(function(){location.reload()},8000)
    return []
}

// Simulate an `input` DOM event for Gradio Textbox component. Needed after you edit its contents in javascript, otherwise your edits
// will only visible on web page and not sent to python.
function updateInput(target){
	let e = new Event("input", { bubbles: true })
	Object.defineProperty(e, "target", {value: target})
	target.dispatchEvent(e);
}

var desiredCheckpointName = null;
function selectCheckpoint(name){
    desiredCheckpointName = name;
    gradioApp().getElementById('change_checkpoint').click()
}

function create_theme_element() {
  el = document.createElement('img');
  el.id = 'theme-preview';
  el.className = 'theme-preview';
  el.onclick = () => el.style.display = 'none';
  document.body.appendChild(el);
  return el;
}

function preview_theme() {
  const name = gradioApp().getElementById('setting_gradio_theme').querySelectorAll('span')[1].innerText; // ugly but we want current value without the need to set apply
  if (name === 'black-orange' || name.startsWith('gradio/')) {
    el = document.getElementById('theme-preview') || create_theme_element();
    el.style.display = el.style.display === 'block' ? 'none' : 'block';
    if (name === 'black-orange') el.src = '/file=javascript/black-orange.jpg';
    else el.src = `/file=javascript/${name.replace('/', '-')}.jpg`;
  } else {
    fetch('/file=javascript/themes.json')
      .then((r) => r.json())
        .then(themes => {
          theme = themes.find((t)=> t.id === name);
          window.open(theme.subdomain, '_blank');
      });
  }
}

function reconnect_ui() {
  const el1 = gradioApp().getElementById('txt2img_gallery_container')
  const el2 = gradioApp().getElementById('txt2img_gallery')
  const task_id = localStorage.getItem('task')
  if (!el1 || !el2) return
  else clearInterval(start_check)
  if (task_id) {
    console.debug('task check:', task_id)
    rememberGallerySelection('txt2img_gallery')
    showSubmitButtons('txt2img', false)
    const atEnd = () => showSubmitButtons('txt2img', true)
    requestProgress(task_id, el1, el2, atEnd, null, true)
  }

  sd_model = gradioApp().getElementById("setting_sd_model_checkpoint")
  let loadingStarted = 0;
  let loadingMonitor = 0;
  const sd_model_callback = () => {
    loading = sd_model.querySelector(".eta-bar")
    if (!loading) {
      loadingStarted = 0
      clearInterval(loadingMonitor)
    } else {
      if (loadingStarted === 0) {
        loadingStarted = Date.now();
        loadingMonitor = setInterval(() => {
          elapsed = Date.now() - loadingStarted;
          console.log('Loading', elapsed)
          if (elapsed > 3000 && loading) loading.style.display = 'none';
        }, 5000);
      }
    }
  };
  const sd_model_observer = new MutationObserver(sd_model_callback);
  sd_model_observer.observe(sd_model, { attributes: true, childList: true, subtree: true });
}

var start_check = setInterval(reconnect_ui, 50);