From 24d57a7ac3e8cc0ab8a6310b3249be59fe5ee77a Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Wed, 1 Sep 2021 14:07:56 -0400
Subject: [PATCH] Clip off ".json" from story name when downloading

---
 aiserver.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/aiserver.py b/aiserver.py
index b3e9d05d..3f111f07 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -507,7 +507,10 @@ def download():
     if(save_format == "plaintext"):
         txt = vars.prompt + "".join(vars.actions.values())
         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)
 
     # Build json to write
@@ -531,7 +534,10 @@ def download():
             })
     
     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)
 
 #============================ METHODS =============================#