Update scene_templates.py

scene_templates.py

@@ -1,336 +1,297 @@
 import logging
-from dataclasses import dataclass, field
 from typing import Dict, List, Optional
+from dataclasses import dataclass
 
 logger = logging.getLogger(__name__)
 
-
 @dataclass
-class InpaintingTemplate:
-    """Data class representing an inpainting template."""
-
+class SceneTemplate:
+    """Data class representing a scene 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
-
-    # 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:
+class SceneTemplateManager:
     """
-    Manages inpainting templates for various use cases.
-
-    Provides categorized presets optimized for different inpainting scenarios
-    including object replacement, removal, style transfer, and enhancement.
-
-    Attributes:
-        TEMPLATES: Dictionary of all available templates
-        CATEGORIES: List of category names in display order
-
-    Example:
-        >>> manager = InpaintingTemplateManager()
-        >>> template = manager.get_template("object_replacement")
-        >>> print(template.prompt_template)
+    Manages curated scene templates for background generation.
+    Provides categorized presets that users can select with one click.
     """
 
-    TEMPLATES: Dict[str, InpaintingTemplate] = {
-        # ========================================
-        # 4 CORE TEMPLATES - Optimized for Speed & Quality
-        # ========================================
+    # 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
+        ),
 
-        # 1. CHANGE COLOR - Pure color transformation
-        "change_color": InpaintingTemplate(
-            key="change_color",
-            name="Change Color",
-            category="Color",
-            icon="🎨",
-            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'"
-            ]
+        # 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
         ),
 
-        # 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"
-            ]
+        # 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
         ),
 
-        # 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"
-            ]
+        # 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",
+            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
         ),
 
-        # 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"
-            ]
+        # 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
+        ),
+        "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
+        ),
+        "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
         ),
     }
 
-
     # Category display order
-    CATEGORIES = ["Color", "Replacement", "Removal"]  # 4 core templates only
+    CATEGORIES = ["Professional", "Nature", "Urban", "Artistic", "Seasonal"]
 
     def __init__(self):
-        """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.
+        """Initialize the scene template manager"""
+        logger.info(f"SceneTemplateManager initialized with {len(self.TEMPLATES)} templates")
 
-        Returns
-        -------
-        dict
-            Dictionary of all templates keyed by template key
-        """
+    def get_all_templates(self) -> Dict[str, SceneTemplate]:
+        """Get all available templates"""
         return self.TEMPLATES
 
-    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
-        """
+    def get_template(self, key: str) -> Optional[SceneTemplate]:
+        """Get a specific template by key"""
         return self.TEMPLATES.get(key)
 
-    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
-        """
+    def get_templates_by_category(self, category: str) -> List[SceneTemplate]:
+        """Get all templates in a specific 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.
-
-        Returns
-        -------
-        list
-            Category names
-        """
+        """Get list of all categories in display order"""
         return self.CATEGORIES
 
     def get_template_choices_sorted(self) -> List[str]:
         """
         Get template choices formatted for Gradio dropdown.
-
-        Returns list of display strings sorted by category then A-Z.
-        Format: "icon Name"
-
-        Returns
-        -------
-        list
-            Formatted display strings for dropdown
+        Returns list of display strings sorted A-Z: "🏢 Modern Office"
         """
         display_list = []
+        for key, template in self.TEMPLATES.items():
+            display_name = f"{template.icon} {template.name}"
+            display_list.append(display_name)
 
-        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)
-
+        # Sort alphabetically by name (ignoring emoji)
+        display_list.sort(key=lambda x: x.split(' ', 1)[1] if ' ' in x else x)
         return display_list
 
     def get_template_key_from_display(self, display_name: str) -> Optional[str]:
         """
         Get template key from display name.
-
-        Parameters
-        ----------
-        display_name : str
-            Display string like "🔄 Object Replacement"
-
-        Returns
-        -------
-        str or None
-            Template key if found
+        Example: "🏢 Modern Office" -> "office_modern"
         """
         if not display_name:
             return None
@@ -340,105 +301,33 @@ class InpaintingTemplateManager:
                 return key
         return None
 
-    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)
-        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
-        """
+    def get_prompt_for_template(self, key: str) -> Optional[str]:
+        """Get the prompt string for a template"""
         template = self.get_template(key)
-        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
-
-        Returns
-        -------
-        str
-            Negative prompt string
-        """
+        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"""
         template = self.get_template(key)
-        if not template:
-            return ""
-        return template.negative_prompt
+        if template and template.negative_extra:
+            return f"{base_negative}, {template.negative_extra}"
+        return base_negative
 
-    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
-        """
+    def get_guidance_scale_for_template(self, key: str) -> float:
+        """Get the recommended guidance scale for a template"""
         template = self.get_template(key)
-        if not template:
-            return []
-        return template.usage_tips
+        return template.guidance_scale if template else 7.5
 
     def build_gallery_html(self) -> str:
         """
-        Build HTML for template gallery display.
-
-        Returns
-        -------
-        str
-            HTML string for Gradio display
+        Build HTML for the scene template gallery.
+        Returns HTML string for display in Gradio.
         """
-        html_parts = ['<div class="inpainting-gallery">']
+        html_parts = ['<div class="scene-gallery">']
 
         for category in self.CATEGORIES:
             templates = self.get_templates_by_category(category)
@@ -446,21 +335,94 @@ class InpaintingTemplateManager:
                 continue
 
             html_parts.append(f'''
-            <div class="inpainting-category">
-                <h4 class="inpainting-category-title">{category}</h4>
-                <div class="inpainting-grid">
+            <div class="scene-category">
+                <h4 class="scene-category-title">{category}</h4>
+                <div class="scene-grid">
             ''')
 
-            for template in sorted(templates, key=lambda t: t.name):
+            for template in templates:
                 html_parts.append(f'''
-                <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>
+                <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>
                 ''')
 
             html_parts.append('</div></div>')
 
         html_parts.append('</div>')
-        return ''.join(html_parts)
+        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);
+            }
+        }
+        """