Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -4,145 +4,107 @@ import requests
 from PIL import Image
 from io import BytesIO
 import scratchcommunication
-import traceback
 
-# --- Turbowarpクラウド接続 ---
-PROJECT_ID = "1293416663"
+# --- Scratchクラウド接続 ---
+PROJECT_ID = "1293030706"
 tw = scratchcommunication.TwCloudConnection(
     project_id=PROJECT_ID,
     username="server",
-    contact_info="contact",
-    cloud_host="wss://clouddata.turbowarp.org/",
-    accept_strs=False,  # 数値を扱いやすくする
-    reconnect=True
+    contact_info="contact"
 )
 
-# Quickaccess有効化（辞書のように扱える）
-tw.enable_quickaccess()
+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 の読み込み ---
-try:
-    with open("turbowarp-server/n-chars.txt", "r", encoding="utf-8") as f:
-        n_chars = [line.strip() for line in f]
-except Exception as e:
-    print(f"[FATAL] n-chars.txt 読み込み失敗: {e}")
-    traceback.print_exc()
-    n_chars = []
+with open("turbowarp-server/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):
-        try:
-            idx = int(encoded_str[i:i+2])
-            if idx < len(n_chars):
-                chars.append(n_chars[idx])
-        except Exception as e:
-            print(f"[WARN] decode失敗: {encoded_str[i:i+2]} -> {e}")
+        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):
-    try:
-        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,
-            timeout=20
-        )
-        resp.raise_for_status()
-        return Image.open(BytesIO(resp.content))
-    except Exception as e:
-        print(f"[ERROR] 画像生成失敗: '{prompt}' -> {e}")
-        traceback.print_exc()
-        return None
+    """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
+    return "10" + encoded  # 先頭に "10"
 
 def split_packets(data, max_length=9998):
+    """10000文字制限に合わせて分割（先頭 '10' 含む）"""
     packets = []
     idx = 0
     while idx < len(data):
-        packets.append(data[idx:idx + max_length])
+        chunk = data[idx:idx + max_length]
+        packets.append(chunk)
         idx += max_length
     return packets
 
 # --- メインループ ---
 while True:
     try:
-        # n1 の取得
-        try:
-            n1_value = tw["n1"]  # 文字列の変数名でアクセス
-            if isinstance(n1_value, int):
-                n1_value = str(n1_value)  # 数値の場合は文字列に変換
-        except KeyError:
-            n1_value = None
-        except Exception as e:
-            print(f"[ERROR] n1取得エラー: {e}")
-            n1_value = None
-        
-        # n0 の取得
-        try:
-            n0_value = tw["n0"]
-            if isinstance(n0_value, int):
-                n0_value = int(n0_value)  # もしくはそのまま整数でOK
-        except KeyError:
-            n0_value = None
-        except Exception as e:
-            print(f"[ERROR] n0取得エラー: {e}")
-            n0_value = None
-
-        if n1 and len(n1) >= 3 and n1[0] == "0" and n0 == 0:
-            tw["n0"] = 1  # 処理中フラグ
-
-            user_id = n1[1:3]
-            encoded_prompt = n1[3:]
-            prompt = decode_prompt(encoded_prompt)
-
-            # 画像生成
-            img = generate_image(prompt)
-            if img is None:
-                tw["n0"] = 0
-                time.sleep(1)
-                continue
-
-            scratch_data = resize_and_encode(img)
-            packets = split_packets(scratch_data)
-
-            # パケット送信
-            for pkt in packets:
-                tw["n1"] = pkt
-                time.sleep(0.2)
-                # 次パケット送信待ち
-                start = time.time()
-                while get_var("n1") != "11":
-                    if time.time() - start > 10:  # 10秒でタイムアウト
-                        break
-                    time.sleep(0.1)
-
-            # 完了
-            tw["n0"] = 0
-
+        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(f"[CRITICAL] メインループ例外: {e}")
-        traceback.print_exc()
+        print("Error:", e)
     time.sleep(0.2)