Update turbowarp-server/gpt.py

turbowarp-server/gpt.py

@@ -1,7 +1,6 @@
 import scratchcommunication
 import time
 import requests
-import traceback
 
 PROJECT_ID = "1290918780"
 
@@ -23,176 +22,100 @@ 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:
-        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())
-
+        if c in chars:
+            i = chars.index(c)
+            out += f"{i:02d}"
+        elif c == "\n":
+            out += "98"
     return out
 
-
 def decode(data):
+    tokens = []
+    current = ""
 
-    text = ""
+    for i in range(0, len(data), 2):
+        num = int(data[i:i+2])
 
-    try:
-
-        for i in range(0, len(data), 2):
-
-            part = data[i:i+2]
-
-            if len(part) < 2:
-                print("decode broken pair:", part)
-                continue
+        if num == 99:  # 履歴区切り
+            tokens.append(current)
+            current = ""
 
-            num = int(part)
+        elif num == 98:  # 改行
+            current += "\n"
 
-            if num == 99:
-                text += "\n"
+        else:
+            if num < len(chars):
+                current += chars[num]
 
-            elif num < len(chars):
-                text += chars[num]
-
-            else:
-                print("decode invalid index:", num)
-
-    except Exception:
-        print("DECODE ERROR")
-        print(traceback.format_exc())
-
-    return text
+    tokens.append(current)
 
+    return tokens
 
 # ---------------------
-# TwCloudConnection
+# TwCloudConnection ラッパー
 # ---------------------
 
 def get_var(name):
-
     try:
-        v = tw.get_variable(name=name, name_literal=False)
-        return v
-
+        return tw.get_variable(name=name, name_literal=False)
     except Exception:
-        print("get_var error:", name)
-        print(traceback.format_exc())
         return None
 
-
 def set_var(name, value):
-
     try:
-        tw.set_variable(name=name, value=value, name_literal=False)
-        return True
-
+        return tw.set_variable(name=name, value=value, name_literal=False)
     except Exception:
-        print("set_var error:", name)
-        print(traceback.format_exc())
-        return False
-
+        return None
 
 # ---------------------
 # 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
 
-    try:
-
-        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
+    )
 
-        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エラー"
+    r.raise_for_status()
+    data = r.json()
 
+    return data["choices"][0]["message"]["content"]
 
 # ---------------------
 # 送信
@@ -200,141 +123,109 @@ def ask_gpt(history):
 
 def send(slot, text):
 
-    print("send to", slot)
-
     encoded = encode(text)
 
-    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)
 
-    print("packet count:", total)
-
-    for index, p in enumerate(packets):
+    for p in packets:
 
         packet = f"1{total}0{p}"
 
-        print("send packet", index+1, "/", total)
-
         start = time.time()
 
-        if not set_var(slot, packet):
-            print("packet send failed")
-            return
+        set_var(slot, packet)
 
         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:
 
-    try:
+    for i, slot in enumerate(slots, 1):
 
-        for i, slot in enumerate(slots, 1):
+        v = get_var(slot)
 
-            v = get_var(slot)
+        if not v:
+            continue
 
-            if not v:
-                continue
+        if len(v) < 3:
+            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)
+        if v[0] != "0":
+            continue
 
-            add_used(i)
+        unread = v[2] == "0"
 
-            try:
-                total = int(v[1])
-            except Exception:
-                print("invalid total packet count")
-                remove_used(i)
-                continue
+        if not unread:
+            continue
 
-            data = v[3:]
+        add_used(i)
 
-            newv = v[:2] + "1" + v[3:]
-            set_var(slot, newv)
-
-            if slot not in buffers:
-                buffers[slot] = []
+        try:
+            total = int(v[1])
+        except Exception:
+            remove_used(i)
+            continue
 
-            buffers[slot].append(data)
+        data = v[3:]
 
-            if len(buffers[slot]) < total:
-                continue
+        newv = v[:2] + "1" + v[3:]
+        set_var(slot, newv)
 
-            joined = "".join(buffers[slot])
+        if slot not in buffers:
+            buffers[slot] = []
 
-            print("joined length:", len(joined))
+        buffers[slot].append(data)
 
-            decoded = decode(joined)
+        if len(buffers[slot]) < total:
+            continue
 
-            print("decoded message:", decoded)
+        joined = "".join(buffers[slot])
 
-            history = []
+        messages = decode(joined)
 
-            parts = decoded.split("\n")
+        history = []
 
-            for j in range(0, len(parts), 2):
+        for j, msg in enumerate(messages):
 
-                if parts[j].strip() == "":
-                    continue
+            msg = msg.strip()
 
-                history.append({
-                    "role": "user",
-                    "content": parts[j]
-                })
+            if msg == "":
+                continue
 
-            try:
-                reply = ask_gpt(history)
+            role = "user" if j % 2 == 0 else "assistant"
 
-            except Exception:
-                print("AI call failed")
-                reply = "AIエラー"
+            history.append({
+                "role": role,
+                "content": msg
+            })
 
-            send(slot, reply)
+        try:
+            reply = ask_gpt(history)
+        except Exception:
+            reply = "エラーが発生しました。"
 
-            buffers[slot] = []
+        send(slot, reply)
 
-            remove_used(i)
+        buffers[slot] = []
 
-    except Exception:
-        print("MAIN LOOP ERROR")
-        print(traceback.format_exc())
+        remove_used(i)
 
     time.sleep(0.2)