88 lines
4.1 KiB
Python
88 lines
4.1 KiB
Python
# Stable Horde for Web UI, a Stable Horde client for AUTOMATIC1111's Stable Diffusion Web UI
|
|
# Copyright (C) 2022 Natan Junges <natanajunges@gmail.com>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
from modules import script_callbacks, scripts, ui
|
|
import gradio
|
|
import json
|
|
import os.path
|
|
|
|
settings_file = os.path.join(scripts.basedir(), "settings.json")
|
|
|
|
def settings():
|
|
with gradio.Blocks(analytics_enabled=False) as settings_tab:
|
|
api_endpoint = gradio.Textbox(max_lines=1, placeholder="https://stablehorde.net/api", label="API endpoint", interactive=True)
|
|
api_key = gradio.Textbox(max_lines=1, placeholder="0000000000", label="API key", interactive=True)
|
|
censor_nsfw = gradio.Checkbox(label="Censor NSFW when NSFW is disabled", interactive=True)
|
|
trusted_workers = gradio.Checkbox(label="Only send requests to trusted workers", interactive=True)
|
|
workers = gradio.Textbox(max_lines=1, label="Only send requests to these workers", interactive=True)
|
|
|
|
with gradio.Row():
|
|
reset = gradio.Button(value=ui.reuse_symbol + " Reset settings")
|
|
reload = gradio.Button(value=ui.refresh_symbol + " Reload settings")
|
|
apply = gradio.Button(value=ui.save_style_symbol + " Apply settings", variant="primary")
|
|
|
|
def reset_click():
|
|
opts = {
|
|
"api_endpoint": "https://stablehorde.net/api",
|
|
"api_key": "0000000000",
|
|
"censor_nsfw": True,
|
|
"trusted_workers": True,
|
|
"workers": []
|
|
}
|
|
|
|
return (gradio.update(value=opts["api_endpoint"]), gradio.update(value=opts["api_key"]), gradio.update(value=opts["censor_nsfw"]), gradio.update(value=opts["trusted_workers"]), gradio.update(value=", ".join(opts["workers"])))
|
|
|
|
def reload_click():
|
|
if os.path.exists(settings_file):
|
|
with open(settings_file) as file:
|
|
opts = json.load(file)
|
|
|
|
return (gradio.update(value=opts["api_endpoint"]), gradio.update(value=opts["api_key"]), gradio.update(value=opts["censor_nsfw"]), gradio.update(value=opts["trusted_workers"]), gradio.update(value=", ".join(opts["workers"])))
|
|
else:
|
|
return reset_click()
|
|
|
|
def apply_click(api_endpoint, api_key, censor_nsfw, trusted_workers, workers):
|
|
if len(api_endpoint) == 0:
|
|
api_endpoint = "https://stablehorde.net/api"
|
|
|
|
if len(api_key) == 0:
|
|
api_key = "0000000000"
|
|
|
|
if len(workers) == 0:
|
|
workers = []
|
|
else:
|
|
workers = list(map(lambda w: w.strip(), workers.split(",")))
|
|
|
|
opts = {
|
|
"api_endpoint": api_endpoint,
|
|
"api_key": api_key,
|
|
"censor_nsfw": censor_nsfw,
|
|
"trusted_workers": trusted_workers,
|
|
"workers": workers
|
|
}
|
|
|
|
with open(settings_file, "w") as file:
|
|
json.dump(opts, file)
|
|
|
|
reset.click(fn=reset_click, outputs=[api_endpoint, api_key, censor_nsfw, trusted_workers, workers])
|
|
reload.click(fn=reload_click, outputs=[api_endpoint, api_key, censor_nsfw, trusted_workers, workers])
|
|
apply.click(fn=apply_click, inputs=[api_endpoint, api_key, censor_nsfw, trusted_workers, workers])
|
|
settings_tab.load(fn=reload_click, outputs=[api_endpoint, api_key, censor_nsfw, trusted_workers, workers])
|
|
|
|
return [(settings_tab, "Stable Horde Settings", "stable_horde_settings")]
|
|
|
|
script_callbacks.on_ui_tabs(settings)
|