Update turbowarp-server/gpt.py

turbowarp-server/gpt.py

@@ -26,14 +26,20 @@ with open("turbowarp-server/n-chars.txt", encoding="utf8") as f:
     for line in f:
         chars.append(line.strip())
 
+# ---------------------
+# 文字エンコード / デコード
+# ---------------------
+
 def encode(text):
     out = ""
     for c in text:
-        if c in chars:
+        if c == " ":
+            out += "96"       # 半角スペースは固定で 96
+        elif c in chars:
             i = chars.index(c)
             out += f"{i:02d}"
         elif c == "\n":
-            out += "98"
+            out += "98"       # 改行
     return out
 
 def decode(data):
@@ -43,19 +49,18 @@ def decode(data):
     for i in range(0, len(data), 2):
         num = int(data[i:i+2])
 
-        if num == 99:  # 履歴区切り
+        if num == 99:          # 履歴区切り
             tokens.append(current)
             current = ""
-
-        elif num == 98:  # 改行
+        elif num == 98:        # 改行
             current += "\n"
-
+        elif num == 96:        # 半角スペース
+            current += " "
         else:
             if num < len(chars):
                 current += chars[num]
 
     tokens.append(current)
-
     return tokens
 
 # ---------------------