Update app.py

app.py

@@ -41,7 +41,7 @@ def segment_dress(image_np):
     return mask
 
 def change_dress_color(image_path, color):
-    """Change the dress color based on the detected dress mask."""
+    """Change the dress color naturally while keeping textures."""
     if image_path is None:
         return None
 
@@ -66,13 +66,16 @@ def change_dress_color(image_path, color):
     img_lab = cv2.cvtColor(img_np, cv2.COLOR_RGB2LAB)
     new_color_lab = cv2.cvtColor(np.uint8([[new_color_bgr]]), cv2.COLOR_BGR2LAB)[0][0]
     
-    # Apply the new color while preserving texture
+    # Preserve texture by only modifying the A & B channels
     img_lab[..., 1] = np.where(mask == 255, new_color_lab[1], img_lab[..., 1])  # Modify A-channel
     img_lab[..., 2] = np.where(mask == 255, new_color_lab[2], img_lab[..., 2])  # Modify B-channel
 
     # Convert back to RGB
     img_recolored = cv2.cvtColor(img_lab, cv2.COLOR_LAB2RGB)
 
+    # Apply Poisson blending for realistic color application
+    img_recolored = cv2.seamlessClone(img_recolored, img_np, mask, (img_np.shape[1]//2, img_np.shape[0]//2), cv2.NORMAL_CLONE)
+
     return Image.fromarray(img_recolored)
 
 # Gradio Interface
@@ -84,7 +87,7 @@ demo = gr.Interface(
     ],
     outputs=gr.Image(type="pil", label="Color Changed Dress"),
     title="Dress Color Changer",
-    description="Upload an image of a dress and select a new color to change its appearance."
+    description="Upload an image of a dress and select a new color to change its appearance naturally."
 )
 
 if __name__ == "__main__":