rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Cookie Saving for Colab - Needs work

Browse Source
This commit is contained in:
ebolam
2022-09-07 20:12:21 -04:00
parent cd67013e3e
commit 9e66f42f5d
3 changed files with 42 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
aiserver.py
View File

@@ -6776,7 +6776,7 @@ def new_ui_index():
def ui2_connect():
#Send all variables to client
koboldai_vars.send_to_ui()
UI_2_load_tweaks()
UI_2_load_cookies()
pass
#==================================================================#
@@ -7692,19 +7692,22 @@ def UI_2_theme_list_refresh(data):
#==================================================================#
# Save Tweaks
#==================================================================#
@socketio.on('save_tweaks')
def UI_2_save_tweaks(data):
with open("./settings/tweaks.settings", "w") as f:
f.write(data)
@socketio.on('save_cookies')
def UI_2_save_cookies(data):
for key in data:
#Note this won't sync to the client automatically as we're modifying a variable rather than setting it
koboldai_vars.cookies[key] = data[key]
with open("./settings/cookies.settings", "w") as f:
f.write(json.dumps(koboldai_vars.cookies))
#==================================================================#
# Load Tweaks
#==================================================================#
def UI_2_load_tweaks():
def UI_2_load_cookies():
if koboldai_vars.on_colab:
if os.path.exists("./settings/tweaks.settings"):
with open("./settings/tweaks.settings", "r") as f:
socketio.emit('load_tweaks', f.read(), room="UI2")
if os.path.exists("./settings/cookies.settings"):
with open("./settings/cookies.settings", "r") as f:
socketio.emit('load_cookies', json.load(f), room="UI2")
#==================================================================#
# Test