Update turbowarp-server/gpt.py

turbowarp-server/gpt.py

@@ -1,6 +1,7 @@
 import scratchcommunication
 import time
 import requests
+import traceback
 
 PROJECT_ID = "1290918780"
 
@@ -22,190 +23,318 @@ 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())
 
+# 高速化用マップ
+char_map = {c: i for i, c in enumerate(chars)}
+
+# ---------------------
+# encode / decode
+# ---------------------
+
 def encode(text):
+
     out = ""
+
     for c in text:
-        if c in chars:
-            i = chars.index(c)
-            out += f"{i:02d}"
+        try:
+            if c in char_map:
+                out += f"{char_map[c]:02d}"
+            else:
+                print("encode skip unknown char:", c)
+        except Exception:
+            print("ENCODE ERROR")
+            print(traceback.format_exc())
+
     return out
 
+
 def decode(data):
+
     text = ""
-    for i in range(0, len(data), 2):
-        num = int(data[i:i+2])
-        if num == 99:
-            text += "\n"
-        else:
-            text += chars[num]
+
+    try:
+
+        for i in range(0, len(data), 2):
+
+            part = data[i:i+2]
+
+            if len(part) < 2:
+                print("decode broken pair:", part)
+                continue
+
+            num = int(part)
+
+            if num == 99:
+                text += "\n"
+
+            elif num < len(chars):
+                text += chars[num]
+
+            else:
+                print("decode invalid index:", num)
+
+    except Exception:
+        print("DECODE ERROR")
+        print(traceback.format_exc())
+
     return text
 
+
 # ---------------------
-# TwCloudConnection の正しいラッパー
+# TwCloudConnection
 # ---------------------
 
 def get_var(name):
-    """
-    TwCloudConnection の get_variable はキーワード引数で使う必要があるので
-    ここでラップする（戻り値はそのまま）。
-    """
+
     try:
-        # name_literal を必要に応じて True に切り替えてください
-        return tw.get_variable(name=name, name_literal=False)
+        v = tw.get_variable(name=name, name_literal=False)
+        return v
+
     except Exception:
-        # 例外は None を返す（呼び出し側でチェック）
+        print("get_var error:", name)
+        print(traceback.format_exc())
         return None
 
+
 def set_var(name, value):
+
     try:
-        return tw.set_variable(name=name, value=value, name_literal=False)
+        tw.set_variable(name=name, value=value, name_literal=False)
+        return True
+
     except Exception:
-        return None
+        print("set_var error:", name)
+        print(traceback.format_exc())
+        return False
+
 
 # ---------------------
 # n0管理
 # ---------------------
 
 def get_used():
+
     v = get_var("n0")
+
     if not v:
         return []
+
     return list(v)
 
+
 def add_used(i):
+
     u = get_used()
+
     if str(i) not in u:
         u.append(str(i))
+
     set_var("n0", "".join(u))
 
+
 def remove_used(i):
+
     u = get_used()
+
     if str(i) in u:
         u.remove(str(i))
+
     set_var("n0", "".join(u))
 
+
 # ---------------------
 # API
 # ---------------------
 
 def ask_gpt(history):
-    messages = [SYSTEM_PROMPT] + history
-    r = requests.post(
-        "https://izuemon-phi-3.hf.space/v1/chat/completions",
-        json={
-            "model": "gpt-3.5-turbo",
-            "messages": messages
-        },
-        timeout=30
-    )
-    r.raise_for_status()
-    data = r.json()
-    return data["choices"][0]["message"]["content"]
+
+    try:
+
+        messages = [SYSTEM_PROMPT] + history
+
+        print("API request start")
+
+        r = requests.post(
+            "https://izuemon-phi-3.hf.space/v1/chat/completions",
+            json={
+                "model": "gpt-3.5-turbo",
+                "messages": messages
+            },
+            timeout=30
+        )
+
+        print("API status:", r.status_code)
+
+        r.raise_for_status()
+
+        data = r.json()
+
+        reply = data["choices"][0]["message"]["content"]
+
+        print("API reply length:", len(reply))
+
+        return reply
+
+    except requests.exceptions.Timeout:
+        print("API timeout")
+        return "APIタイムアウト"
+
+    except requests.exceptions.RequestException:
+        print("API request error")
+        print(traceback.format_exc())
+        return "API通信エラー"
+
+    except Exception:
+        print("API unknown error")
+        print(traceback.format_exc())
+        return "AIエラー"
+
 
 # ---------------------
 # 送信
 # ---------------------
 
 def send(slot, text):
+
+    print("send to", slot)
+
     encoded = encode(text)
 
-    # TurboWarp 環境なら問題ないとのことなのでそのまま
+    if not encoded:
+        print("encoded text empty")
+        return
+
     size = 99996
 
     packets = [encoded[i:i+size] for i in range(0, len(encoded), size)]
+
     total = len(packets)
 
-    for p in packets:
+    print("packet count:", total)
+
+    for index, p in enumerate(packets):
+
         packet = f"1{total}0{p}"
+
+        print("send packet", index+1, "/", total)
+
         start = time.time()
-        set_var(slot, packet)
+
+        if not set_var(slot, packet):
+            print("packet send failed")
+            return
 
         while True:
+
             v = get_var(slot)
-            # 値が存在して、長さが期待を満たすか、安全に確認
+
             if v and len(v) > 2 and v[2] == "1":
+                print("packet confirmed")
                 break
+
             if time.time() - start > 10:
-                # タイムアウト
+                print("packet timeout")
                 return
+
             time.sleep(0.1)
 
+
 # ---------------------
 # メインループ
 # ---------------------
 
 buffers = {}
 
+print("SERVER STARTED")
+
 while True:
-    for i, slot in enumerate(slots, 1):
-        v = get_var(slot)
-        
-        if not v:
-            continue
-        
-        if len(v) < 3:
-            continue
-        
-        # 追加: 送信パケットは無視
-        if v[0] != "0":
-            continue
-        
-        unread = v[2] == "0"
-        if not unread:
-            continue
-
-        add_used(i)
-
-        # v のフォーマットは "フラグ(1桁) + total(1桁) + '0' + データ..." の前提
-        try:
-            total = int(v[1])
-        except Exception:
-            # 不正データならスキップして使用解除
-            remove_used(i)
-            continue
 
-        data = v[3:]
+    try:
+
+        for i, slot in enumerate(slots, 1):
+
+            v = get_var(slot)
+
+            if not v:
+                continue
+
+            if len(v) < 3:
+                print("invalid packet:", v)
+                continue
+
+            if v[0] != "0":
+                continue
+
+            unread = v[2] == "0"
+
+            if not unread:
+                continue
+
+            print("new message slot", slot)
+
+            add_used(i)
+
+            try:
+                total = int(v[1])
+            except Exception:
+                print("invalid total packet count")
+                remove_used(i)
+                continue
+
+            data = v[3:]
 
-        # 既読フラグを立てる（インデックス 2 を "1" にする）
-        # 文字列操作で安全にセット
-        if len(v) >= 3:
             newv = v[:2] + "1" + v[3:]
             set_var(slot, newv)
 
-        if slot not in buffers:
-            buffers[slot] = []
+            if slot not in buffers:
+                buffers[slot] = []
 
-        buffers[slot].append(data)
+            buffers[slot].append(data)
 
-        if len(buffers[slot]) < total:
-            continue
+            if len(buffers[slot]) < total:
+                continue
 
-        joined = "".join(buffers[slot])
-        decoded = decode(joined)
+            joined = "".join(buffers[slot])
 
-        history = []
-        parts = decoded.split("\n")
+            print("joined length:", len(joined))
 
-        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]
-            })
+            decoded = decode(joined)
 
-        try:
-            reply = ask_gpt(history)
-        except Exception:
-            reply = "エラーが発生しました。"
+            print("decoded message:", decoded)
+
+            history = []
+
+            parts = decoded.split("\n")
+
+            for j in range(0, len(parts), 2):
+
+                if parts[j].strip() == "":
+                    continue
+
+                history.append({
+                    "role": "user",
+                    "content": parts[j]
+                })
+
+            try:
+                reply = ask_gpt(history)
 
-        send(slot, reply)
-        buffers[slot] = []
-        remove_used(i)
+            except Exception:
+                print("AI call failed")
+                reply = "AIエラー"
+
+            send(slot, reply)
+
+            buffers[slot] = []
+
+            remove_used(i)
+
+    except Exception:
+        print("MAIN LOOP ERROR")
+        print(traceback.format_exc())
 
     time.sleep(0.2)