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

Clean up code for saving story as plaintext

Browse Source
This commit is contained in:
Gnome Ann
2021-09-01 13:49:04 -04:00
parent 543acf9ba4
commit 8ae9304cda
1 changed files with 2 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
aiserver.py
View File

@@ -504,19 +504,7 @@ def download():
save_format = request.args.get("format", "json").strip().lower()
if(save_format == "plaintext"):
ln = len(vars.actions)
if(ln > 0):
chunks = collections.deque()
for key in reversed(vars.actions):
chunk = vars.actions[key]
chunks.appendleft(chunk)
if(ln > 0):
txt = vars.prompt + "".join(chunks)
elif(ln == 0):
txt = vars.prompt
txt = vars.prompt + "".join(vars.actions.values())
save = Response(txt)
save.headers.set('Content-Disposition', 'attachment', filename='%s.txt' % path.basename(vars.savedir))
return(save)
@@ -1981,18 +1969,7 @@ def saveRequest(savpath):
"constant": wi["constant"]
})
ln = len(vars.actions)
if(ln > 0):
chunks = collections.deque()
for key in reversed(vars.actions):
chunk = vars.actions[key]
chunks.appendleft(chunk)
if(ln > 0):
txt = vars.prompt + "".join(chunks)
elif(ln == 0):
txt = vars.prompt
txt = vars.prompt + "".join(vars.actions.values())
# Write it
try: