54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from modules import shared, script_callbacks
|
|
from modules import script_callbacks
|
|
import gradio as gr
|
|
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
js = """function close_window() {
|
|
if (confirm("Close the application?")) {
|
|
|
|
document.body.style.display = 'none';
|
|
var messageDiv = document.createElement('div');
|
|
messageDiv.textContent = 'You may close this window';
|
|
messageDiv.style.textAlign = 'center';
|
|
messageDiv.style.fontWeight = 'bold';
|
|
|
|
// Set the background color
|
|
document.body.style.backgroundColor = '#f2f2f2'; // Greyish background color
|
|
|
|
// Append the message div to the body
|
|
document.body.appendChild(messageDiv);
|
|
return true;
|
|
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
"""
|
|
|
|
def stop_button(component, **kwargs):
|
|
after_this_compo = "setting_{}".format(shared.opts.data["quicksettings_list"][-1])
|
|
|
|
# print(after_this_compo)
|
|
|
|
if kwargs.get("elem_id") == after_this_compo:
|
|
with gr.Row(elem_id="quicksettings", variant="compact"):
|
|
btn = gr.Button("Exit ⭕", elem_id="stop_button",size="sm",variant='stop')
|
|
hidden_checkbox = gr.Checkbox(visible=False)
|
|
|
|
def when(hidden_state):
|
|
if hidden_state:
|
|
os._exit(0)
|
|
return False
|
|
|
|
btn.click(when, hidden_checkbox, hidden_checkbox, _js=js)
|
|
|
|
|
|
|
|
|
|
script_callbacks.on_after_component(stop_button)
|