mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-02-26 00:17:41 +01:00
Fix for aidg.club website being taken read-only
This commit is contained in:
ebolam
parent
68d143b80c
commit
23a031d852
166
aiserver.py
166
aiserver.py
@ -5782,85 +5782,101 @@ def importgame():
|
|||||||
def importAidgRequest(id):
|
def importAidgRequest(id):
|
||||||
exitModes()
|
exitModes()
|
||||||
|
|
||||||
urlformat = "https://prompts.aidg.club/api/"
|
#prompts.aidg.club is dead now. They are serving up a sqllite db, so we'll download that if it's not already saved and use that instead
|
||||||
req = requests.get(urlformat+id)
|
#urlformat = "https://prompts.aidg.club/api/"
|
||||||
|
#req = requests.get(urlformat+id)
|
||||||
|
|
||||||
|
if not os.path.exists("aidgclub.db"):
|
||||||
|
import urllib.request
|
||||||
|
urllib.request.urlretrieve("https://prompts.aidg.club/backup.db", "aidgclub.db")
|
||||||
|
|
||||||
if(req.status_code == 200):
|
import sqlite3
|
||||||
js = req.json()
|
con = sqlite3.connect("aidgclub.db")
|
||||||
|
cur = con.cursor()
|
||||||
|
js = {}
|
||||||
|
sql = "select * from Prompts where CorrelationId = {}".format(id)
|
||||||
|
for row in cur.execute(sql):
|
||||||
|
js['promptContent'] = row[6]
|
||||||
|
js['memory'] = row[3]
|
||||||
|
js['authorsNote'] = row[1]
|
||||||
|
js['worldInfos'] = []
|
||||||
|
for row in cur.execute("select * from WorldInfos where PromptId = {}".format(id)):
|
||||||
|
js['worldInfos'].append({'keys': row[2], 'entry': row[1]})
|
||||||
|
|
||||||
|
|
||||||
# Import game state
|
# Import game state
|
||||||
vars.gamestarted = True
|
vars.gamestarted = True
|
||||||
vars.prompt = js["promptContent"]
|
vars.prompt = js["promptContent"]
|
||||||
vars.memory = js["memory"]
|
vars.memory = js["memory"]
|
||||||
vars.authornote = js["authorsNote"]
|
vars.authornote = js["authorsNote"]
|
||||||
vars.authornotetemplate = "[Author's note: <|>]"
|
vars.authornotetemplate = "[Author's note: <|>]"
|
||||||
vars.actions = structures.KoboldStoryRegister()
|
vars.actions = structures.KoboldStoryRegister()
|
||||||
vars.actions_metadata = {}
|
vars.actions_metadata = {}
|
||||||
vars.worldinfo = []
|
vars.worldinfo = []
|
||||||
vars.worldinfo_i = []
|
vars.worldinfo_i = []
|
||||||
vars.worldinfo_u = {}
|
vars.worldinfo_u = {}
|
||||||
vars.wifolders_d = {}
|
vars.wifolders_d = {}
|
||||||
vars.wifolders_l = []
|
vars.wifolders_l = []
|
||||||
vars.wifolders_u = {uid: [] for uid in vars.wifolders_d}
|
vars.wifolders_u = {uid: [] for uid in vars.wifolders_d}
|
||||||
vars.lastact = ""
|
vars.lastact = ""
|
||||||
vars.submission = ""
|
vars.submission = ""
|
||||||
vars.lastctx = ""
|
vars.lastctx = ""
|
||||||
|
|
||||||
if not vars.memory:
|
if not vars.memory:
|
||||||
vars.memory = ""
|
vars.memory = ""
|
||||||
if not vars.authornote:
|
if not vars.authornote:
|
||||||
vars.authornote = ""
|
vars.authornote = ""
|
||||||
|
|
||||||
num = 0
|
num = 0
|
||||||
for wi in js["worldInfos"]:
|
for wi in js["worldInfos"]:
|
||||||
vars.worldinfo.append({
|
vars.worldinfo.append({
|
||||||
"key": wi["keys"],
|
"key": wi["keys"],
|
||||||
"keysecondary": wi.get("keysecondary", ""),
|
"keysecondary": wi.get("keysecondary", ""),
|
||||||
"content": wi["entry"],
|
"content": wi["entry"],
|
||||||
"comment": wi.get("comment", ""),
|
"comment": wi.get("comment", ""),
|
||||||
"folder": wi.get("folder", None),
|
"folder": wi.get("folder", None),
|
||||||
"num": num,
|
"num": num,
|
||||||
"init": True,
|
"init": True,
|
||||||
"selective": wi.get("selective", False),
|
"selective": wi.get("selective", False),
|
||||||
"constant": wi.get("constant", False),
|
"constant": wi.get("constant", False),
|
||||||
"uid": None,
|
"uid": None,
|
||||||
})
|
})
|
||||||
while(True):
|
while(True):
|
||||||
uid = int.from_bytes(os.urandom(4), "little", signed=True)
|
uid = int.from_bytes(os.urandom(4), "little", signed=True)
|
||||||
if(uid not in vars.worldinfo_u):
|
if(uid not in vars.worldinfo_u):
|
||||||
break
|
break
|
||||||
vars.worldinfo_u[uid] = vars.worldinfo[-1]
|
vars.worldinfo_u[uid] = vars.worldinfo[-1]
|
||||||
vars.worldinfo[-1]["uid"] = uid
|
vars.worldinfo[-1]["uid"] = uid
|
||||||
if(vars.worldinfo[-1]["folder"]) is not None:
|
if(vars.worldinfo[-1]["folder"]) is not None:
|
||||||
vars.wifolders_u[vars.worldinfo[-1]["folder"]].append(vars.worldinfo[-1])
|
vars.wifolders_u[vars.worldinfo[-1]["folder"]].append(vars.worldinfo[-1])
|
||||||
num += 1
|
num += 1
|
||||||
|
|
||||||
for uid in vars.wifolders_l + [None]:
|
for uid in vars.wifolders_l + [None]:
|
||||||
vars.worldinfo.append({"key": "", "keysecondary": "", "content": "", "comment": "", "folder": uid, "num": None, "init": False, "selective": False, "constant": False, "uid": None})
|
vars.worldinfo.append({"key": "", "keysecondary": "", "content": "", "comment": "", "folder": uid, "num": None, "init": False, "selective": False, "constant": False, "uid": None})
|
||||||
while(True):
|
while(True):
|
||||||
uid = int.from_bytes(os.urandom(4), "little", signed=True)
|
uid = int.from_bytes(os.urandom(4), "little", signed=True)
|
||||||
if(uid not in vars.worldinfo_u):
|
if(uid not in vars.worldinfo_u):
|
||||||
break
|
break
|
||||||
vars.worldinfo_u[uid] = vars.worldinfo[-1]
|
vars.worldinfo_u[uid] = vars.worldinfo[-1]
|
||||||
vars.worldinfo[-1]["uid"] = uid
|
vars.worldinfo[-1]["uid"] = uid
|
||||||
if(vars.worldinfo[-1]["folder"] is not None):
|
if(vars.worldinfo[-1]["folder"] is not None):
|
||||||
vars.wifolders_u[vars.worldinfo[-1]["folder"]].append(vars.worldinfo[-1])
|
vars.wifolders_u[vars.worldinfo[-1]["folder"]].append(vars.worldinfo[-1])
|
||||||
stablesortwi()
|
stablesortwi()
|
||||||
vars.worldinfo_i = [wi for wi in vars.worldinfo if wi["init"]]
|
vars.worldinfo_i = [wi for wi in vars.worldinfo if wi["init"]]
|
||||||
|
|
||||||
# Reset current save
|
# Reset current save
|
||||||
vars.savedir = getcwd()+"\\stories"
|
vars.savedir = getcwd()+"\\stories"
|
||||||
|
|
||||||
# Refresh game screen
|
# Refresh game screen
|
||||||
vars.laststory = None
|
vars.laststory = None
|
||||||
emit('from_server', {'cmd': 'setstoryname', 'data': vars.laststory}, broadcast=True)
|
emit('from_server', {'cmd': 'setstoryname', 'data': vars.laststory}, broadcast=True)
|
||||||
setgamesaved(False)
|
setgamesaved(False)
|
||||||
sendwi()
|
sendwi()
|
||||||
emit('from_server', {'cmd': 'setmemory', 'data': vars.memory}, broadcast=True)
|
emit('from_server', {'cmd': 'setmemory', 'data': vars.memory}, broadcast=True)
|
||||||
emit('from_server', {'cmd': 'setanote', 'data': vars.authornote}, broadcast=True)
|
emit('from_server', {'cmd': 'setanote', 'data': vars.authornote}, broadcast=True)
|
||||||
emit('from_server', {'cmd': 'setanotetemplate', 'data': vars.authornotetemplate}, broadcast=True)
|
emit('from_server', {'cmd': 'setanotetemplate', 'data': vars.authornotetemplate}, broadcast=True)
|
||||||
refresh_story()
|
refresh_story()
|
||||||
emit('from_server', {'cmd': 'setgamestate', 'data': 'ready'}, broadcast=True)
|
emit('from_server', {'cmd': 'setgamestate', 'data': 'ready'}, broadcast=True)
|
||||||
|
|
||||||
#==================================================================#
|
#==================================================================#
|
||||||
# Import World Info JSON file
|
# Import World Info JSON file
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user