Update turbowarp-server/gpt.py

turbowarp-server/gpt.py

@@ -20,59 +20,44 @@ 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())
 
-# 半角・全角スペースを 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
-    """
+def encode(text):
     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 == 98:
+        if num == 99:
             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):
@@ -84,6 +69,7 @@ def set_var(name, value):
 # ---------------------
 # n0管理
 # ---------------------
+
 def get_used():
     v = get_var("n0")
     if not v:
@@ -103,8 +89,9 @@ def remove_used(i):
     set_var("n0", "".join(u))
 
 # ---------------------
-# ChatGPT API
+# API
 # ---------------------
+
 def ask_gpt(history):
     messages = [SYSTEM_PROMPT] + history
     r = requests.post(
@@ -122,9 +109,13 @@ 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)
 
@@ -135,21 +126,28 @@ 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 or len(v) < 3:
+
+        if not v:
+            continue
+
+        if len(v) < 3:
             continue
 
         unread = v[2] == "0"
@@ -158,18 +156,25 @@ while True:
 
         add_used(i)
 
+        # v のフォーマットは "フラグ(1桁) + total(1桁) + '0' + データ..." の前提
         try:
             total = int(v[1])
         except Exception:
+            # 不正データならスキップして使用解除
             remove_used(i)
             continue
 
         data = v[3:]
-        newv = v[:2] + "1" + v[3:]  # 既読フラグ
-        set_var(slot, newv)
+
+        # 既読フラグを立てる（インデックス 2 を "1" にする）
+        # 文字列操作で安全にセット
+        if len(v) >= 3:
+            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:
@@ -178,18 +183,20 @@ while True:
         joined = "".join(buffers[slot])
         decoded = decode(joined)
 
-        # 履歴単位で分割
-        histories = decoded.split("__HIST__")
+        history = []
+        parts = decoded.split("\n")
 
-        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})
+        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]
+            })
 
         try:
-            reply = ask_gpt(history_for_api)
+            reply = ask_gpt(history)
         except Exception:
             reply = "エラーが発生しました。"