rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Clip off ".json" from story name when downloading

This commit is contained in:
Gnome Ann 2021-09-01 14:07:56 -04:00
parent 7e1b1add11
commit 24d57a7ac3
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
aiserver.py
View File

@ -507,7 +507,10 @@ def download():
if(save_format == "plaintext"): if(save_format == "plaintext"):
txt = vars.prompt + "".join(vars.actions.values()) txt = vars.prompt + "".join(vars.actions.values())
save = Response(txt) save = Response(txt)
save.headers.set('Content-Disposition', 'attachment', filename='%s.txt' % path.basename(vars.savedir)) filename = path.basename(vars.savedir)
if filename[-5:] == ".json":
filename = filename[:-5]
save.headers.set('Content-Disposition', 'attachment', filename='%s.txt' % filename)
return(save) return(save)
# Build json to write # Build json to write
@ -531,7 +534,10 @@ def download():
}) })
save = Response(json.dumps(js, indent=3)) save = Response(json.dumps(js, indent=3))
save.headers.set('Content-Disposition', 'attachment', filename='%s.json' % path.basename(vars.savedir)) filename = path.basename(vars.savedir)
if filename[-5:] == ".json":
filename = filename[:-5]
save.headers.set('Content-Disposition', 'attachment', filename='%s.json' % filename)
return(save) return(save)
#============================ METHODS =============================# #============================ METHODS =============================#