Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -7,27 +7,37 @@ import scratchcommunication
 
 # --- Scratchクラウド接続 ---
 PROJECT_ID = "1293030706"
+print(f"[INFO] Connecting to Scratch project {PROJECT_ID}...")
 tw = scratchcommunication.TwCloudConnection(
     project_id=PROJECT_ID,
     username="server",
     contact_info="contact"
 )
+print("[INFO] Connected successfully.")
 
 def get_var(name):
     try:
-        return tw.get_variable(name=name, name_literal=False)
-    except Exception:
+        value = tw.get_variable(name=name, name_literal=False)
+        print(f"[GET] Variable '{name}' = {value}")
+        return value
+    except Exception as e:
+        print(f"[ERROR] Failed to get variable '{name}': {e}")
         return None
 
 def set_var(name, value):
     try:
-        return tw.set_variable(name=name, value=value, name_literal=False)
-    except Exception:
-        return None
+        tw.set_variable(name=name, value=value, name_literal=False)
+        print(f"[SET] Variable '{name}' = {value}")
+        return True
+    except Exception as e:
+        print(f"[ERROR] Failed to set variable '{name}': {e}")
+        return False
 
 # --- n-chars.txt の読み込み ---
+print("[INFO] Loading n-chars.txt...")
 with open("turbowarp-server/n-chars.txt", "r", encoding="utf-8") as f:
     n_chars = [line.strip() for line in f]
+print(f"[INFO] Loaded {len(n_chars)} characters.")
 
 def decode_prompt(encoded_str):
     """クラウド変数の数字列を元にプロンプト文字列に復号"""
@@ -36,7 +46,9 @@ def decode_prompt(encoded_str):
         idx = int(encoded_str[i:i+2])
         if idx < len(n_chars):
             chars.append(n_chars[idx])
-    return "".join(chars)
+    decoded = "".join(chars)
+    print(f"[DECODE] Encoded prompt '{encoded_str[:20]}...' -> '{decoded[:30]}...'")
+    return decoded
 
 def rgb_to_scratch_number(rgb):
     """RGBを0-999999の10進数に変換"""
@@ -46,6 +58,7 @@ def rgb_to_scratch_number(rgb):
 def generate_image(prompt):
     """APIから352x352の画像生成"""
     seed = random.randint(0, 999999)
+    print(f"[INFO] Generating image with prompt: '{prompt[:50]}...', seed={seed}")
     params = {
         "prompt": prompt,
         "negative_prompt": "nsfw, low quality",
@@ -58,15 +71,19 @@ def generate_image(prompt):
     }
     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))
+    img = Image.open(BytesIO(resp.content))
+    print(f"[INFO] Image generated: size={img.size}, mode={img.mode}")
+    return img
 
 def resize_and_encode(img):
     """90x90にリサイズして数字列に変換"""
     img = img.resize((90, 90))
+    print("[INFO] Resized image to 90x90.")
     encoded = ""
     for y in range(90):
         for x in range(90):
             encoded += rgb_to_scratch_number(img.getpixel((x, y)))
+    print(f"[INFO] Image encoded to string of length {len(encoded)}.")
     return "10" + encoded  # 先頭に "10"
 
 def split_packets(data, max_length=9998):
@@ -77,14 +94,18 @@ def split_packets(data, max_length=9998):
         chunk = data[idx:idx + max_length]
         packets.append(chunk)
         idx += max_length
+    print(f"[INFO] Data split into {len(packets)} packet(s).")
     return packets
 
 # --- メインループ ---
+print("[INFO] Starting main loop...")
 while True:
     try:
         n1 = get_var("n1")
         if n1 and len(n1) >= 3 and n1[0] == "0":  # ID受信フラグ
+            print("[INFO] Received new request from n1.")
             if get_var("n0") == "0":  # 他ユーザー未使用
+                print("[INFO] n0 is free, starting processing...")
                 set_var("n0", "1")
                 user_id = n1[1:3]  # 実際のIDはここを調整
                 encoded_prompt = n1[3:]
@@ -96,15 +117,18 @@ while True:
                 
                 # パケット分割
                 packets = split_packets(scratch_data)
-                for pkt in packets:
+                for i, pkt in enumerate(packets):
+                    print(f"[INFO] Sending packet {i+1}/{len(packets)}")
                     set_var("n1", pkt)  # 送信
                     time.sleep(0.2)
                     # 次パケット送信待ち
                     while get_var("n1") != "11":
+                        print("[INFO] Waiting for ACK '11'...")
                         time.sleep(0.1)
                 
                 # 完了後リセット
+                print("[INFO] Transmission complete. Resetting n0.")
                 set_var("n0", "0")
     except Exception as e:
-        print("Error:", e)
+        print(f"[ERROR] Exception in main loop: {e}")
     time.sleep(0.2)