Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -1,290 +1,290 @@
-import scratchcommunication
-import requests
-from PIL import Image
-from io import BytesIO
+import os
 import time
 import random
-import math
+import threading
+from io import BytesIO
 
-# プロジェクトID
-PROJECT_ID = 1293416663
+import requests
+from PIL import Image
 
-# 接続設定
+# 既存コードを流用
 tw = scratchcommunication.TwCloudConnection(
     project_id=PROJECT_ID,
     username="server",
     contact_info="contact"
 )
 
-# n-chars.txtの読み込み
-def load_char_map():
-    """n-chars.txtから文字マッピングを読み込む"""
-    char_map = {}
-    try:
-        with open("turbowarp-server/n-chars.txt", "r", encoding="utf-8") as f:
-            chars = f.read().strip().split("\n")
-            for i, char in enumerate(chars):
-                char_map[char] = str(i).zfill(2)  # 00, 01, 02...
-                char_map[str(i).zfill(2)] = char  # 逆引きも保存
-        return char_map
-    except Exception as e:
-        print(f"文字マップ読み込みエラー: {e}")
-        return {}
-
-CHAR_MAP = load_char_map()
-
-def encode_prompt(prompt):
-    """プロンプトを数字列にエンコード"""
-    encoded = []
-    for char in prompt:
-        if char in CHAR_MAP:
-            encoded.append(CHAR_MAP[char])
-        else:
-            encoded.append("99")  # 不明な文字は99
-    return "".join(encoded)
-
-def decode_prompt(encoded):
-    """数字列をプロンプトにデコード"""
-    decoded = []
-    for i in range(0, len(encoded), 2):
-        code = encoded[i:i+2]
-        if code in CHAR_MAP:
-            decoded.append(CHAR_MAP[code])
-        else:
-            decoded.append("?")
-    return "".join(decoded)
-
 def get_var(name):
-    """クラウド変数の値を取得（安全版）"""
     try:
-        return tw.get_variable(name=name, name_literal=False)
+        v = tw.get_variable(name=name, name_literal=False)
+        return None if v is None else str(v)
     except Exception:
         return None
 
 def set_var(name, value):
-    """クラウド変数の値を設定（安全版）"""
     try:
-        return tw.set_variable(name=name, value=value, name_literal=False)
+        return tw.set_variable(name=name, value=str(value), name_literal=False)
     except Exception:
         return None
 
-def wait_for_var_change(var_name, expected_prefix=None, timeout=30):
-    """変数の変更を待機"""
-    start_time = time.time()
-    last_value = get_var(var_name)
-    
-    while time.time() - start_time < timeout:
-        current_value = get_var(var_name)
-        if current_value != last_value:
-            if expected_prefix is None or str(current_value).startswith(expected_prefix):
-                return current_value
-            last_value = current_value
-        time.sleep(0.1)
-    return None
-
-def generate_image(prompt, negative_prompt="nsfw, low quality"):
-    """画像生成APIを呼び出し"""
-    try:
-        # ランダムシードを生成
-        seed = random.randint(0, 2**32 - 1)
-        
-        params = {
-            "prompt": prompt,
-            "negative_prompt": negative_prompt,
-            "seed": seed,
-            "randomize_seed": "false",
-            "width": 352,
-            "height": 352,
-            "guidance_scale": 0.0,  # CFG=0
-            "num_inference_steps": 2
-        }
-        
-        response = requests.get(
-            "https://izuemon-pixart-alpha-pixart-sigma-xl-2-1024-ms.hf.space/gen",
-            params=params,
-            timeout=60
-        )
-        
-        if response.status_code == 200:
-            return response.content
-        else:
-            print(f"画像生成APIエラー: {response.status_code}")
-            return None
-            
-    except Exception as e:
-        print(f"画像生成エラー: {e}")
-        return None
 
-def resize_image(image_data, size=(90, 90)):
-    """画像をリサイズ"""
-    try:
-        img = Image.open(BytesIO(image_data))
-        img = img.resize(size, Image.Resampling.LANCZOS)
-        return img
-    except Exception as e:
-        print(f"リサイズエラー: {e}")
-        return None
+# ========= 設定 =========
+API_URL = "https://izuemon-pixart-alpha-pixart-sigma-xl-2-1024-ms.hf.space/gen"
+N_CHARS_PATH = "turbowarp-server/n-chars.txt"
 
-def image_to_rgb_string(img):
-    """画像をRGB値の文字列に変換（000000-999999形式）"""
-    pixels = []
-    width, height = img.size
-    
-    for y in range(height):
-        for x in range(width):
-            r, g, b = img.getpixel((x, y))[:3]
-            # RGBを6桁の数字に変換（000000-999999）
-            rgb_value = r * 10000 + g * 100 + b
-            pixels.append(str(rgb_value).zfill(6))
-    
-    return "".join(pixels)
-
-def send_packets(data, var_name="n1"):
-    """データをパケットに分割して送信"""
-    # パケットサイズ（n1のプレフィックス"10"を除く）
-    packet_size = 9998 - 2  # "10"分を引く
-    
-    # パケット数
-    total_packets = math.ceil(len(data) / packet_size)
-    
-    for i in range(total_packets):
-        start = i * packet_size
-        end = min((i + 1) * packet_size, len(data))
-        packet_data = data[start:end]
-        
-        # パケット番号を付与（2桁）
-        packet_num = str(i + 1).zfill(2)
-        packet_value = f"10{packet_num}{packet_data}"
-        
-        # 送信
-        set_var(var_name, packet_value)
-        print(f"パケット {i+1}/{total_packets} 送信: {len(packet_data)}文字")
-        
-        # 次のパケットの前に既読を待機
-        if i < total_packets - 1:
-            # "11"に変更されるのを待機
-            start_wait = time.time()
-            while True:
-                current = get_var(var_name)
-                if current and str(current).startswith("11"):
-                    print(f"パケット {i+1} 既読確認")
-                    break
-                if time.time() - start_wait > 30:
-                    print("既読待機タイムアウト")
-                    break
-                time.sleep(0.1)
-            
-            # 0.5秒間隔
-            time.sleep(0.5)
-    
-    return total_packets
-
-def process_request(received_value):
-    """受信したデータを処理"""
-    try:
-        # 受信データ: "0[ID][encoded_prompt]"
-        # 先頭が'0'か確認
-        value_str = str(received_value)
-        if not value_str.startswith('0'):
-            print(f"無効なデータ形式: {value_str[:10]}...")
-            return
-        
-        # ID抽出（2桁）
-        if len(value_str) < 3:
-            print("IDが不足しています")
-            return
-        
-        user_id = value_str[1:3]
-        encoded_prompt = value_str[3:]
-        
-        print(f"処理開始 - ID: {user_id}, エンコード長: {len(encoded_prompt)}")
-        
-        # プロンプトをデコード
-        prompt = decode_prompt(encoded_prompt)
-        print(f"デコードされたプロンプト: {prompt}")
-        
-        # 画像生成
-        print("画像生成中...")
-        image_data = generate_image(prompt)
-        if not image_data:
-            print("画像生成失敗")
-            return
-        
-        # リサイズ
-        print("画像リサイズ中...")
-        resized_img = resize_image(image_data)
-        if not resized_img:
-            print("リサイズ失敗")
+NEGATIVE_PROMPT = "nsfw, low quality"
+WIDTH = 352
+HEIGHT = 352
+CFG = 0
+SEED_RANDOM = True
+SAMPLING_STEPS = 2
+
+OUT_SIZE = (90, 90)
+PACKET_DATA_LIMIT = 9998   # 「10」を除いたデータ部の上限として扱う
+POLL_INTERVAL = 0.5        # クラウド変数の連続送信を防ぐ
+ACK_VALUE = "11"           # Scratch側が既読にした合図
+BUSY_LOCK = "1"
+FREE_LOCK = "0"
+
+
+# ========= 文字対応表 =========
+def load_n_chars(path: str):
+    if not os.path.exists(path):
+        raise FileNotFoundError(f"n-chars.txt が見つかりません: {path}")
+
+    with open(path, "r", encoding="utf-8") as f:
+        chars = [line.rstrip("\n") for line in f]
+
+    if len(chars) < 100:
+        raise ValueError("n-chars.txt は最低 100 行必要です（00〜99）。")
+    return chars[:100]
+
+N_CHARS = load_n_chars(N_CHARS_PATH)
+N_CHARS_MAP = {f"{i:02d}": ch for i, ch in enumerate(N_CHARS)}
+N_CHARS_REV = {ch: f"{i:02d}" for i, ch in enumerate(N_CHARS)}
+
+
+def decode_numeric_text(encoded: str) -> str:
+    """
+    00〜99 の2桁列を文字列に戻す。
+    例: '000102' -> ['00','01','02']
+    """
+    encoded = "" if encoded is None else str(encoded)
+    encoded = "".join(ch for ch in encoded if ch.isdigit())
+
+    if len(encoded) % 2 != 0:
+        encoded = encoded[:-1]  # 末尾が余ったら切り捨て
+
+    out = []
+    for i in range(0, len(encoded), 2):
+        key = encoded[i:i+2]
+        out.append(N_CHARS_MAP.get(key, ""))
+    return "".join(out)
+
+
+def encode_text_to_numeric(text: str) -> str:
+    """
+    文字列 -> 00〜99 の列
+    """
+    result = []
+    for ch in str(text):
+        if ch not in N_CHARS_REV:
+            raise ValueError(f"対応表にない文字です: {repr(ch)}")
+        result.append(N_CHARS_REV[ch])
+    return "".join(result)
+
+
+# ========= Cloud 値の正規化 =========
+def normalize_value(v):
+    if v is None:
+        return ""
+    return str(v).strip()
+
+
+def get_n1():
+    return normalize_value(get_var("n1"))
+
+def get_n0():
+    return normalize_value(get_var("n0"))
+
+def set_n1(v):
+    return set_var("n1", v)
+
+def set_n0(v):
+    return set_var("n0", v)
+
+
+# ========= 画像生成 =========
+def fetch_generated_image(prompt: str) -> Image.Image:
+    params = {
+        "prompt": prompt,
+        "negative_prompt": NEGATIVE_PROMPT,
+        "seed": random.randint(0, 2**31 - 1) if SEED_RANDOM else 0,
+        "randomize_seed": "true" if SEED_RANDOM else "false",
+        "width": WIDTH,
+        "height": HEIGHT,
+        "guidance_scale": CFG,
+        "num_inference_steps": SAMPLING_STEPS,
+    }
+
+    r = requests.get(API_URL, params=params, timeout=300)
+    r.raise_for_status()
+
+    # 生画像データを想定
+    img = Image.open(BytesIO(r.content)).convert("RGB")
+    return img
+
+
+def resize_to_90(img: Image.Image) -> Image.Image:
+    return img.resize(OUT_SIZE, Image.Resampling.LANCZOS)
+
+
+def rgb_to_6digits(r: int, g: int, b: int) -> str:
+    """
+    RGB を 6桁の10進数に変換。
+    仕様が曖昧なため、各チャンネルを 0〜99 に線形変換して 2桁ずつ連結する。
+    """
+    rr = round(r * 99 / 255)
+    gg = round(g * 99 / 255)
+    bb = round(b * 99 / 255)
+    return f"{rr:02d}{gg:02d}{bb:02d}"
+
+
+def image_to_payload(img: Image.Image) -> str:
+    """
+    90x90 の画像を左上から走査し、
+    '10' + 90*90個の6桁数値 を返す。
+    """
+    pixels = img.load()
+    w, h = img.size
+    parts = ["10"]
+
+    for y in range(h):
+        for x in range(w):
+            r, g, b = pixels[x, y]
+            parts.append(rgb_to_6digits(r, g, b))
+
+    return "".join(parts)
+
+
+# ========= パケット送信 =========
+def split_packets(data: str, limit: int = PACKET_DATA_LIMIT):
+    return [data[i:i+limit] for i in range(0, len(data), limit)]
+
+
+def wait_for_ack():
+    while True:
+        time.sleep(POLL_INTERVAL)
+        v = get_n1()
+        if v == ACK_VALUE:
             return
-        
-        # RGB文字列に変換
-        print("RGB変換中...")
-        rgb_string = image_to_rgb_string(resized_img)
-        print(f"RGB文字列長: {len(rgb_string)}文字")
-        
-        # パケット送信
-        print("パケット送信開始...")
-        packet_count = send_packets(rgb_string)
-        print(f"送信完了: {packet_count}パケット")
-        
-        return True
-        
-    except Exception as e:
-        print(f"処理エラー: {e}")
-        import traceback
-        traceback.print_exc()
-        return False
-
-def main():
-    """メインループ"""
-    print("サーバー起動...")
-    print("プロジェクトID:", PROJECT_ID)
-    
-    # 初期化
-    set_var("n0", "0")
-    set_var("n1", "00")
-    
-    print("待機中...")
-    
+        # 事故防止: 途中で n1 が空や想定外に戻った場合は少し待つ
+        # 必要ならここでエラー処理にしてもよい
+
+
+def send_packets(data_with_prefix: str):
+    packets = split_packets(data_with_prefix, PACKET_DATA_LIMIT)
+
+    for idx, packet in enumerate(packets, start=1):
+        # 送信開始
+        set_n1("10" + packet)
+
+        # Scratch側が既読にして "11" にするまで待つ
+        wait_for_ack()
+
+        # 次の通信へ行く前に少し空ける
+        time.sleep(POLL_INTERVAL)
+
+    # 全送信完了
+    set_n0(FREE_LOCK)
+
+
+# ========= メイン処理 =========
+def process_incoming_value(raw_value: str):
+    """
+    n1 の 3文字目以降をデコードして prompt として使う。
+    もしここで「ID -> prompt」変換を入れたい場合は、
+    decode_numeric_text の結果を ID として別処理に差し替えてください。
+    """
+    raw_value = normalize_value(raw_value)
+
+    if len(raw_value) < 3:
+        return
+
+    kind = raw_value[0]   # 0: ID送信系 / 1: レスポンス系
+    read_flag = raw_value[1]
+    payload = raw_value[2:]
+
+    # 今回は「送られてきたものをデコードしたもの」を prompt として使う
+    decoded_prompt = decode_numeric_text(payload)
+
+    # 必要なら kind / read_flag を使って分岐できる
+    # kind == "0" -> 依頼受信
+    # kind == "1" -> 返信など
+    _ = read_flag
+
+    # 画像生成
+    img = fetch_generated_image(decoded_prompt)
+
+    # 90x90 に縮小
+    img = resize_to_90(img)
+
+    # 送信用の数値列へ変換
+    payload_90 = image_to_payload(img)
+
+    # 送信前にロックをかける
+    set_n0(BUSY_LOCK)
+
+    # パケット分割送信
+    send_packets(payload_90)
+
+
+def server_loop():
+    last_n1 = None
+
     while True:
         try:
-            # n1の変更を監視
-            n1_value = wait_for_var_change("n1", expected_prefix="0", timeout=5)
-            
-            if n1_value is not None:
-                # 使用中フラグを立てる
-                set_var("n0", "1")
-                print(f"受信データ検出: {str(n1_value)[:50]}...")
-                
-                # 既読にする（n1を"1x"に変更）
-                # 元の値から既読フラグを立てる
-                current_n1 = str(n1_value)
-                if len(current_n1) >= 2:
-                    read_value = "1" + current_n1[1:]
-                    set_var("n1", read_value)
-                    print(f"既読設定: {read_value}")
-                
-                # リクエスト処理
-                success = process_request(n1_value)
-                
-                if success:
-                    print("処理完了")
+            current_n1 = get_n1()
+
+            # 変化を検知したときだけ処理
+            if current_n1 and current_n1 != last_n1:
+                last_n1 = current_n1
+
+                # 他ユーザー使用中なら何もしない
+                if get_n0() == BUSY_LOCK:
+                    time.sleep(POLL_INTERVAL)
+                    continue
+
+                # n1 の先頭が 0 のときだけ受信処理
+                if current_n1.startswith("0"):
+                    # まず使用中にする
+                    set_n0(BUSY_LOCK)
+
+                    # 既読化の扱いが必要ならここで Scratch 側仕様に合わせて書き換え
+                    # 例: 受信確認として n1 を一度 "11" にする場合はここで行う
+                    # set_n1(ACK_VALUE)
+
+                    process_incoming_value(current_n1)
+
+                    # process_incoming_value 側で最後に n0=0 に戻す
                 else:
-                    print("処理失敗")
-                
-                # 使用中フラグを解除
-                set_var("n0", "0")
-                print("待機再開...")
-            
-            time.sleep(0.1)
-            
-        except KeyboardInterrupt:
-            print("\nサーバー停止")
-            set_var("n0", "0")
-            break
-        except Exception as e:
-            print(f"メインループエラー: {e}")
-            time.sleep(1)
+                    time.sleep(POLL_INTERVAL)
+
+            time.sleep(POLL_INTERVAL)
+
+        except Exception:
+            # 例外時はロック解除して継続
+            try:
+                set_n0(FREE_LOCK)
+            except Exception:
+                pass
+            time.sleep(POLL_INTERVAL)
+
 
 if __name__ == "__main__":
-    main()
+    # 初期化
+    set_n0(FREE_LOCK)
+    # n1 は必要ならここで初期値を入れる
+    set_n1("0")
+
+    server_loop()