Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -17,8 +17,14 @@ tw = scratchcommunication.TwCloudConnection(
 def get_var(name):
     try:
         return tw.get_variable(name=name, name_literal=False)
+    except scratchcommunication.exceptions.NotSupported as e:
+        print(f"[WARN] NotSupported error while getting '{name}': {e}")
+        return None
+    except KeyError as e:
+        print(f"[ERROR] KeyError while getting '{name}': {e}")
+        return None
     except Exception as e:
-        print(f"[ERROR] get_var('{name}') failed: {e}")
+        print(f"[ERROR] Unexpected error while getting '{name}': {e}")
         traceback.print_exc()
         return None
 
@@ -26,7 +32,7 @@ def set_var(name, value):
     try:
         return tw.set_variable(name=name, value=value, name_literal=False)
     except Exception as e:
-        print(f"[ERROR] set_var('{name}', {value}) failed: {e}")
+        print(f"[ERROR] Failed to set '{name}' to '{value}': {e}")
         traceback.print_exc()
         return None
 
@@ -35,35 +41,26 @@ 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"[ERROR] Failed to load n-chars.txt: {e}")
+    print(f"[FATAL] Failed to load n-chars.txt: {e}")
     traceback.print_exc()
     n_chars = []
 
 def decode_prompt(encoded_str):
-    """クラウド変数の数字列を元にプロンプト文字列に復号"""
-    try:
-        chars = []
-        for i in range(0, len(encoded_str), 2):
+    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])
-        return "".join(chars)
-    except Exception as e:
-        print(f"[ERROR] decode_prompt failed for '{encoded_str}': {e}")
-        traceback.print_exc()
-        return ""
+        except Exception as e:
+            print(f"[WARN] Failed to decode chunk '{encoded_str[i:i+2]}': {e}")
+    return "".join(chars)
 
 def rgb_to_scratch_number(rgb):
-    try:
-        r, g, b = rgb
-        return f"{r:03}{g:03}{b:03}"
-    except Exception as e:
-        print(f"[ERROR] rgb_to_scratch_number failed for {rgb}: {e}")
-        traceback.print_exc()
-        return "000000000"
+    r, g, b = rgb
+    return f"{r:03}{g:03}{b:03}"
 
 def generate_image(prompt):
-    """APIから352x352の画像生成"""
     try:
         seed = random.randint(0, 999999)
         params = {
@@ -79,84 +76,60 @@ def generate_image(prompt):
         resp = requests.get(
             "https://izuemon-pixart-alpha-pixart-sigma-xl-2-1024-ms.hf.space/gen",
             params=params,
-            timeout=30
+            timeout=20
         )
         resp.raise_for_status()
         return Image.open(BytesIO(resp.content))
     except Exception as e:
-        print(f"[ERROR] generate_image failed for prompt '{prompt}': {e}")
+        print(f"[ERROR] Image generation failed for prompt '{prompt}': {e}")
         traceback.print_exc()
         return None
 
 def resize_and_encode(img):
-    """90x90にリサイズして数字列に変換"""
-    try:
-        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"
-    except Exception as e:
-        print(f"[ERROR] resize_and_encode failed: {e}")
-        traceback.print_exc()
-        return "10"
+    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
 
 def split_packets(data, max_length=9998):
-    """10000文字制限に合わせて分割（先頭 '10' 含む）"""
-    try:
-        packets = []
-        idx = 0
-        while idx < len(data):
-            chunk = data[idx:idx + max_length]
-            packets.append(chunk)
-            idx += max_length
-        return packets
-    except Exception as e:
-        print(f"[ERROR] split_packets failed: {e}")
-        traceback.print_exc()
-        return [data]
+    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 = get_var("n1")
-        if n1 and len(n1) >= 3 and n1[0] == "0":  # ID受信フラグ
-            if get_var("n0") == "0":  # 他ユーザー未使用
+        if n1 and len(n1) >= 3 and n1[0] == "0":
+            if get_var("n0") == "0":
                 set_var("n0", "1")
-                user_id = n1[1:3]  # 実際のIDはここを調整
+                user_id = n1[1:3]
                 encoded_prompt = n1[3:]
                 prompt = decode_prompt(encoded_prompt)
-
-                # 画像生成
+                
                 img = generate_image(prompt)
                 if img is None:
-                    print(f"[WARNING] Image generation failed for user {user_id}. Skipping.")
                     set_var("n0", "0")
+                    time.sleep(1)
                     continue
-
+                
                 scratch_data = resize_and_encode(img)
-
-                # パケット分割
                 packets = split_packets(scratch_data)
+                
                 for pkt in packets:
-                    set_var("n1", pkt)  # 送信
+                    set_var("n1", pkt)
                     time.sleep(0.2)
-                    # 次パケット送信待ち
-                    try:
-                        timeout = time.time() + 10  # 10秒でタイムアウト
-                        while get_var("n1") != "11":
-                            if time.time() > timeout:
-                                print(f"[WARNING] Packet acknowledgement timeout for user {user_id}.")
-                                break
-                            time.sleep(0.1)
-                    except Exception as e:
-                        print(f"[ERROR] Waiting for packet ack failed: {e}")
-                        traceback.print_exc()
-
-                # 完了後リセット
+                    while get_var("n1") != "11":
+                        time.sleep(0.1)
+                
                 set_var("n0", "0")
     except Exception as e:
-        print(f"[ERROR] Main loop exception: {e}")
+        print(f"[CRITICAL] Main loop error: {e}")
         traceback.print_exc()
     time.sleep(0.2)