Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -18,16 +18,35 @@ print("[INFO] Connected successfully.")
 def get_var(name):
     try:
         value = tw.get_variable(name=name, name_literal=False)
-        print(f"[GET] Variable '{name}' = {value[:50]}..." if value and len(value) > 50 else f"[GET] Variable '{name}' = {value}")
+        
+        # デバッグ用に生の値を出力
+        print(f"[DEBUG] Raw value of '{name}': {value} (type: {type(value).__name__})")
+        
+        # Noneの場合は"0"として扱う
+        if value is None:
+            print(f"[WARNING] Variable '{name}' is None, treating as '0'")
+            return "0"
+        
+        # 様々な型を文字列に変換
+        if isinstance(value, (int, float)):
+            value = str(int(value))
+            print(f"[GET] Variable '{name}' converted from number to string: '{value}'")
+        elif not isinstance(value, str):
+            value = str(value)
+            print(f"[GET] Variable '{name}' converted from {type(value).__name__} to string: '{value}'")
+        
+        # 表示用に切り詰め
+        display_value = value[:50] + "..." if len(value) > 50 else value
+        print(f"[GET] Variable '{name}' = '{display_value}'")
         return value
     except Exception as e:
         print(f"[ERROR] Failed to get variable '{name}': {e}")
-        return None
+        return "0"
 
 def set_var(name, value):
     try:
         tw.set_variable(name=name, value=value, name_literal=False)
-        print(f"[SET] Variable '{name}' = {value[:50]}..." if len(value) > 50 else f"[SET] Variable '{name}' = {value}")
+        print(f"[SET] Variable '{name}' = '{value[:50]}...'" if len(value) > 50 else f"[SET] Variable '{name}' = '{value}'")
         return True
     except Exception as e:
         print(f"[ERROR] Failed to set variable '{name}': {e}")
@@ -39,21 +58,47 @@ 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 encode_prompt(prompt):
+    """プロンプト文字列を数字列にエンコード（半角スペースは96として特別処理）"""
+    encoded = ""
+    for char in prompt:
+        if char == " ":
+            encoded += "96"
+        else:
+            try:
+                idx = n_chars.index(char)
+                encoded += f"{idx:02d}"
+            except ValueError:
+                print(f"[WARNING] Character '{char}' not found in n_chars, skipping")
+                continue
+    print(f"[ENCODE] Prompt encoded to length {len(encoded)}")
+    return encoded
+
 def decode_prompt(encoded_str):
-    """クラウド変数の数字列を元にプロンプト文字列に復号"""
+    """クラウド変数の数字列を元にプロンプト文字列に復号（96は半角スペースとして処理）"""
     chars = []
-    for i in range(0, len(encoded_str), 2):
+    i = 0
+    while i < len(encoded_str):
         idx = int(encoded_str[i:i+2])
-        if idx < len(n_chars):
+        if idx == 96:
+            chars.append(" ")
+        elif idx < len(n_chars):
             chars.append(n_chars[idx])
+        else:
+            print(f"[WARNING] Invalid index {idx} at position {i}")
+        i += 2
     decoded = "".join(chars)
     print(f"[DECODE] Encoded prompt -> '{decoded[:50]}...'")
     return decoded
 
 def rgb_to_scratch_number(rgb):
-    """RGBを0-999999の10進数に変換"""
+    """RGBを6桁の10進数に変換 (R:00-99, G:00-99, B:00-99)"""
     r, g, b = rgb
-    return f"{r:03}{g:03}{b:03}"
+    # 各色を0-99の範囲に収める（元が0-255の場合）
+    r_scaled = int(r * 99 / 255)
+    g_scaled = int(g * 99 / 255)
+    b_scaled = int(b * 99 / 255)
+    return f"{r_scaled:02d}{g_scaled:02d}{b_scaled:02d}"
 
 def generate_image(prompt):
     """APIから352x352の画像生成"""
@@ -76,122 +121,161 @@ def generate_image(prompt):
     return img
 
 def resize_and_encode(img):
-    """90x90にリサイズして数字列に変換"""
+    """90x90にリサイズして6桁の数字列に変換（各ピクセル6桁）"""
     img = img.resize((90, 90))
     print("[INFO] Resized image to 90x90.")
+    
     encoded = ""
+    total_pixels = 0
+    
     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"
+            pixel = img.getpixel((x, y))
+            # RGB値を取得（RGBAの場合はRGBのみ使用）
+            if len(pixel) >= 3:
+                r, g, b = pixel[:3]
+            else:
+                r, g, b = pixel, pixel, pixel
+            encoded += rgb_to_scratch_number((r, g, b))
+            total_pixels += 1
+    
+    expected_length = total_pixels * 6
+    print(f"[INFO] Encoded {total_pixels} pixels to string of length {len(encoded)} (expected: {expected_length})")
+    return encoded
 
 def split_packets(data, max_length=9998):
     """10000文字制限に合わせて分割（先頭 '10' 含む）"""
     packets = []
     idx = 0
+    total_length = len(data)
+    expected_packets = (total_length + max_length - 1) // max_length
+    
     while idx < len(data):
         chunk = data[idx:idx + max_length]
         packets.append(chunk)
         idx += max_length
-    print(f"[INFO] Data split into {len(packets)} packet(s).")
+    
+    print(f"[INFO] Data length: {total_length}, max packet size: {max_length}")
+    print(f"[INFO] Expected {expected_packets} packets, actual {len(packets)} packets")
     return packets
 
 # --- メインループ ---
 print("[INFO] Starting main loop...")
-last_processed_id = ""  # 最後に処理したIDを記録
+last_processed_id = ""
+
+# 初期化
+set_var("n0", "0")
+set_var("n1", "0")
+time.sleep(1)
 
 while True:
     try:
         n1 = get_var("n1")
         n0 = get_var("n0")
         
-        # n1が存在し、長さが3以上で、先頭が"0"（新規リクエスト）
+        print(f"[DEBUG] n0='{n0}', n1='{n1[:30]}...' if n1 and len(n1)>30 else n1")
+        
+        is_n0_free = (n0 == "0")
+        print(f"[DEBUG] is_n0_free: {is_n0_free}")
+        
         if n1 and len(n1) >= 3 and n1[0] == "0":
-            user_id = n1[1:3]  # ユーザーID（2桁）
-            request_data = n1  # 完全なリクエストデータ
+            user_id = n1[1:3]
+            request_data = n1
             
             print(f"[INFO] Received request from user {user_id}")
             print(f"[INFO] Request data length: {len(request_data)}")
             
-            # 同じリクエストを繰り返し処理しないようにチェック
             if request_data == last_processed_id:
-                print("[INFO] Already processed this request, marking as read...")
-                # 既読マークをつける（先頭を"01"に変更）
+                print("[INFO] Already processed, marking as read...")
                 marked_as_read = "01" + request_data[2:]
                 set_var("n1", marked_as_read)
                 time.sleep(0.1)
                 continue
             
-            # n0が利用可能かチェック
-            if n0 == "0" or n0 is None:
+            if is_n0_free:
                 print("[INFO] n0 is free, starting processing...")
                 
-                # 処理中フラグをセット
+                # 処理中フラグ
                 set_var("n0", "1")
                 
-                # リクエストを既読にする（先頭を"01"に変更）
+                # リクエストを既読に
                 marked_as_read = "01" + request_data[2:]
                 set_var("n1", marked_as_read)
                 
-                # プロンプトをデコード（先頭の"0" + ユーザーID(2桁)を除去）
+                # プロンプトをデコード
                 encoded_prompt = request_data[3:]
+                print(f"[INFO] Encoded prompt length: {len(encoded_prompt)}")
                 prompt = decode_prompt(encoded_prompt)
+                print(f"[INFO] Decoded prompt: '{prompt}'")
                 
-                # 画像生成
+                # 画像生成とエンコード
                 img = generate_image(prompt)
                 scratch_data = resize_and_encode(img)
                 
                 # パケット分割
                 packets = split_packets(scratch_data)
+                print(f"[INFO] Will send {len(packets)} packets")
                 
-                # 最初のパケットを送信
+                # パケット送信
                 for i, pkt in enumerate(packets):
-                    print(f"[INFO] Sending packet {i+1}/{len(packets)}")
-                    set_var("n1", pkt)
+                    packet_num = i + 1
+                    print(f"[INFO] Sending packet {packet_num}/{len(packets)} (size: {len(pkt)} chars)")
+                    
+                    # パケット送信
+                    set_var("n1", "10" + pkt)
                     
                     if i < len(packets) - 1:
-                        # 次のパケットのACKを待つ
-                        print("[INFO] Waiting for ACK '11'...")
+                        # ACK待ち
+                        print(f"[INFO] Waiting for ACK for packet {packet_num}...")
                         ack_timeout = 0
-                        while ack_timeout < 50:  # 5秒タイムアウト
+                        ack_received = False
+                        while ack_timeout < 50:
                             current_n1 = get_var("n1")
                             if current_n1 == "11":
-                                print("[INFO] Received ACK")
+                                print(f"[INFO] Received ACK for packet {packet_num}")
+                                ack_received = True
                                 break
-                            time.sleep(0.1)
+                            time.sleep(0.2)
                             ack_timeout += 1
                         
-                        if ack_timeout >= 50:
-                            print("[ERROR] ACK timeout")
+                        if not ack_received:
+                            print(f"[ERROR] ACK timeout for packet {packet_num}")
                             set_var("n0", "0")
                             break
                     else:
-                        # 最後のパケットは完了を待つ
+                        # 完了待ち
                         print("[INFO] Waiting for completion '99'...")
                         complete_timeout = 0
+                        complete_received = False
                         while complete_timeout < 50:
                             current_n1 = get_var("n1")
                             if current_n1 == "99":
                                 print("[INFO] Received completion signal")
+                                complete_received = True
                                 break
-                            time.sleep(0.1)
+                            time.sleep(0.2)
                             complete_timeout += 1
+                        
+                        if not complete_received:
+                            print("[ERROR] Completion timeout")
                 
-                # 完了後リセット
+                # 完了
                 print("[INFO] Transmission complete. Resetting n0.")
                 set_var("n0", "0")
-                last_processed_id = request_data  # 処理済みとして記録
+                last_processed_id = request_data
+                print(f"[INFO] Set last_processed_id")
                 
             else:
                 print("[INFO] n0 is busy, waiting...")
         else:
-            # 何もしない
-            pass
+            if n1:
+                print(f"[DEBUG] n1 condition not met: first char='{n1[0] if n1 else 'None'}', length={len(n1) if n1 else 0}")
+            else:
+                print("[DEBUG] n1 is None or empty")
             
     except Exception as e:
         print(f"[ERROR] Exception in main loop: {e}")
         import traceback
         traceback.print_exc()
     
-    time.sleep(0.2)
+    time.sleep(0.3)