Update scene_templates.py

scene_templates.py

@@ -1,297 +1,336 @@
 import logging
+from dataclasses import dataclass, field
 from typing import Dict, List, Optional
-from dataclasses import dataclass
 
 logger = logging.getLogger(__name__)
 
+
 @dataclass
-class SceneTemplate:
-    """Data class representing a scene template"""
+class InpaintingTemplate:
+    """Data class representing an inpainting template."""
+
     key: str
     name: str
-    prompt: str
-    negative_extra: str
     category: str
     icon: str
+    description: str
+
+    # Prompt templates
+    prompt_template: str
+    negative_prompt: str
+
+    # Recommended parameters
+    controlnet_conditioning_scale: float = 0.7
+    feather_radius: int = 8
     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
 
-class SceneTemplateManager:
-    """
-    Manages curated scene templates for background generation.
-    Provides categorized presets that users can select with one click.
+    # 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
+
+    # Difficulty level for UI display
+    difficulty: str = "medium"  # "easy", "medium", "advanced"
+
+    # Tips for users
+    usage_tips: List[str] = field(default_factory=list)
+
+
+class InpaintingTemplateManager:
     """
+    Manages inpainting templates for various use cases.
 
-    # Scene template definitions
-    TEMPLATES: Dict[str, SceneTemplate] = {
-        # Professional Category 
-        "office_modern": SceneTemplate(
-            key="office_modern",
-            name="Modern Office",
-            prompt="modern minimalist office interior, clean white desk, large floor-to-ceiling windows, natural daylight, professional corporate environment, soft shadows, contemporary furniture",
-            negative_extra="messy, cluttered, dark, old",
-            category="Professional",
-            icon="🏢",
-            guidance_scale=7.5
-        ),
-        "office_executive": SceneTemplate(
-            key="office_executive",
-            name="Executive Suite",
-            prompt="luxurious executive office, mahogany desk, leather chair, city skyline view through windows, warm ambient lighting, bookshelf, elegant professional setting",
-            negative_extra="cheap, cramped, messy",
-            category="Professional",
-            icon="👔",
-            guidance_scale=7.5
-        ),
-        "studio_white": SceneTemplate(
-            key="studio_white",
-            name="White Studio",
-            prompt="clean white photography studio background, professional lighting setup, seamless white backdrop, soft diffused light, minimal shadows",
-            negative_extra="colored, textured, dirty",
-            category="Professional",
-            icon="📷",
-            guidance_scale=8.0
-        ),
-        "coworking": SceneTemplate(
-            key="coworking",
-            name="Coworking Space",
-            prompt="modern coworking space, open plan office, plants, exposed brick, industrial chic design, natural light, collaborative environment",
-            negative_extra="empty, dark, boring",
-            category="Professional",
-            icon="💼",
-            guidance_scale=7.0
-        ),
-        "conference": SceneTemplate(
-            key="conference",
-            name="Conference Room",
-            prompt="modern conference room, large meeting table, glass walls, professional presentation screen, bright corporate lighting, clean minimal design",
-            negative_extra="small, cramped, outdated",
-            category="Professional",
-            icon="🤝",
-            guidance_scale=7.5
-        ),
+    Provides categorized presets optimized for different inpainting scenarios
+    including object replacement, removal, style transfer, and enhancement.
 
-        # Nature Category 
-        "beach_sunset": SceneTemplate(
-            key="beach_sunset",
-            name="Sunset Beach",
-            prompt="beautiful tropical beach at golden hour sunset, palm trees silhouette, calm turquoise ocean waves, warm orange and pink sky, soft sand, paradise vacation vibes",
-            negative_extra="storm, rain, crowded, trash",
-            category="Nature",
-            icon="🏖️",
-            guidance_scale=7.0
-        ),
-        "forest_enchanted": SceneTemplate(
-            key="forest_enchanted",
-            name="Enchanted Forest",
-            prompt="magical enchanted forest, sunlight streaming through tall trees, lush green foliage, mystical atmosphere, morning mist, fairy tale woodland",
-            negative_extra="dead trees, dark, scary, barren",
-            category="Nature",
-            icon="🌲",
-            guidance_scale=7.0
-        ),
-        "mountain_scenic": SceneTemplate(
-            key="mountain_scenic",
-            name="Mountain Vista",
-            prompt="breathtaking mountain landscape, snow-capped peaks, alpine meadow, clear blue sky, majestic scenic view, pristine nature, peaceful atmosphere",
-            negative_extra="industrial, polluted, crowded",
-            category="Nature",
-            icon="🏔️",
-            guidance_scale=7.5
-        ),
-        "garden_spring": SceneTemplate(
-            key="garden_spring",
-            name="Spring Garden",
-            prompt="beautiful spring flower garden, colorful blooming flowers, roses and tulips, manicured hedges, sunny day, botanical paradise, fresh and vibrant",
-            negative_extra="dead, winter, wilted, dry",
-            category="Nature",
-            icon="🌸",
-            guidance_scale=7.0
-        ),
-        "lake_serene": SceneTemplate(
-            key="lake_serene",
-            name="Serene Lake",
-            prompt="peaceful serene lake at dawn, mirror-like water reflection, surrounding mountains, soft morning light, tranquil atmosphere, pristine natural beauty",
-            negative_extra="stormy, polluted, industrial",
-            category="Nature",
-            icon="🏞️",
-            guidance_scale=7.0
-        ),
-        "cherry_blossom": SceneTemplate(
-            key="cherry_blossom",
-            name="Cherry Blossom",
-            prompt="stunning cherry blossom trees in full bloom, pink sakura petals falling gently, Japanese garden aesthetic, soft spring sunlight, romantic atmosphere",
-            negative_extra="winter, dead, brown, wilted",
-            category="Nature",
-            icon="🌸",
-            guidance_scale=7.0
-        ),
+    Attributes:
+        TEMPLATES: Dictionary of all available templates
+        CATEGORIES: List of category names in display order
 
-        # Urban Category 
-        "city_skyline": SceneTemplate(
-            key="city_skyline",
-            name="City Skyline",
-            prompt="modern city skyline at blue hour, impressive skyscrapers, glass buildings reflecting sunset, urban metropolitan view, cinematic atmosphere",
-            negative_extra="slums, dirty, abandoned, ruins",
-            category="Urban",
-            icon="🌆",
-            guidance_scale=7.5
-        ),
-        "cafe_cozy": SceneTemplate(
-            key="cafe_cozy",
-            name="Cozy Cafe",
-            prompt="warm cozy coffee shop interior, wooden furniture, ambient lighting, exposed brick walls, plants, comfortable atmosphere, artisan cafe vibes",
-            negative_extra="fast food, plastic, harsh lighting",
-            category="Urban",
-            icon="☕",
-            guidance_scale=7.0
-        ),
-        "street_european": SceneTemplate(
-            key="street_european",
-            name="European Street",
-            prompt="charming European cobblestone street, historic buildings, outdoor cafe, flowers on balconies, warm afternoon light, romantic Paris or Rome vibes",
-            negative_extra="modern, industrial, ugly, dirty",
-            category="Urban",
-            icon="🏛️",
-            guidance_scale=7.0
-        ),
-        "night_neon": SceneTemplate(
-            key="night_neon",
-            name="Neon Nightlife",
-            prompt="vibrant city nightlife scene, neon lights and signs, urban night atmosphere, colorful reflections on wet street, cyberpunk aesthetic, electric energy",
-            negative_extra="daytime, boring, plain",
-            category="Urban",
-            icon="🌃",
-            guidance_scale=6.5
-        ),
-        "rooftop_view": SceneTemplate(
-            key="rooftop_view",
-            name="Rooftop Terrace",
-            prompt="luxury rooftop terrace, city panoramic view, modern outdoor furniture, string lights, sunset golden hour, sophisticated urban oasis",
-            negative_extra="cheap, dirty, crowded",
-            category="Urban",
-            icon="🏙️",
-            guidance_scale=7.5
-        ),
+    Example:
+        >>> manager = InpaintingTemplateManager()
+        >>> template = manager.get_template("object_replacement")
+        >>> print(template.prompt_template)
+    """
 
-        # Artistic Category 
-        "gradient_soft": SceneTemplate(
-            key="gradient_soft",
-            name="Soft Gradient",
-            prompt="smooth soft gradient background, pastel colors blending beautifully, pink to blue to purple transition, dreamy aesthetic, professional portrait backdrop",
-            negative_extra="harsh, noisy, textured, busy",
-            category="Artistic",
+    TEMPLATES: Dict[str, InpaintingTemplate] = {
+        # ========================================
+        # 4 CORE TEMPLATES - Optimized for Speed & Quality
+        # ========================================
+
+        # 1. CHANGE COLOR - Pure color transformation
+        "change_color": InpaintingTemplate(
+            key="change_color",
+            name="Change Color",
+            category="Color",
             icon="🎨",
-            guidance_scale=8.0
-        ),
-        "abstract_modern": SceneTemplate(
-            key="abstract_modern",
-            name="Modern Abstract",
-            prompt="modern abstract art background, geometric shapes, bold colors, contemporary design, artistic composition, museum gallery aesthetic",
-            negative_extra="realistic, plain, boring",
-            category="Artistic",
-            icon="🖼️",
-            guidance_scale=6.5
-        ),
-        "vintage_retro": SceneTemplate(
-            key="vintage_retro",
-            name="Vintage Retro",
-            prompt="vintage retro aesthetic background, warm sepia tones, nostalgic 70s vibes, film grain texture, classic photography style, timeless elegance",
-            negative_extra="modern, digital, cold, harsh",
-            category="Artistic",
-            icon="📻",
-            guidance_scale=7.0
-        ),
-        "watercolor_dream": SceneTemplate(
-            key="watercolor_dream",
-            name="Watercolor Dream",
-            prompt="beautiful watercolor painting background, soft flowing colors, artistic brush strokes, dreamy ethereal atmosphere, delicate artistic aesthetic",
-            negative_extra="digital, sharp, photorealistic",
-            category="Artistic",
-            icon="🖌️",
-            guidance_scale=6.5
+            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, "
+                "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.3,    # Low-medium: allow color freedom in masked area
+            feather_radius=4,                     # Low: clean color boundaries
+            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
+            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: Fill the masked area with a solid, uniform color.",
+                "",
+                "📝 Example Prompts:",
+                "  • '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",
+                "  • Paint the entire area you want to recolor",
+                "  • Use modifiers: 'bright', 'dark', 'pastel', 'vivid'"
+            ]
         ),
 
-        # Seasonal Category 
-        "autumn_foliage": SceneTemplate(
-            key="autumn_foliage",
-            name="Autumn Foliage",
-            prompt="beautiful autumn scenery, vibrant fall foliage, orange red and golden leaves, maple trees, warm sunlight filtering through, cozy seasonal atmosphere",
-            negative_extra="spring, summer, green, snow",
-            category="Seasonal",
-            icon="🍂",
-            guidance_scale=7.0
-        ),
-        "winter_snow": SceneTemplate(
-            key="winter_snow",
-            name="Winter Wonderland",
-            prompt="magical winter wonderland, fresh white snow covering everything, snow-laden pine trees, soft snowfall, peaceful cold atmosphere, holiday season vibes",
-            negative_extra="summer, green, rain, mud",
-            category="Seasonal",
-            icon="❄️",
-            guidance_scale=7.0
+        # 2. CLOTHING CHANGE - Style and garment transformation
+        "clothing_change": InpaintingTemplate(
+            key="clothing_change",
+            name="Clothing Change",
+            category="Replacement",
+            icon="👕",
+            description="Change clothing style, material, or design - can include color change",
+            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, "
+                "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
+            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",
+            usage_tips=[
+                "🎯 Purpose: Replace clothing with a different style, material, or design.",
+                "",
+                "📝 Example Prompts:",
+                "  • '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 clothing type + color + details for best results",
+                "  • For suits: mention 'tailored', 'fitted', specific fabric like 'wool' or 'silk'",
+                "  • Body structure is preserved automatically"
+            ]
         ),
-        "summer_tropical": SceneTemplate(
-            key="summer_tropical",
-            name="Tropical Summer",
-            prompt="vibrant tropical summer scene, lush palm trees, bright sunny day, exotic flowers, paradise vacation destination, warm and inviting atmosphere",
-            negative_extra="winter, cold, snow, gray",
-            category="Seasonal",
-            icon="🌴",
-            guidance_scale=7.0
+
+        # 3. OBJECT REPLACEMENT - Replace one object with another
+        "object_replacement": InpaintingTemplate(
+            key="object_replacement",
+            name="Object Replacement",
+            category="Replacement",
+            icon="🔄",
+            description="Replace objects (one type at a time) - all masked areas become the SAME object",
+            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, "
+                "cartoon, anime, illustration, drawing, painted"
+            ),
+            controlnet_conditioning_scale=0.25,   # Low-medium: allows complete object replacement
+            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",
+            usage_tips=[
+                "🎯 Purpose: Replace an object with something completely different.",
+                "",
+                "📝 Example Prompts:",
+                "  • '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",
+                "  • Describe what you want, not what you're removing",
+                "  • Include material and style for realistic results"
+            ]
         ),
-        "spring_meadow": SceneTemplate(
-            key="spring_meadow",
-            name="Spring Meadow",
-            prompt="beautiful spring meadow, wildflowers blooming, fresh green grass, butterflies, soft warm sunlight, renewal and new beginnings, pastoral beauty",
-            negative_extra="winter, autumn, dead, dry",
-            category="Seasonal",
-            icon="🌷",
-            guidance_scale=7.0
+
+        # 4. REMOVAL - Remove objects and fill with background
+        "removal": InpaintingTemplate(
+            key="removal",
+            name="Remove Object",
+            category="Removal",
+            icon="🗑️",
+            description="Remove objects and naturally fill with background - describe the background material",
+            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, "
+                "cartoon, anime, illustration, drawing, painted"
+            ),
+            controlnet_conditioning_scale=0.20,   # Low: allows creative background filling
+            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",
+            usage_tips=[
+                "🎯 Purpose: Remove unwanted objects and fill with background.",
+                "",
+                "📝 Example Prompts:",
+                "  • '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 texture, not the object",
+                "  • Leave empty to auto-match surrounding area",
+                "  • Works best with uniform backgrounds"
+            ]
         ),
     }
 
+
     # Category display order
-    CATEGORIES = ["Professional", "Nature", "Urban", "Artistic", "Seasonal"]
+    CATEGORIES = ["Color", "Replacement", "Removal"]  # 4 core templates only
 
     def __init__(self):
-        """Initialize the scene template manager"""
-        logger.info(f"SceneTemplateManager initialized with {len(self.TEMPLATES)} templates")
+        """Initialize the InpaintingTemplateManager."""
+        logger.info(f"InpaintingTemplateManager initialized with {len(self.TEMPLATES)} templates")
+
+    def get_all_templates(self) -> Dict[str, InpaintingTemplate]:
+        """
+        Get all available templates.
 
-    def get_all_templates(self) -> Dict[str, SceneTemplate]:
-        """Get all available templates"""
+        Returns
+        -------
+        dict
+            Dictionary of all templates keyed by template key
+        """
         return self.TEMPLATES
 
-    def get_template(self, key: str) -> Optional[SceneTemplate]:
-        """Get a specific template by key"""
+    def get_template(self, key: str) -> Optional[InpaintingTemplate]:
+        """
+        Get a specific template by key.
+
+        Parameters
+        ----------
+        key : str
+            Template identifier
+
+        Returns
+        -------
+        InpaintingTemplate or None
+            Template if found, None otherwise
+        """
         return self.TEMPLATES.get(key)
 
-    def get_templates_by_category(self, category: str) -> List[SceneTemplate]:
-        """Get all templates in a specific category"""
+    def get_templates_by_category(self, category: str) -> List[InpaintingTemplate]:
+        """
+        Get all templates in a specific category.
+
+        Parameters
+        ----------
+        category : str
+            Category name
+
+        Returns
+        -------
+        list
+            List of templates in the category
+        """
         return [t for t in self.TEMPLATES.values() if t.category == category]
 
     def get_categories(self) -> List[str]:
-        """Get list of all categories in display order"""
+        """
+        Get list of all categories in display order.
+
+        Returns
+        -------
+        list
+            Category names
+        """
         return self.CATEGORIES
 
     def get_template_choices_sorted(self) -> List[str]:
         """
         Get template choices formatted for Gradio dropdown.
-        Returns list of display strings sorted A-Z: "🏢 Modern Office"
+
+        Returns list of display strings sorted by category then A-Z.
+        Format: "icon Name"
+
+        Returns
+        -------
+        list
+            Formatted display strings for dropdown
         """
         display_list = []
-        for key, template in self.TEMPLATES.items():
-            display_name = f"{template.icon} {template.name}"
-            display_list.append(display_name)
 
-        # Sort alphabetically by name (ignoring emoji)
-        display_list.sort(key=lambda x: x.split(' ', 1)[1] if ' ' in x else x)
+        for category in self.CATEGORIES:
+            templates = self.get_templates_by_category(category)
+            for template in sorted(templates, key=lambda t: t.name):
+                display_name = f"{template.icon} {template.name}"
+                display_list.append(display_name)
+
         return display_list
 
     def get_template_key_from_display(self, display_name: str) -> Optional[str]:
         """
         Get template key from display name.
-        Example: "🏢 Modern Office" -> "office_modern"
+
+        Parameters
+        ----------
+        display_name : str
+            Display string like "🔄 Object Replacement"
+
+        Returns
+        -------
+        str or None
+            Template key if found
         """
         if not display_name:
             return None
@@ -301,33 +340,105 @@ class SceneTemplateManager:
                 return key
         return None
 
-    def get_prompt_for_template(self, key: str) -> Optional[str]:
-        """Get the prompt string for a template"""
+    def get_parameters_for_template(self, key: str) -> Dict[str, any]:
+        """
+        Get recommended parameters for a template.
+
+        Parameters
+        ----------
+        key : str
+            Template key
+
+        Returns
+        -------
+        dict
+            Dictionary of parameter names and values
+        """
         template = self.get_template(key)
-        return template.prompt if template else None
-
-    def get_negative_prompt_for_template(
-        self,
-        key: str,
-        base_negative: str = "blurry, low quality, distorted, people, characters"
-    ) -> str:
-        """Get combined negative prompt for a template"""
+        if not template:
+            return {}
+
+        return {
+            "controlnet_conditioning_scale": template.controlnet_conditioning_scale,
+            "feather_radius": template.feather_radius,
+            "guidance_scale": template.guidance_scale,
+            "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
+        }
+
+    def build_prompt(self, key: str, content: str) -> str:
+        """
+        Build complete prompt from template and user content.
+
+        Parameters
+        ----------
+        key : str
+            Template key
+        content : str
+            User-provided content description
+
+        Returns
+        -------
+        str
+            Formatted prompt with content inserted
+        """
         template = self.get_template(key)
-        if template and template.negative_extra:
-            return f"{base_negative}, {template.negative_extra}"
-        return base_negative
+        if not template:
+            return content
+
+        return template.prompt_template.format(content=content)
+
+    def get_negative_prompt(self, key: str) -> str:
+        """
+        Get negative prompt for a template.
+
+        Parameters
+        ----------
+        key : str
+            Template key
 
-    def get_guidance_scale_for_template(self, key: str) -> float:
-        """Get the recommended guidance scale for a template"""
+        Returns
+        -------
+        str
+            Negative prompt string
+        """
         template = self.get_template(key)
-        return template.guidance_scale if template else 7.5
+        if not template:
+            return ""
+        return template.negative_prompt
+
+    def get_usage_tips(self, key: str) -> List[str]:
+        """
+        Get usage tips for a template.
+
+        Parameters
+        ----------
+        key : str
+            Template key
+
+        Returns
+        -------
+        list
+            List of tip strings
+        """
+        template = self.get_template(key)
+        if not template:
+            return []
+        return template.usage_tips
 
     def build_gallery_html(self) -> str:
         """
-        Build HTML for the scene template gallery.
-        Returns HTML string for display in Gradio.
+        Build HTML for template gallery display.
+
+        Returns
+        -------
+        str
+            HTML string for Gradio display
         """
-        html_parts = ['<div class="scene-gallery">']
+        html_parts = ['<div class="inpainting-gallery">']
 
         for category in self.CATEGORIES:
             templates = self.get_templates_by_category(category)
@@ -335,95 +446,21 @@ class SceneTemplateManager:
                 continue
 
             html_parts.append(f'''
-            <div class="scene-category">
-                <h4 class="scene-category-title">{category}</h4>
-                <div class="scene-grid">
+            <div class="inpainting-category">
+                <h4 class="inpainting-category-title">{category}</h4>
+                <div class="inpainting-grid">
             ''')
 
-            for template in templates:
+            for template in sorted(templates, key=lambda t: t.name):
                 html_parts.append(f'''
-                <button class="scene-card" data-template="{template.key}" onclick="selectTemplate('{template.key}')">
-                    <span class="scene-icon">{template.icon}</span>
-                    <span class="scene-name">{template.name}</span>
-                </button>
+                <div class="inpainting-card" data-template="{template.key}">
+                    <span class="inpainting-icon">{template.icon}</span>
+                    <span class="inpainting-name">{template.name}</span>
+                    <span class="inpainting-desc">{template.description[:50]}...</span>
+                </div>
                 ''')
 
             html_parts.append('</div></div>')
 
         html_parts.append('</div>')
-        return ''.join(html_parts)
-
-    def get_gallery_css(self) -> str:
-        """Get CSS styles for the scene gallery"""
-        return """
-        /* Scene Gallery Styles */
-        .scene-gallery {
-            margin: 16px 0;
-        }
-
-        .scene-category {
-            margin-bottom: 20px;
-        }
-
-        .scene-category-title {
-            font-size: 0.9rem;
-            font-weight: 600;
-            color: #475569;
-            margin-bottom: 12px;
-            padding-bottom: 8px;
-            border-bottom: 1px solid #e2e8f0;
-        }
-
-        .scene-grid {
-            display: grid;
-            grid-template-columns: repeat(auto-fill, minmax(100px, 1fr));
-            gap: 8px;
-        }
-
-        .scene-card {
-            display: flex;
-            flex-direction: column;
-            align-items: center;
-            justify-content: center;
-            padding: 12px 8px;
-            background: #f8fafc;
-            border: 1px solid #e2e8f0;
-            border-radius: 8px;
-            cursor: pointer;
-            transition: all 0.2s ease;
-            min-height: 70px;
-        }
-
-        .scene-card:hover {
-            background: #dbeafe;
-            border-color: #3b82f6;
-            transform: translateY(-2px);
-            box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);
-        }
-
-        .scene-card.selected {
-            background: #dbeafe;
-            border-color: #3b82f6;
-            box-shadow: 0 0 0 2px rgba(59, 130, 246, 0.3);
-        }
-
-        .scene-icon {
-            font-size: 1.5rem;
-            margin-bottom: 4px;
-        }
-
-        .scene-name {
-            font-size: 0.75rem;
-            font-weight: 500;
-            color: #1e293b;
-            text-align: center;
-            line-height: 1.2;
-        }
-
-        @media (max-width: 768px) {
-            .scene-grid {
-                grid-template-columns: repeat(3, 1fr);
-            }
-        }
-        """
-        
+        return ''.join(html_parts)