Update app.py

app.py

@@ -3442,20 +3442,36 @@ SPRITE_DIR/"Zebra.sprite3"/"f3e322a25b9f79801066056de6f33fb1.png"
     # # ============================== #
     # #      EMBED SPRITE IMAGES       #
     # # ============================== #
+    # sprite_features = []
+    # for b64 in sprite_base64:
+    #     if "," in b64:  # strip data URI prefix if present
+    #         b64 = b64.split(",", 1)[1]
+    
+    #     img_bytes = base64.b64decode(b64)
+    #     pil_img = Image.open(BytesIO(img_bytes)).convert("RGB")
+    
+    #     # optional re-encode to PNG for CLIP
+    #     buf = BytesIO()
+    #     pil_img.save(buf, format="PNG")
+    #     buf.seek(0)
+    
+    #     feats = clip_embd.embed_image([buf])[0]  # extract CLIP embedding
+    #     sprite_features.append(feats)
+
     sprite_features = []
-    for b64 in sprite_base64:
-        if "," in b64:  # strip data URI prefix if present
+    for sprite in sprites_data.values():  # assuming dict like {"Sprite 1": {...}, ...}
+        b64 = sprite["base64"]
+        if "," in b64:
             b64 = b64.split(",", 1)[1]
     
         img_bytes = base64.b64decode(b64)
         pil_img = Image.open(BytesIO(img_bytes)).convert("RGB")
     
-        # optional re-encode to PNG for CLIP
         buf = BytesIO()
         pil_img.save(buf, format="PNG")
         buf.seek(0)
     
-        feats = clip_embd.embed_image([buf])[0]  # extract CLIP embedding
+        feats = clip_embd.embed_image([buf])[0]
         sprite_features.append(feats)
     
     # sprite_matrix = np.array(sprite_features, dtype=np.float32)