From d75284611643183786e66a4b9627d9292248f77c Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Tue, 7 Jan 2025 22:11:15 +0800
Subject: [PATCH] fixed ask save file

---
 koboldcpp.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/koboldcpp.py b/koboldcpp.py
index 33b9f6559..566299fb7 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -3582,7 +3582,9 @@ def show_gui():
         filename = asksaveasfile(filetypes=file_type, defaultextension=file_type)
         if filename is None:
             return
-        file = open(str(filename.name), 'a')
+        filenamestr = str(filename.name).strip()
+        filenamestr = f"{filenamestr}.kcppt" if ".kcppt" not in filenamestr.lower() else filenamestr
+        file = open(filenamestr, 'a')
         file.write(json.dumps(savdict))
         file.close()
         pass
@@ -3936,7 +3938,9 @@ def show_gui():
         filename = asksaveasfile(filetypes=file_type, defaultextension=file_type)
         if filename is None:
             return
-        file = open(str(filename.name), 'a')
+        filenamestr = str(filename.name).strip()
+        filenamestr = f"{filenamestr}.kcpps" if ".kcpps" not in filenamestr.lower() else filenamestr
+        file = open(filenamestr, 'a')
         file.write(json.dumps(savdict))
         file.close()
         pass