Update scene_templates.py

scene_templates.py

@@ -1,620 +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
-
-    # Conditioning type preference
-    preferred_conditioning: str = "canny"  # "canny" or "depth"
-
-    # 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] = {
-        # Object Replacement Category
-        "object_replacement": InpaintingTemplate(
-            key="object_replacement",
-            name="Object Replacement",
-            category="Replacement",
-            icon="🔄",
-            description="Replace selected objects with new content while preserving context",
-            prompt_template="{content}, seamlessly integrated into scene, matching lighting and perspective, realistic placement",
-            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"
-            ),
-            controlnet_conditioning_scale=0.32,  # Optimized: Lower for clean replacement (was 0.38)
-            feather_radius=10,                    # Optimized: Tighter blending (was 14)
-            guidance_scale=11.0,                  # Optimized: Higher for object accuracy (was 10.0)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",       # ✅ KEPT: Edge-based replacement
-            difficulty="medium",
-            usage_tips=[
-                "Draw mask PRECISELY around the object to replace (include small margin)",
-                "Low conditioning (0.32) helps remove original object completely",
-                "Be specific: Instead of 'plant', use 'large green potted fern with detailed leaves'",
-                "📝 Example prompts:",
-                "  • 'elegant white ceramic teacup with delicate gold rim and floral pattern'",
-                "  • 'modern silver laptop computer with sleek metallic finish'",
-                "  • 'vintage wooden desk lamp with warm brass details'"
-            ]
+    # 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
         ),
-
-        "face_swap": InpaintingTemplate(
-            key="face_swap",
-            name="Face Enhancement",
-            category="Replacement",
-            icon="👤",
-            description="Enhance or modify facial features - SUBTLE adjustments only (skin, expression, lighting)",
-            prompt_template="{content}, natural skin texture, proper facial proportions, realistic lighting, detailed facial features",
-            negative_prompt=(
-                "deformed face, asymmetric features, unnatural skin, "
-                "plastic appearance, wrong eye direction, blurry features, "
-                "artificial smoothing, uncanny valley, distorted proportions"
-            ),
-            controlnet_conditioning_scale=0.88,  # Very high - preserves facial structure
-            feather_radius=6,
-            guidance_scale=8.5,
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",
-            difficulty="advanced",                # ⚠️ Advanced: Faces are sensitive
-            usage_tips=[
-                "⚠️ ADVANCED: Best for SUBTLE enhancements (skin tone, blemishes, expression)",
-                "NOT for major changes (face shape, eye size, feature positions)",
-                "Draw mask CAREFULLY around face outline (avoid hair and neck)",
-                "High conditioning (0.88) preserves facial structure - only small modifications",
-                "📝 Example prompts:",
-                "  • 'warm friendly smile with bright eyes and natural expression'",
-                "  • 'professional headshot with confident neutral expression, clear smooth skin'",
-                "  • 'gentle smile with soft natural lighting on face'"
-            ]
+        "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
         ),
-
-        "clothing_change": InpaintingTemplate(
-            key="clothing_change",
-            name="Clothing Change",
-            category="Replacement",
-            icon="👕",
-            description="Change clothing color, pattern, style - preserves fabric folds and texture",
-            prompt_template="CHANGE to {content}, vivid saturated color, natural fabric texture, proper folds and wrinkles, correct fit",
-            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"
-            ),
-            controlnet_conditioning_scale=0.30,  # Optimized: Much lower for color change freedom (was 0.42)
-            feather_radius=14,
-            guidance_scale=11.5,                  # Optimized: Higher for color accuracy (was 10.5)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="depth",       # ✅ KEPT: Preserves fabric folds and texture
-            difficulty="easy",                    # ✅ Easy: Good success rate
-            usage_tips=[
-                "✅ BALANCED: Low conditioning (0.30) allows color change while depth preserves folds",
-                "Works for both moderate and dramatic color changes (black→red now possible!)",
-                "For extreme changes, be VERY specific with color: 'BRIGHT red', 'PURE white'",
-                "📝 Example prompts:",
-                "  • 'BRIGHT red polo shirt with vivid saturated color, clean collar and soft fabric'",
-                "  • 'deep navy blue formal blazer with fine texture and professional fit'",
-                "  • 'PURE white dress shirt with crisp clean fabric'"
-            ]
+        "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
         ),
-
-        "dramatic_color_change": InpaintingTemplate(
-            key="dramatic_color_change",
-            name="Dramatic Color Change",
-            category="Replacement",
-            icon="🎨",
-            description="Extreme color transformations (dark↔light, black↔white) - especially for illustrations",
-            prompt_template="COMPLETELY SOLID {content}, entire area pure color, all fabric same vivid color, no original color remaining",
-            negative_prompt=(
-                "original color, keeping same shade, partial change, color bleeding, "
-                "faded colors, mixed tones, subtle change, gradual transition, "
-                "original appearance, unchanged, dark remnants, light patches, "
-                "black clothing, dark colors, keeping original darkness, "
-                "black fabric, keeping black, original black color, dark body, "
-                "partial color change, colored trim only, colored edges only, colored outline only, "
-                "black body with colored decoration, original color body"
-            ),
-            controlnet_conditioning_scale=0.15,  # Optimized: Extremely low for anime/illustration (was 0.25)
-            feather_radius=16,
-            guidance_scale=15.5,                  # Optimized: Very high for strong color adherence (was 13.5)
-            num_inference_steps=10,               # Optimized: 10 steps sufficient (was 12)
-            preferred_conditioning="canny",       # Optimized: Canny for color freedom (was depth)
-            difficulty="medium",                  # ⚠️ Medium: Extreme changes challenging
-            usage_tips=[
-                "✅ OPTIMIZED for anime/illustration extreme color changes (black↔white, black↔red)",
-                "Use COMPLETE descriptors: 'solid PURE WHITE', 'completely BRIGHT RED', 'entire garment JET BLACK'",
-                "Very low conditioning (0.15) allows complete color transformation on flat-color illustrations",
-                "Add 'completely', 'solid', 'entire', 'pure' to ensure full coverage",
-                "📝 Example prompts:",
-                "  • 'solid PURE WHITE dress shirt, completely white fabric'",
-                "  • 'completely JET BLACK leather jacket, entire garment black'",
-                "  • 'solid BRIGHT RED polo shirt, all fabric vivid red, no black'"
-            ]
+        "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
         ),
-
-        "clothing_addition": InpaintingTemplate(
-            key="clothing_addition",
-            name="Add Accessories",
-            category="Replacement",
-            icon="👔",
-            description="⚡ EXPERIMENTAL: Add ties, pockets, buttons to clothing - challenging, may require multiple attempts",
-            prompt_template="{content}, clearly visible, highly detailed accessory, seamlessly integrated into clothing, proper placement and perspective",
-            negative_prompt=(
-                "missing details, incomplete, floating objects, disconnected, "
-                "unrealistic placement, wrong perspective, blurry, poorly integrated, "
-                "invisible, faint, unclear, hidden, absent, not visible"
-            ),
-            controlnet_conditioning_scale=0.20,  # Optimized: Extremely low for object generation (was 0.25)
-            feather_radius=12,
-            guidance_scale=15.0,                  # Optimized: Very high for strong prompt adherence (was 14.0)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",       # Optimized: Canny for maximum freedom (was depth)
-            difficulty="advanced",                # ⚠️⚠️ ADVANCED: Low success rate, experimental
-            usage_tips=[
-                "⚡⚡ EXPERIMENTAL FEATURE: Adding new objects is very challenging",
-                "⚠️ Lower success rate due to complexity - may require multiple attempts",
-                "Draw mask from NECK to CHEST (vertical strip) for ties, not just collar area",
-                "Very low conditioning (0.20) gives model maximum freedom to generate new objects",
-                "📝 Example prompts:",
-                "  • 'burgundy silk necktie with diagonal stripes and Windsor knot, hanging down from collar to chest'",
-                "  • 'white pocket square with neat fold, visible in breast pocket'",
-                "  • 'silver lapel pin with detailed engraving on left collar'",
-                "💡 TIP: For ties, mask should cover the entire length where tie should appear"
-            ]
+        "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
         ),
 
-        # Object Removal Category
-        "object_removal": InpaintingTemplate(
-            key="object_removal",
-            name="Object Removal",
-            category="Removal",
-            icon="🗑️",
-            description="Remove unwanted objects and fill with matching background",
-            prompt_template="clean background, seamless continuation, {content}",
-            negative_prompt=(
-                "visible patches, color mismatch, texture inconsistency, "
-                "ghost artifacts, blur spots, repeated patterns, visible seams, remnants"
-            ),
-            controlnet_conditioning_scale=0.42,  # Optimized: Lower for cleaner removal (was 0.48)
-            feather_radius=14,
-            guidance_scale=8.5,                   # Optimized: Higher for accuracy (was 8.0)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",       # ✅ KEPT: Edge-based removal
-            difficulty="easy",                    # ✅ Easy: Inpainting core strength
-            usage_tips=[
-                "Draw mask slightly BEYOND object edges for better blending",
-                "Lower conditioning (0.42) ensures complete removal without traces",
-                "Describe what background SHOULD look like (e.g., 'grass lawn', 'wooden floor')",
-                "📝 Example prompts:",
-                "  • 'clean grass lawn with natural green color'",
-                "  • 'smooth wooden floor with consistent grain pattern'",
-                "  • 'plain white wall with even texture'"
-            ]
+        # 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
         ),
-
-        "watermark_removal": InpaintingTemplate(
-            key="watermark_removal",
-            name="Watermark Removal",
-            category="Removal",
-            icon="💧",
-            description="Remove watermarks and text overlays",
-            prompt_template="clean image, no text, seamless background, {content}",
-            negative_prompt=(
-                "text, watermark, logo, signature, letters, numbers, visible artifacts, "
-                "color inconsistency, blur, remnants, ghost text"
-            ),
-            controlnet_conditioning_scale=0.40,  # Optimized: Lower for complete removal (was 0.45)
-            feather_radius=12,
-            guidance_scale=9.0,                   # Optimized: Higher for clean result (was 8.5)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",       # ✅ KEPT: Edge-based removal
-            difficulty="easy",                    # ✅ Easy: Good success rate
-            usage_tips=[
-                "Draw mask covering ALL watermark/text areas precisely",
-                "Lower conditioning (0.40) ensures watermark disappears completely",
-                "Describe what SHOULD be there instead (e.g., 'sky', 'fabric texture')",
-                "📝 Example prompts:",
-                "  • 'clean blue sky with smooth gradient'",
-                "  • 'natural skin texture without marks'",
-                "  • 'smooth fabric surface with consistent color'"
-            ]
+        "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
         ),
 
-        "blemish_removal": InpaintingTemplate(
-            key="blemish_removal",
-            name="Blemish Removal",
-            category="Removal",
-            icon="✨",
-            description="Remove skin blemishes, scratches, or small imperfections",
-            prompt_template="clean smooth surface, natural texture, {content}",
-            negative_prompt=(
-                "artificial smoothing, plastic texture, visible editing, "
-                "color patches, unnatural appearance, over-processed"
-            ),
-            controlnet_conditioning_scale=0.6,
-            feather_radius=6,
-            guidance_scale=6.5,
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",
-            difficulty="easy",                    # ✅ Easy: Small areas, high success
-            usage_tips=[
-                "Draw small precise masks for EACH blemish/imperfection",
-                "Lower guidance (6.5) preserves natural skin texture",
-                "📝 Example prompts:",
-                "  • 'natural clean skin with smooth texture'",
-                "  • 'smooth surface without scratches or marks'",
-                "  • 'clear skin with natural pores and texture'"
-            ]
+        # 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
         ),
 
-        # Style Transfer Category
-        "style_artistic": InpaintingTemplate(
-            key="style_artistic",
-            name="Artistic Style",
-            category="Style",
+        # 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="🎨",
-            description="Apply artistic style to selected region",
-            prompt_template="{content}, distinctive artistic style, strong painterly effect, creative interpretation, visible brushstrokes",
-            negative_prompt=(
-                "photorealistic, plain, boring, low contrast, unchanged, "
-                "inconsistent style, harsh transitions, original appearance, realistic photo"
-            ),
-            controlnet_conditioning_scale=0.42,  # Optimized: Lower for style freedom (was 0.52)
-            feather_radius=12,
-            guidance_scale=12.5,                  # Optimized: Higher for style accuracy (was 11.5)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",       # ✅ KEPT: Edge-based styling
-            difficulty="medium",                  # ⚠️ Medium: Style changes need steps
-            usage_tips=[
-                "Works best on larger areas (faces, clothing, backgrounds) for visible transformation",
-                "Lower conditioning (0.42) allows artistic reinterpretation",
-                "Be VERY specific about art style for best results",
-                "📝 Example prompts:",
-                "  • 'impressionist oil painting with visible thick brushstrokes and vibrant colors'",
-                "  • 'watercolor painting with soft edges and delicate color washes'",
-                "  • 'Van Gogh style with swirling brushstrokes and bold color contrasts'"
-            ]
+            guidance_scale=8.0
         ),
-
-        "style_vintage": InpaintingTemplate(
-            key="style_vintage",
-            name="Vintage Look",
-            category="Style",
-            icon="📻",
-            description="Apply vintage or retro aesthetic to selected area",
-            prompt_template="{content}, strong vintage aesthetic, warm sepia tones, film grain texture, nostalgic atmosphere",
-            negative_prompt=(
-                "modern, digital, cold colors, harsh contrast, "
-                "oversaturated, neon colors, contemporary look, clean digital, crisp"
-            ),
-            controlnet_conditioning_scale=0.45,  # Optimized: Lower for style freedom (was 0.55)
-            feather_radius=14,
-            guidance_scale=11.0,                  # Optimized: Higher for era accuracy (was 10.5)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="depth",       # Optimized: Depth for texture/grain (was canny)
-            difficulty="medium",                  # ⚠️ Medium: Color/texture adjustments
-            usage_tips=[
-                "Works best on medium to large regions for visible aesthetic change",
-                "Depth conditioning preserves texture while applying vintage look",
-                "Specify era and style for best results",
-                "📝 Example prompts:",
-                "  • '1920s sepia photograph with faded brown tones and soft grain'",
-                "  • '1970s vintage photo with warm orange tones and slight film grain'",
-                "  • '1950s Kodachrome with saturated warm colors and nostalgic feel'"
-            ]
+        "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
         ),
-
-        "style_anime": InpaintingTemplate(
-            key="style_anime",
-            name="Anime Style",
-            category="Style",
-            icon="🎌",
-            description="⚡ Transform to anime style - dramatic change, may need multiple attempts",
-            prompt_template="{content}, anime illustration style, clean sharp lines, vibrant saturated colors, cel-shaded with flat colors",
-            negative_prompt=(
-                "photorealistic, blurry lines, muddy colors, realistic photo, "
-                "3D render, uncanny valley, western cartoon, gradient shading, photographic"
-            ),
-            controlnet_conditioning_scale=0.35,  # Optimized: Lower for dramatic style change (was 0.48)
-            feather_radius=10,
-            guidance_scale=14.0,                  # Optimized: Higher for style accuracy (was 12.5)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",
-            difficulty="advanced",                # ⚠️ Advanced: Extreme style transformation
-            usage_tips=[
-                "⚡ DRAMATIC transformation - best for portraits and characters",
-                "⚠️ Significant stylistic changes require good source image quality",
-                "Lower conditioning (0.35) allows dramatic style transformation",
-                "Expect anime-style interpretation, not exact anime conversion",
-                "📝 Example prompts:",
-                "  • 'modern anime style with large expressive eyes and vibrant colors'",
-                "  • 'Studio Ghibli style with soft features and warm color palette'",
-                "  • 'manga style with clean black lines and cel-shaded coloring'"
-            ]
+        "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
         ),
-
-        # Detail Enhancement Category
-        "detail_enhance": InpaintingTemplate(
-            key="detail_enhance",
-            name="Detail Enhancement",
-            category="Enhancement",
-            icon="🔍",
-            description="Add fine details and textures to selected area",
-            prompt_template="{content}, highly detailed, intricate textures, fine details, sharp focus",
-            negative_prompt=(
-                "blurry, smooth, low detail, soft focus, "
-                "oversimplified, lacking texture"
-            ),
-            controlnet_conditioning_scale=0.85,  # Very high - preserves structure
-            feather_radius=4,
-            guidance_scale=8.0,
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="depth",
-            difficulty="easy",                    # ✅ Easy: Enhancement not generation
-            usage_tips=[
-                "High conditioning (0.85) preserves overall structure while adding detail",
-                "Best for adding fine details to existing objects",
-                "📝 Example prompts:",
-                "  • 'highly detailed fabric with visible weave and fine threads'",
-                "  • 'intricate wood grain with natural knots and detailed texture'",
-                "  • 'sharp facial features with fine skin pores and detail'"
-            ]
+        "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
         ),
 
-        "texture_add": InpaintingTemplate(
-            key="texture_add",
-            name="Texture Addition",
-            category="Enhancement",
-            icon="🧱",
-            description="Add or enhance surface textures",
-            prompt_template="{content} texture, realistic surface detail, natural material appearance",
-            negative_prompt=(
-                "flat, smooth, unrealistic, plastic, "
-                "wrong material, inconsistent texture"
-            ),
-            controlnet_conditioning_scale=0.8,   # High - preserves shape
-            feather_radius=5,
-            guidance_scale=7.5,
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="depth",
-            difficulty="easy",                    # ✅ Easy: Texture overlay
-            usage_tips=[
-                "Specify material type clearly for best results",
-                "Depth conditioning preserves 3D form while changing texture",
-                "📝 Example prompts:",
-                "  • 'rough wood texture with natural grain and knots'",
-                "  • 'soft cotton fabric with gentle weave pattern'",
-                "  • 'smooth marble surface with subtle veining'"
-            ]
+        # 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
         ),
-
-        "lighting_fix": InpaintingTemplate(
-            key="lighting_fix",
-            name="Lighting Correction",
-            category="Enhancement",
-            icon="💡",
-            description="Correct or enhance lighting in selected area",
-            prompt_template="{content}, proper lighting, natural shadows, balanced exposure",
-            negative_prompt=(
-                "harsh shadows, overexposed, underexposed, "
-                "flat lighting, unnatural highlights"
-            ),
-            controlnet_conditioning_scale=0.65,
-            feather_radius=15,                    # Large for smooth transitions
-            guidance_scale=7.0,
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="depth",
-            difficulty="easy",                    # ✅ Easy: Lighting adjustments
-            usage_tips=[
-                "Use large feather (15px) for smooth lighting transitions",
-                "Best for fixing uneven lighting or adding natural light",
-                "📝 Example prompts:",
-                "  • 'soft natural lighting from window, gentle shadows'",
-                "  • 'balanced exposure with warm golden hour light'",
-                "  • 'even studio lighting with soft diffused shadows'"
-            ]
+        "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
         ),
-
-        # Background Category
-        "background_extend": InpaintingTemplate(
-            key="background_extend",
-            name="Background Extension",
-            category="Background",
-            icon="📐",
-            description="Extend image background seamlessly",
-            prompt_template="seamless background extension, {content}, consistent style and lighting",
-            negative_prompt=(
-                "visible seams, style mismatch, lighting inconsistency, "
-                "repeated elements, unnatural continuation, abrupt changes"
-            ),
-            controlnet_conditioning_scale=0.45,  # Optimized: Lower for extension freedom (was 0.55)
-            feather_radius=20,                    # Largest for smooth blending
-            guidance_scale=8.5,                   # Optimized: Higher for consistency (was 8.0)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="canny",       # ✅ KEPT: Edge-based extension
-            difficulty="medium",                  # ⚠️ Medium: Needs consistency
-            usage_tips=[
-                "Draw mask on area to extend (edges of image)",
-                "Large feather (20px) ensures smooth blending with existing background",
-                "Canny conditioning focuses on edges for seamless continuation",
-                "📝 Example prompts:",
-                "  • 'continue the wooden floor with same grain pattern'",
-                "  • 'extend blue sky with matching clouds and lighting'",
-                "  • 'seamless continuation of brick wall texture'"
-            ]
+        "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
         ),
-
-        "background_replace": InpaintingTemplate(
-            key="background_replace",
-            name="Background Replacement",
-            category="Background",
-            icon="🖼️",
-            description="Replace background while keeping subject intact",
-            prompt_template="{content}, professional background scene, seamless integration with subject, matching lighting and atmosphere",
-            negative_prompt=(
-                "floating subject, inconsistent lighting, disconnected, "
-                "wrong perspective, visible edges, color mismatch, original background, "
-                "poor integration, obvious composite"
-            ),
-            controlnet_conditioning_scale=0.48,  # Optimized: Lower for background freedom (was 0.60)
-            feather_radius=12,
-            guidance_scale=10.5,                  # Optimized: Higher for scene accuracy (was 9.5)
-            num_inference_steps=10,               # Optimized: 10 steps (was 12)
-            preferred_conditioning="depth",       # ✅ KEPT: Preserves depth relationships
-            difficulty="medium",                  # ⚠️ Medium: Large area replacement
-            usage_tips=[
-                "Draw mask around ENTIRE background (leave subject unmasked with small margin)",
-                "Depth conditioning (0.48) preserves subject-background spatial relationship",
-                "Include lighting description to match subject for natural results",
-                "📝 Example prompts:",
-                "  • 'professional photography studio with white backdrop and soft lighting'",
-                "  • 'modern minimalist office with white walls and bright natural lighting'",
-                "  • 'sunny beach with blue ocean and golden hour lighting'"
-            ]
+        "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 = ["Replacement", "Removal", "Style", "Enhancement", "Background"]
+    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
@@ -624,102 +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,
-            "preferred_conditioning": template.preferred_conditioning
-        }
-
-    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)
@@ -727,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);
+            }
+        }
+        """