Update app.py

app.py

@@ -16,43 +16,42 @@ CLOTH_DIR = f"{ROOT}/test_color"
 PAIRS = f"{ROOT}/test_pairs.txt"
 
 def prepare_onepair_dataset(user_img_path: str, cloth_img_path: str):
-    # Reset the temp dataset folder each run
+    # Reset temp dataset
     if os.path.exists(ROOT):
         shutil.rmtree(ROOT)
 
-    split_dir = f"{ROOT}/test"
-    img_dir = f"{split_dir}/image"
-    cloth_dir = f"{split_dir}/cloth"
-    cloth_maskdir = f"{split_dir}/cloth-mask"
-
+    # Expected DCI-VTON structure
+    img_dir   = os.path.join(ROOT, "test", "image")
+    cloth_dir = os.path.join(ROOT, "test", "cloth")
+    cm_dir    = os.path.join(ROOT, "test", "cloth-mask")
     os.makedirs(img_dir, exist_ok=True)
     os.makedirs(cloth_dir, exist_ok=True)
-    os.makedirs(cloth_maskdir, exist_ok=True)
+    os.makedirs(cm_dir, exist_ok=True)
+
+    user_name  = "person.jpg"  # person image filename
+    cloth_name = "cloth.jpg"   # garment image filename
 
-    user_name = "person.jpg"
-    cloth_name = "cloth.jpg"
+    # Write pairs: CLOTH FIRST, then PERSON (as CPDataset expects)
+    with open(os.path.join(ROOT, "test_pairs.txt"), "w") as f:
+        f.write(f"{cloth_name} {user_name}\n")
 
-    # Save the inputs into expected folders
-    shutil.copy(user_img_path, f"{img_dir}/{user_name}")
-    shutil.copy(cloth_img_path, f"{cloth_dir}/{cloth_name}")
+    # Save images in expected places
+    shutil.copy(user_img_path,  os.path.join(img_dir,   user_name))
+    shutil.copy(cloth_img_path, os.path.join(cloth_dir, cloth_name))
 
-    # --- Build cloth mask using rembg alpha ---
-    cut_rgba = remove(Image.open(cloth_img_path).convert("RGBA"))
-    alpha = np.array(cut_rgba.split()[-1])  # alpha channel
-    binmask = Image.fromarray((alpha > 10).astype(np.uint8) * 255, mode="L")
-    binmask.save(f"{cloth_maskdir}/{cloth_name}", quality=95)
+    # --- Compatibility shim ---
+    # Some codepaths also try to read image/cloth.jpg; make a duplicate.
+    shutil.copy(user_img_path, os.path.join(img_dir, cloth_name))
 
-    # IMPORTANT: person (image) first, then cloth
-    with open(PAIRS, "w") as f:
-        f.write(f"{cloth_name} {user_name}\n")
+    # Reuse the garment as a coarse cloth-mask (you can replace with a real mask later)
+    shutil.copy(cloth_img_path, os.path.join(cm_dir, cloth_name))
 
-    print("PAIR LINE:", open(PAIRS).read().strip())
-    print("TREE:")
-    
+    # Debug print
+    print("PAIR LINE:", open(os.path.join(ROOT, "test_pairs.txt")).read().strip())
     for p in ["test/image", "test/cloth", "test/cloth-mask"]:
-        print(p, "=>", os.listdir(os.path.join(ROOT, p)))
+        full = os.path.join(ROOT, p)
+        print(p, "=>", os.listdir(full) if os.path.exists(full) else "MISSING")
 
-    
     return ROOT