Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -1,92 +1,110 @@
 import time
+import random
 import requests
+from PIL import Image
+from io import BytesIO
 import scratchcommunication
 
+# --- Scratchクラウド接続 ---
 PROJECT_ID = 1293030706
-
-# Scratchクラウド接続
 tw = scratchcommunication.TwCloudConnection(
     project_id=PROJECT_ID,
     username="server",
-    contact_info="contact",
-    reconnect=True,
-    receive_from_websocket=True,
-    accept_strs=True  # 文字列も取得可能
+    contact_info="contact"
 )
 
-# short_id を a-z0-9 → 00~35 に変換
-def encode_short_id(short_id):
-    chars = "abcdefghijklmnopqrstuvwxyz0123456789"
-    result = ""
-    for c in short_id.lower():  # 念のため小文字化
-        if c in chars:
-            result += f"{chars.index(c):02d}"
-        else:
-            result += "00"  # 不明文字は00
-    return result
-
-last_n1 = None
+def get_var(name):
+    try:
+        return tw.get_variable(name=name, name_literal=False)
+    except Exception:
+        return None
 
+def set_var(name, value):
+    try:
+        return tw.set_variable(name=name, value=value, name_literal=False)
+    except Exception:
+        return None
+
+# --- n-chars.txt の読み込み ---
+with open("n-chars.txt", "r", encoding="utf-8") as f:
+    n_chars = [line.strip() for line in f]
+
+def decode_prompt(encoded_str):
+    """クラウド変数の数字列を元にプロンプト文字列に復号"""
+    chars = []
+    for i in range(0, len(encoded_str), 2):
+        idx = int(encoded_str[i:i+2])
+        if idx < len(n_chars):
+            chars.append(n_chars[idx])
+    return "".join(chars)
+
+def rgb_to_scratch_number(rgb):
+    """RGBを0-999999の10進数に変換"""
+    r, g, b = rgb
+    return f"{r:03}{g:03}{b:03}"
+
+def generate_image(prompt):
+    """APIから352x352の画像生成"""
+    seed = random.randint(0, 999999)
+    params = {
+        "prompt": prompt,
+        "negative_prompt": "nsfw, low quality",
+        "width": 352,
+        "height": 352,
+        "num_inference_steps": 2,
+        "guidance_scale": 0,
+        "seed": seed,
+        "randomize_seed": "true"
+    }
+    resp = requests.get("https://izuemon-pixart-alpha-pixart-sigma-xl-2-1024-ms.hf.space/gen", params=params)
+    resp.raise_for_status()
+    return Image.open(BytesIO(resp.content))
+
+def resize_and_encode(img):
+    """90x90にリサイズして数字列に変換"""
+    img = img.resize((90, 90))
+    encoded = ""
+    for y in range(90):
+        for x in range(90):
+            encoded += rgb_to_scratch_number(img.getpixel((x, y)))
+    return "10" + encoded  # 先頭に "10"
+
+def split_packets(data, max_length=9998):
+    """10000文字制限に合わせて分割（先頭 '10' 含む）"""
+    packets = []
+    idx = 0
+    while idx < len(data):
+        chunk = data[idx:idx + max_length]
+        packets.append(chunk)
+        idx += max_length
+    return packets
+
+# --- メインループ ---
 while True:
     try:
-        # クラウド変数 n1 の取得
-        n1 = tw.get_variable(name="n1", name_literal=False)
-        if n1 is None:
-            time.sleep(0.2)
-            continue
-        n1 = str(n1).strip()
-
-        if n1 != last_n1:
-            last_n1 = n1
-
-            if len(n1) < 3:
-                continue  # 3桁以上ない場合はスキップ
-
-            mode = n1[0]        # 0: 受信/送信待ち, 1: レスポンス返却
-            read_flag = n1[1]   # 0: 未読, 1: 既読
-            content = n1[2:]    # 3桁目以降 = ID
-
-            # -----------------------------
-            # Scratch → サーバー
-            # -----------------------------
-            if mode == "0" and read_flag == "0":
-                # 使用中フラグ
-                tw.set_variable(name="n0", value=1)
-                print("受信ID:", content)
-
-                # 既読に変更
-                tw.set_variable(name="n1", value="0" + "1" + content)
-
-                # APIアクセス
-                url = f"https://20.rf.gd/set.php?project_id={content}"
-                try:
-                    res = requests.get(url, cookies={"__test": "872039d283902860e895efb6018d2632"}, timeout=5)
-                    res.raise_for_status()
-                    data = res.json()
-                    short_id = data.get("short_id", "000")
-                except requests.exceptions.RequestException as e:
-                    print("API接続エラー:", e)
-                    short_id = "000"
-                except ValueError:
-                    print("API JSON解析エラー")
-                    short_id = "000"
-
-                print("short_id:", short_id)
-
-                encoded = encode_short_id(short_id)
-
-                # サーバー → Scratch にレスポンス書き込み
-                tw.set_variable(name="n1", value="1" + "0" + encoded)
-
-            # -----------------------------
-            # Scratch が既読にしたら n0 を解放
-            # -----------------------------
-            elif mode == "1" and read_flag == "1":
-                tw.set_variable(name="n0", value=0)
-                print("通信完了。n0を0に戻しました。")
-
-        time.sleep(0.2)
-
+        n1 = get_var("n1")
+        if n1 and len(n1) >= 3 and n1[0] == "0":  # ID受信フラグ
+            if get_var("n0") == "0":  # 他ユーザー未使用
+                set_var("n0", "1")
+                user_id = n1[1:3]  # 実際のIDはここを調整
+                encoded_prompt = n1[3:]
+                prompt = decode_prompt(encoded_prompt)
+                
+                # 画像生成
+                img = generate_image(prompt)
+                scratch_data = resize_and_encode(img)
+                
+                # パケット分割
+                packets = split_packets(scratch_data)
+                for pkt in packets:
+                    set_var("n1", pkt)  # 送信
+                    time.sleep(0.2)
+                    # 次パケット送信待ち
+                    while get_var("n1") != "11":
+                        time.sleep(0.1)
+                
+                # 完了後リセット
+                set_var("n0", "0")
     except Exception as e:
-        print("エラー:", e)
-        time.sleep(1)
+        print("Error:", e)
+    time.sleep(0.2)