Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -52,13 +52,35 @@ 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 == " ":  # 半角スペースを特別に96として処理
+            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:  # 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
@@ -97,7 +119,7 @@ def resize_and_encode(img):
         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 encoded  # 先頭に "10"
+    return encoded
 
 def split_packets(data, max_length=9998):
     """10000文字制限に合わせて分割（先頭 '10' 含む）"""