Update app_main.py

app_main.py

@@ -305,35 +305,35 @@ def similarity_matching(input_json_path: str) -> str:
     # ============================== #
     clip_embd = OpenCLIPEmbeddings()
 
-    # ========================================= #
-    #  Walk folders to collect all image paths  #
-    # ========================================= #
-    folder_image_paths = []
-    for image_dir in image_dirs:
-        for root, _, files in os.walk(image_dir):
-            for fname in files:
-                if fname.lower().endswith((".png", ".jpg", ".jpeg")):
-                    folder_image_paths.append(os.path.join(root, fname))
-
-    # ============================== #
-    #   EMBED FOLDER IMAGES (REF)    #
-    # ============================== #
-    img_features = clip_embd.embed_image(folder_image_paths)
-
-    # ============================== #
-    #     Store image embeddings     #
-    # ============================== #
-    embedding_json = []
-    for i, path in enumerate(folder_image_paths):
-        embedding_json.append({
-            "name":os.path.basename(path),
-            "file-path": path,
-            "embeddings": list(img_features[i])
-        })
+    # # ========================================= #
+    # #  Walk folders to collect all image paths  #
+    # # ========================================= #
+    # folder_image_paths = []
+    # for image_dir in image_dirs:
+    #     for root, _, files in os.walk(image_dir):
+    #         for fname in files:
+    #             if fname.lower().endswith((".png", ".jpg", ".jpeg")):
+    #                 folder_image_paths.append(os.path.join(root, fname))
+
+    # # ============================== #
+    # #   EMBED FOLDER IMAGES (REF)    #
+    # # ============================== #
+    # img_features = clip_embd.embed_image(folder_image_paths)
+
+    # # ============================== #
+    # #     Store image embeddings     #
+    # # ============================== #
+    # embedding_json = []
+    # for i, path in enumerate(folder_image_paths):
+    #     embedding_json.append({
+    #         "name":os.path.basename(path),
+    #         "file-path": path,
+    #         "embeddings": list(img_features[i])
+    #     })
     
-    # Save to embeddings.json
-    with open(f"{OUTPUT_FOLDER}/embeddings.json", "w") as f:
-        json.dump(embedding_json, f, indent=2)
+    # # Save to embeddings.json
+    # with open(f"{OUTPUT_FOLDER}/embeddings.json", "w") as f:
+    #     json.dump(embedding_json, f, indent=2)
         
     # ============================== #
     #      DECODE SPRITE IMAGES      #
@@ -355,8 +355,8 @@ def similarity_matching(input_json_path: str) -> str:
     # ============================== #
     #     COMPUTE SIMILARITIES       #
     # ============================== #
-    # with open(f"{OUTPUT_FOLDER}/embeddings.json", "r") as f:
-    #     embedding_json = json.load(f)
+    with open(f"{OUTPUT_FOLDER}/embeddings.json", "r") as f:
+        embedding_json = json.load(f)
         
     img_matrix = np.array([img["embeddings"] for img in embedding_json])
     sprite_matrix = np.array(sprite_features)