Update inpainting_templates.py

inpainting_templates.py

@@ -32,6 +32,10 @@ class InpaintingTemplate:
     # Conditioning type preference
     preferred_conditioning: str = "canny"  # "canny" or "depth"
 
+    # Structure preservation in masked area
+    # True = keep edges in mask (for color change), False = clear edges (for replacement/removal)
+    preserve_structure_in_mask: bool = False
+
     # Prompt enhancement control
     enhance_prompt: bool = True  # Whether to use OpenCLIP prompt enhancement
 
@@ -64,34 +68,31 @@ class InpaintingTemplateManager:
         # 4 CORE TEMPLATES - Optimized for Speed & Quality
         # ========================================
 
-        # 1. CHANGE COLOR - Pure color transformation
+        # 1. CHANGE COLOR - Pure color transformation (preserves structure)
         "change_color": InpaintingTemplate(
             key="change_color",
             name="Change Color",
             category="Color",
             icon="🎨",
-            description="Change color ONLY - works on any object (clothes, walls, furniture) based on mask",
-            prompt_template="repaint as {content} color, solid {content} only",  # Clear action verb + color instruction
+            description="Change color ONLY - preserves original structure, only changes the fill color",
+            prompt_template="{content} colored, solid {content} color, flat color, same structure",
             negative_prompt=(
                 "original color, keeping same color, unchanged color, "
                 "black, dark, keeping black, maintaining black color, "
                 "black clothing, dark colors, dark fabric, black fabric, "
-                "partial color change, incomplete transformation, "
-                "texture details, fabric patterns, material description, object names, "
-                "realistic lighting effects, complex shadows, photorealistic rendering, "
-                "style description, clothing type, object description, "
-                "uniform, military uniform, military clothing, military outfit, army uniform, "
-                "style change, clothing design change, fashion change, outfit change, "
-                "costume, formal wear, changing garment type, different clothing style, "
-                "suit, jacket, vest, coat, dress uniform"
+                "patterns, floral, stripes, plaid, checkered, decorative patterns, "
+                "texture change, material change, fabric change, "
+                "new design, different style, style change, "
+                "complex patterns, printed patterns, embroidery"
             ),
-            controlnet_conditioning_scale=0.08,   # Minimal: maximum color freedom for extreme transformations
-            feather_radius=6,                     # Low: reduces blending with original color
-            guidance_scale=20.0,                  # Extremely high: forces complete color transformation
+            controlnet_conditioning_scale=0.6,    # Medium-high: preserve structure/shape from edges
+            feather_radius=4,                     # Low: clean color boundaries
+            guidance_scale=12.0,                  # High: follow color prompt
             num_inference_steps=10,               # Optimized for speed
-            strength=1.0,                         # Full repaint: completely replace original color
-            preferred_conditioning="canny",       # Edge-based + mask-aware: preserves shape, allows color change
-            enhance_prompt=False,                 # Disabled: avoid interference from object descriptions
+            strength=1.0,                         # Full repaint for color change
+            preferred_conditioning="canny",       # Edge-based: preserves shape
+            preserve_structure_in_mask=True,      # KEY: keep edges to preserve clothing structure
+            enhance_prompt=False,                 # Disabled: use color prompt directly
             difficulty="easy",
             usage_tips=[
                 "🎯 Purpose: Pure color transformation (any object type)",
@@ -377,6 +378,7 @@ class InpaintingTemplateManager:
             "num_inference_steps": template.num_inference_steps,
             "strength": template.strength,
             "preferred_conditioning": template.preferred_conditioning,
+            "preserve_structure_in_mask": template.preserve_structure_in_mask,
             "enhance_prompt": template.enhance_prompt
         }