Conditional K Value

app.py

@@ -53,8 +53,10 @@ def depth_to_normal(depth):
 # CORE PROCESSING FUNCTION
 # ===============================
 def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image, k: int = 5):
-    img_pil = base_image.convert("RGB")
-    img_np = np.array(img_pil)
+    # Auto-set k for BASE-BOTTOM images
+    filename = getattr(base_image, "filename", "") or ""
+    if any(name in filename for name in ["BASE-BOTTOM-1.png"]):
+        k = 8
 
     # Prepare tensor
     img_resized = img_pil.resize((384, 384))