Update turbowarp-server/gpt.py

turbowarp-server/gpt.py

@@ -20,44 +20,59 @@ slots = [f"n{i}" for i in range(1, 10)]
 # ---------------------
 # 文字テーブル
 # ---------------------
-
 chars = []
 with open("turbowarp-server/n-chars.txt", encoding="utf8") as f:
     for line in f:
         chars.append(line.strip())
 
-def encode(text):
+# 半角・全角スペースを chars に追加
+if " " not in chars: chars.append(" ")
+if "　" not in chars: chars.append("　")
+
+def encode(text, for_history=False):
+    """
+    text を数字列に変換
+    for_history=True の場合は履歴区切り文字 '99' を使う
+    改行は送受信用 98
+    """
     out = ""
     for c in text:
         if c in chars:
             i = chars.index(c)
             out += f"{i:02d}"
+        elif c == "\n":
+            out += "98"  # メッセージ内改行
+        elif for_history and c == "\x1e":  # 履歴区切り用の特別文字
+            out += "99"
+        else:
+            # 未登録文字は無視
+            pass
     return out
 
 def decode(data):
+    """
+    数字列を文字列に戻す
+    98 → メッセージ内改行
+    99 → 履歴区切りマーカー
+    """
     text = ""
     for i in range(0, len(data), 2):
         num = int(data[i:i+2])
-        if num == 99:
+        if num == 98:
             text += "\n"
+        elif num == 99:
+            text += "__HIST__"  # 履歴区切りとして特殊文字
         else:
             text += chars[num]
     return text
 
 # ---------------------
-# TwCloudConnection の正しいラッパー
+# TwCloudConnection のラッパー
 # ---------------------
-
 def get_var(name):
-    """
-    TwCloudConnection の get_variable はキーワード引数で使う必要があるので
-    ここでラップする（戻り値はそのまま）。
-    """
     try:
-        # name_literal を必要に応じて True に切り替えてください
         return tw.get_variable(name=name, name_literal=False)
     except Exception:
-        # 例外は None を返す（呼び出し側でチェック）
         return None
 
 def set_var(name, value):
@@ -69,7 +84,6 @@ def set_var(name, value):
 # ---------------------
 # n0管理
 # ---------------------
-
 def get_used():
     v = get_var("n0")
     if not v:
@@ -89,9 +103,8 @@ def remove_used(i):
     set_var("n0", "".join(u))
 
 # ---------------------
-# API
+# ChatGPT API
 # ---------------------
-
 def ask_gpt(history):
     messages = [SYSTEM_PROMPT] + history
     r = requests.post(
@@ -109,13 +122,9 @@ def ask_gpt(history):
 # ---------------------
 # 送信
 # ---------------------
-
 def send(slot, text):
     encoded = encode(text)
-
-    # TurboWarp 環境なら問題ないとのことなのでそのまま
     size = 99996
-
     packets = [encoded[i:i+size] for i in range(0, len(encoded), size)]
     total = len(packets)
 
@@ -126,28 +135,21 @@ def send(slot, text):
 
         while True:
             v = get_var(slot)
-            # 値が存在して、長さが期待を満たすか、安全に確認
             if v and len(v) > 2 and v[2] == "1":
                 break
             if time.time() - start > 10:
-                # タイムアウト
                 return
             time.sleep(0.1)
 
 # ---------------------
 # メインループ
 # ---------------------
-
 buffers = {}
 
 while True:
     for i, slot in enumerate(slots, 1):
         v = get_var(slot)
-
-        if not v:
-            continue
-
-        if len(v) < 3:
+        if not v or len(v) < 3:
             continue
 
         unread = v[2] == "0"
@@ -156,25 +158,18 @@ while True:
 
         add_used(i)
 
-        # v のフォーマットは "フラグ(1桁) + total(1桁) + '0' + データ..." の前提
         try:
             total = int(v[1])
         except Exception:
-            # 不正データならスキップして使用解除
             remove_used(i)
             continue
 
         data = v[3:]
-
-        # 既読フラグを立てる（インデックス 2 を "1" にする）
-        # 文字列操作で安全にセット
-        if len(v) >= 3:
-            newv = v[:2] + "1" + v[3:]
-            set_var(slot, newv)
+        newv = v[:2] + "1" + v[3:]  # 既読フラグ
+        set_var(slot, newv)
 
         if slot not in buffers:
             buffers[slot] = []
-
         buffers[slot].append(data)
 
         if len(buffers[slot]) < total:
@@ -183,20 +178,18 @@ while True:
         joined = "".join(buffers[slot])
         decoded = decode(joined)
 
-        history = []
-        parts = decoded.split("\n")
+        # 履歴単位で分割
+        histories = decoded.split("__HIST__")
 
-        for j in range(0, len(parts), 2):
-            # parts[j] がユーザーメッセージ、parts[j+1] が（将来の拡張用など）
-            if parts[j].strip() == "":
-                continue
-            history.append({
-                "role": "user",
-                "content": parts[j]
-            })
+        history_for_api = []
+        for h in histories:
+            lines = h.split("\n")  # 98 が改行としてデコ��ド済み
+            for line in lines:
+                if line.strip():
+                    history_for_api.append({"role": "user", "content": line})
 
         try:
-            reply = ask_gpt(history)
+            reply = ask_gpt(history_for_api)
         except Exception:
             reply = "エラーが発生しました。"