Update inpainting_templates.py

inpainting_templates.py

@@ -68,46 +68,47 @@ class InpaintingTemplateManager:
         # 4 CORE TEMPLATES - Optimized for Speed & Quality
         # ========================================
 
-        # 1. CHANGE COLOR - Pure color transformation (preserves structure)
+        # 1. CHANGE COLOR - Pure color transformation
         "change_color": InpaintingTemplate(
             key="change_color",
             name="Change Color",
             category="Color",
             icon="🎨",
-            description="Change color ONLY - preserves original structure, only changes the fill color",
-            prompt_template="{content} colored, solid {content} color, flat color, same structure",
+            description="Change color ONLY - fills the masked area with a solid, flat color",
+            prompt_template="{content} color, solid flat {content}, uniform color, no patterns, smooth surface",
             negative_prompt=(
                 "original color, keeping same color, unchanged color, "
                 "black, dark, keeping black, maintaining black color, "
                 "black clothing, dark colors, dark fabric, black fabric, "
                 "patterns, floral, stripes, plaid, checkered, decorative patterns, "
-                "texture change, material change, fabric change, "
-                "new design, different style, style change, "
+                "diamond pattern, grid pattern, geometric patterns, "
+                "texture, textured, wrinkles, folds, creases, "
+                "gradients, shading variations, color variations, "
                 "complex patterns, printed patterns, embroidery"
             ),
-            controlnet_conditioning_scale=0.5,    # Medium: balance structure preservation and color freedom
+            controlnet_conditioning_scale=0.3,    # Low-medium: allow color freedom in masked area
             feather_radius=4,                     # Low: clean color boundaries
-            guidance_scale=13.0,                  # High: follow color prompt strongly
+            guidance_scale=15.0,                  # Very high: strongly follow color prompt
             num_inference_steps=10,               # Optimized for speed
             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
+            preferred_conditioning="canny",       # Edge-based
+            preserve_structure_in_mask=False,     # KEY: clear edges in mask for pure color fill
             enhance_prompt=False,                 # Disabled: use color prompt directly
             difficulty="easy",
             usage_tips=[
-                "🎯 Purpose: Change the color while keeping the original shape and structure.",
+                "🎯 Purpose: Fill the masked area with a solid, uniform color.",
                 "",
                 "📝 Example Prompts:",
-                "  • 'vibrant red' - for a bold, saturated red color",
-                "  • 'soft pastel pink' - for a gentle, light pink tone",
-                "  • 'deep navy blue' - for a rich, dark blue shade",
-                "  • 'bright yellow' - for an eye-catching yellow",
-                "  • 'pure white' - for a clean, solid white",
+                "  • 'vibrant red' - bold, saturated red",
+                "  • 'soft pastel pink' - gentle, light pink",
+                "  • 'deep navy blue' - rich, dark blue",
+                "  • 'bright yellow' - eye-catching yellow",
+                "  • 'pure white' - clean, solid white",
                 "",
                 "💡 Tips:",
-                "  • Describe ONLY the color, not the object itself",
+                "  • Describe ONLY the color, not the object",
                 "  • Paint the entire area you want to recolor",
-                "  • Add modifiers like 'bright', 'dark', 'pastel' for better results"
+                "  • Use modifiers: 'bright', 'dark', 'pastel', 'vivid'"
             ]
         ),
 
@@ -118,12 +119,13 @@ class InpaintingTemplateManager:
             category="Replacement",
             icon="👕",
             description="Change clothing style, material, or design - can include color change",
-            prompt_template="{content} with proper fit and natural appearance",
+            prompt_template="{content}, photorealistic, realistic fabric texture, natural fit, high quality",
             negative_prompt=(
                 "wrong body proportions, floating fabric, unrealistic wrinkles, "
                 "mismatched lighting, visible edges, original clothing style, "
                 "keeping same color, original color, faded colors, unchanged appearance, partial change, "
-                "black clothing, dark original color, distorted body, naked, nudity"
+                "black clothing, dark original color, distorted body, naked, nudity, "
+                "cartoon, anime, illustration, drawing, painted"
             ),
             controlnet_conditioning_scale=0.30,   # Medium: preserves body structure, allows clothing change
             feather_radius=14,                    # Medium: natural blending with body
@@ -134,18 +136,19 @@ class InpaintingTemplateManager:
             enhance_prompt=True,                  # Enabled: enriches clothing details
             difficulty="easy",
             usage_tips=[
-                "🎯 Purpose: Replace clothing with a completely different style or design.",
+                "🎯 Purpose: Replace clothing with a different style, material, or design.",
                 "",
                 "📝 Example Prompts:",
-                "  • 'white polo shirt with collar' - casual business look",
-                "  • 'formal black suit with tie' - professional attire",
-                "  • 'cozy knit sweater in cream color' - warm casual style",
-                "  • 'vintage denim jacket with patches' - retro fashion",
-                "  • 'elegant silk blouse in emerald green' - sophisticated look",
+                "  • 'tailored charcoal suit with silk tie and white shirt' - formal business",
+                "  • 'navy blazer with gold buttons over light blue oxford shirt' - smart casual",
+                "  • 'black tuxedo with bow tie and white dress shirt' - elegant formal",
+                "  • 'white polo shirt with collar' - casual business",
+                "  • 'cozy cream knit sweater' - warm casual style",
+                "  • 'vintage denim jacket' - retro fashion",
                 "",
                 "💡 Tips:",
-                "  • Include both clothing type AND color for best results",
-                "  • Add details like 'with buttons', 'v-neck', 'long sleeves' for specificity",
+                "  • Include clothing type + color + details for best results",
+                "  • For suits: mention 'tailored', 'fitted', specific fabric like 'wool' or 'silk'",
                 "  • Body structure is preserved automatically"
             ]
         ),
@@ -157,12 +160,13 @@ class InpaintingTemplateManager:
             category="Replacement",
             icon="🔄",
             description="Replace objects (one type at a time) - all masked areas become the SAME object",
-            prompt_template="{content} in natural lighting, fitting the scene",
+            prompt_template="{content}, photorealistic, natural lighting, seamlessly integrated into scene, high quality",
             negative_prompt=(
                 "inconsistent lighting, wrong perspective, mismatched colors, "
                 "visible seams, floating objects, unrealistic placement, original object, "
                 "poorly integrated, disconnected from scene, keeping original, remnants of original, "
-                "multiple different objects, mixed objects, various items"
+                "multiple different objects, mixed objects, various items, "
+                "cartoon, anime, illustration, drawing, painted"
             ),
             controlnet_conditioning_scale=0.25,   # Low-medium: allows complete object replacement
             feather_radius=10,                    # Medium: natural scene integration
@@ -173,19 +177,19 @@ class InpaintingTemplateManager:
             enhance_prompt=True,                  # Enabled: enriches object details
             difficulty="medium",
             usage_tips=[
-                "🎯 Purpose: Replace an object in the scene with something completely different.",
+                "🎯 Purpose: Replace an object with something completely different.",
                 "",
                 "📝 Example Prompts:",
-                "  • 'ceramic vase with colorful flowers' - decorative item",
-                "  • 'modern laptop on a stand' - tech gadget",
-                "  • 'stack of vintage books' - classic decoration",
-                "  • 'potted succulent plant' - natural element",
-                "  • 'antique brass lamp' - lighting fixture",
+                "  • 'elegant ceramic vase with fresh roses' - decorative item",
+                "  • 'modern silver laptop on wooden stand' - tech gadget",
+                "  • 'stack of leather-bound vintage books' - classic decoration",
+                "  • 'healthy green potted succulent' - natural element",
+                "  • 'antique brass table lamp with fabric shade' - lighting",
                 "",
                 "💡 Tips:",
-                "  • Replace ONE object type at a time for best results",
+                "  • Replace ONE object type at a time",
                 "  • Describe what you want, not what you're removing",
-                "  • Include material and style details for realism"
+                "  • Include material and style for realistic results"
             ]
         ),
 
@@ -196,12 +200,13 @@ class InpaintingTemplateManager:
             category="Removal",
             icon="🗑️",
             description="Remove objects and naturally fill with background - describe the background material",
-            prompt_template="continue the background with {content}, seamless blending, natural continuation",
+            prompt_template="continue the background with {content}, photorealistic, seamless blending, natural texture continuation, high quality",
             negative_prompt=(
                 "new object appearing, adding items, inserting objects, "
                 "foreground elements, visible object, thing, item, "
                 "unnatural filling, visible patches, inconsistent texture, "
-                "mismatched pattern, color discontinuity, artificial blending"
+                "mismatched pattern, color discontinuity, artificial blending, "
+                "cartoon, anime, illustration, drawing, painted"
             ),
             controlnet_conditioning_scale=0.20,   # Low: allows creative background filling
             feather_radius=12,                    # Medium: smooth background blending
@@ -212,19 +217,19 @@ class InpaintingTemplateManager:
             enhance_prompt=False,                 # Disabled: avoid generating new objects
             difficulty="medium",
             usage_tips=[
-                "🎯 Purpose: Remove unwanted objects and fill the area with background.",
+                "🎯 Purpose: Remove unwanted objects and fill with background.",
                 "",
                 "📝 Example Prompts:",
-                "  • 'smooth wooden floor texture' - for indoor floors",
-                "  • 'clean white painted wall' - for wall backgrounds",
-                "  • 'natural green grass' - for outdoor lawn areas",
-                "  • 'beige carpet texture' - for carpeted floors",
-                "  • 'clear blue sky with clouds' - for sky backgrounds",
+                "  • 'polished hardwood floor with natural grain' - indoor floors",
+                "  • 'smooth white painted wall' - wall backgrounds",
+                "  • 'lush green grass lawn' - outdoor areas",
+                "  • 'soft beige carpet texture' - carpeted floors",
+                "  • 'clear blue sky with soft clouds' - sky backgrounds",
                 "",
                 "💡 Tips:",
-                "  • Describe the BACKGROUND, not the object being removed",
-                "  • Leave empty to auto-match surrounding textures",
-                "  • Works best with uniform, simple backgrounds"
+                "  • Describe the BACKGROUND texture, not the object",
+                "  • Leave empty to auto-match surrounding area",
+                "  • Works best with uniform backgrounds"
             ]
         ),
     }