Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -4,6 +4,7 @@ import requests
 from PIL import Image
 from io import BytesIO
 import scratchcommunication
+import traceback
 
 # --- Scratchクラウド接続 ---
 PROJECT_ID = 1293416663
@@ -16,68 +17,105 @@ tw = scratchcommunication.TwCloudConnection(
 def get_var(name):
     try:
         return tw.get_variable(name=name, name_literal=False)
-    except Exception:
+    except Exception as e:
+        print(f"[ERROR] get_var('{name}') failed: {e}")
+        traceback.print_exc()
         return None
 
 def set_var(name, value):
     try:
         return tw.set_variable(name=name, value=value, name_literal=False)
-    except Exception:
+    except Exception as e:
+        print(f"[ERROR] set_var('{name}', {value}) failed: {e}")
+        traceback.print_exc()
         return None
 
 # --- n-chars.txt の読み込み ---
-with open("turbowarp-server/n-chars.txt", "r", encoding="utf-8") as f:
-    n_chars = [line.strip() for line in f]
+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}")
+    traceback.print_exc()
+    n_chars = []
 
 def decode_prompt(encoded_str):
     """クラウド変数の数字列を元にプロンプト文字列に復号"""
-    chars = []
-    for i in range(0, len(encoded_str), 2):
-        idx = int(encoded_str[i:i+2])
-        if idx < len(n_chars):
-            chars.append(n_chars[idx])
-    return "".join(chars)
+    try:
+        chars = []
+        for i in range(0, len(encoded_str), 2):
+            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 ""
 
 def rgb_to_scratch_number(rgb):
-    """RGBを0-999999の10進数に変換"""
-    r, g, b = rgb
-    return f"{r:03}{g:03}{b:03}"
+    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"
 
 def generate_image(prompt):
     """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))
+    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=30
+        )
+        resp.raise_for_status()
+        return Image.open(BytesIO(resp.content))
+    except Exception as e:
+        print(f"[ERROR] generate_image failed for prompt '{prompt}': {e}")
+        traceback.print_exc()
+        return None
 
 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  # 先頭に "10"
+    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"
 
 def split_packets(data, max_length=9998):
     """10000文字制限に合わせて分割（先頭 '10' 含む）"""
-    packets = []
-    idx = 0
-    while idx < len(data):
-        chunk = data[idx:idx + max_length]
-        packets.append(chunk)
-        idx += max_length
-    return packets
+    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]
 
 # --- メインループ ---
 while True:
@@ -89,22 +127,36 @@ while True:
                 user_id = n1[1:3]  # 実際のIDはここを調整
                 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")
+                    continue
+
                 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)
-                
+                    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()
+
                 # 完了後リセット
                 set_var("n0", "0")
     except Exception as e:
-        print("Error:", e)
+        print(f"[ERROR] Main loop exception: {e}")
+        traceback.print_exc()
     time.sleep(0.2)