Update inpainting_templates.py

inpainting_templates.py

@@ -25,6 +25,10 @@ class InpaintingTemplate:
     guidance_scale: float = 7.5
     num_inference_steps: int = 25
 
+    # Inpainting strength (0.0-1.0)
+    # 1.0 = fully repaint masked area, 0.0 = keep original
+    strength: float = 1.0
+
     # Conditioning type preference
     preferred_conditioning: str = "canny"  # "canny" or "depth"
 
@@ -85,6 +89,7 @@ class InpaintingTemplateManager:
             feather_radius=6,                     # Low: reduces blending with original color
             guidance_scale=20.0,                  # Extremely high: forces complete color transformation
             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
             difficulty="easy",
@@ -144,6 +149,7 @@ class InpaintingTemplateManager:
             feather_radius=14,                    # Medium: natural blending with body
             guidance_scale=11.5,                  # Medium-high: accurate clothing generation
             num_inference_steps=10,               # Optimized for speed
+            strength=1.0,                         # Full repaint: completely replace clothing
             preferred_conditioning="depth",       # Depth: preserves fabric folds and body structure
             enhance_prompt=True,                  # Enabled: enriches clothing details
             difficulty="easy",
@@ -180,6 +186,7 @@ class InpaintingTemplateManager:
             feather_radius=10,                    # Medium: natural scene integration
             guidance_scale=13.0,                  # Medium-high: accurate object generation
             num_inference_steps=10,               # Optimized for speed
+            strength=1.0,                         # Full repaint: completely replace object
             preferred_conditioning="canny",       # Edge-based: preserves scene perspective
             enhance_prompt=True,                  # Enabled: enriches object details
             difficulty="medium",
@@ -218,6 +225,7 @@ class InpaintingTemplateManager:
             feather_radius=12,                    # Medium: smooth background blending
             guidance_scale=12.0,                  # Medium: balanced control and naturalness
             num_inference_steps=10,               # Optimized for speed
+            strength=1.0,                         # Full repaint: completely remove and fill
             preferred_conditioning="depth",       # Depth: preserves spatial perspective
             enhance_prompt=False,                 # Disabled: avoid generating new objects
             difficulty="medium",
@@ -367,7 +375,9 @@ class InpaintingTemplateManager:
             "feather_radius": template.feather_radius,
             "guidance_scale": template.guidance_scale,
             "num_inference_steps": template.num_inference_steps,
-            "preferred_conditioning": template.preferred_conditioning
+            "strength": template.strength,
+            "preferred_conditioning": template.preferred_conditioning,
+            "enhance_prompt": template.enhance_prompt
         }
 
     def build_prompt(self, key: str, content: str) -> str: