From 183e5d9deafc59b6bde45795730f7eb6a9668bc0 Mon Sep 17 00:00:00 2001 From: ebolam Date: Wed, 7 Sep 2022 09:09:40 -0400 Subject: [PATCH] Saving tweaks on colab --- aiserver.py | 18 +++++++++++++++++- static/koboldai.js | 8 ++++++++ templates/index_new.html | 1 + templates/popups.html | 10 +++++----- 4 files changed, 31 insertions(+), 6 deletions(-) diff --git a/aiserver.py b/aiserver.py index 74bb498c..dbecd5e3 100644 --- a/aiserver.py +++ b/aiserver.py @@ -6771,7 +6771,7 @@ def new_ui_index(): def ui2_connect(): #Send all variables to client koboldai_vars.send_to_ui() - + UI_2_load_tweaks() pass #==================================================================# @@ -7684,6 +7684,22 @@ def UI_2_sp_list_refresh(data): def UI_2_theme_list_refresh(data): koboldai_vars.theme_list = [".".join(f.split(".")[:-1]) for f in os.listdir("./themes") if os.path.isfile(os.path.join("./themes", f))] +#==================================================================# +# Save Tweaks +#==================================================================# +@socketio.on('save_tweaks') +def UI_2_save_tweaks(data): + with open("./settings/tweaks.settings", "w") as f: + f.write(data) + +#==================================================================# +# Load Tweaks +#==================================================================# +def UI_2_load_tweaks(): + if os.path.exists("./settings/tweaks.settings"): + with open("./settings/tweaks.settings", "r") as f: + socketio.emit('load_tweaks', f.read(), room="UI2") + #==================================================================# # Test #==================================================================# diff --git a/static/koboldai.js b/static/koboldai.js index 7541d118..f9955f2b 100644 --- a/static/koboldai.js +++ b/static/koboldai.js @@ -25,6 +25,7 @@ socket.on("world_info_folder", function(data){world_info_folder(data);}); socket.on("delete_new_world_info_entry", function(data){document.getElementById("world_info_-1").remove();}); socket.on("delete_world_info_entry", function(data){document.getElementById("world_info_"+data).remove();}); socket.on("error", function(data){show_error_message(data);}); +socket.on('load_tweaks', function(data){load_tweaks(data);}); //socket.onAny(function(event_name, data) {console.log({"event": event_name, "class": data.classname, "data": data});}); var presets = {}; @@ -2007,6 +2008,10 @@ function send_world_info(uid) { socket.emit("edit_world_info", world_info_data[uid]); } +function load_tweaks(data) { + +} + //--------------------------------------------General UI Functions------------------------------------ function autoResize(element) { element.style.height = 'auto'; @@ -3356,6 +3361,9 @@ $(document).ready(function(){ } setCookie("enabledTweaks", JSON.stringify(out)); + if (document.getElementById("on_colab").textContent == "true") { + socket.emit("save_tweaks", JSON.stringify(out)); + } } diff --git a/templates/index_new.html b/templates/index_new.html index 1af01ab6..f8be2eec 100644 --- a/templates/index_new.html +++ b/templates/index_new.html @@ -19,6 +19,7 @@ +
diff --git a/templates/popups.html b/templates/popups.html index b3d4e86d..7b98378a 100644 --- a/templates/popups.html +++ b/templates/popups.html @@ -65,8 +65,8 @@ The story name you have entered already exists. Would you like to overwrite?
- - + +
@@ -81,8 +81,8 @@
- - + +
@@ -95,7 +95,7 @@
- +