Upload app (2).py

app (2).py

@@ -0,0 +1,699 @@
+import os
+import gradio as gr
+import numpy as np
+import torch
+import random
+from PIL import Image, ImageEnhance, ImageFilter
+from gradio.themes import Soft
+from gradio.themes.utils import colors, fonts, sizes
+
+colors.steel_blue = colors.Color(
+    name="steel_blue",
+    c50="#EBF3F8",
+    c100="#D3E5F0",
+    c200="#A8CCE1",
+    c300="#7DB3D2",
+    c400="#529AC3",
+    c500="#4682B4",
+    c600="#3E72A0",
+    c700="#36638C",
+    c800="#2E5378",
+    c900="#264364",
+    c950="#1E3450",
+)
+
+class SteelBlueTheme(Soft):
+    def __init__(self, **kwargs):
+        super().__init__(
+            primary_hue=colors.gray,
+            secondary_hue=colors.steel_blue,
+            neutral_hue=colors.slate,
+            text_size=sizes.text_lg,
+            font=(fonts.GoogleFont("Outfit"), "Arial", "sans-serif"),
+            font_mono=(fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace"),
+        )
+
+steel_blue_theme = SteelBlueTheme()
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
+
+print("=" * 50)
+print("🎨 Enhanced Image-to-Image Photo Restoration")
+print("=" * 50)
+print("Using device:", device)
+print("=" * 50)
+
+from diffusers import (
+    AutoPipelineForImage2Image,
+    StableDiffusionXLImg2ImgPipeline,
+    StableDiffusionImg2ImgPipeline,
+    AutoencoderKL
+)
+import gc
+
+# Initialize model
+pipe = None
+model_name = None
+vae = None
+
+try:
+    print("🔄 Loading SDXL Refiner with enhanced VAE...")
+    
+    # Try to load better VAE (compatible with SDXL)
+    try:
+        print("📦 Loading high-quality VAE: madebyollin/sdxl-vae-fp16-fix")
+        vae = AutoencoderKL.from_pretrained(
+            "madebyollin/sdxl-vae-fp16-fix",
+            torch_dtype=dtype
+        )
+        print("✅ High-quality VAE loaded successfully!")
+        vae_name = "SDXL-VAE-FP16-Fix"
+    except Exception as vae_error:
+        print(f"⚠️ Could not load custom VAE: {vae_error}")
+        print("📦 Using default SDXL VAE")
+        vae = None
+        vae_name = "Default-VAE"
+    
+    # Load SDXL Refiner with custom VAE (if available)
+    if vae is not None:
+        pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+            "stabilityai/stable-diffusion-xl-refiner-1.0",
+            vae=vae,
+            torch_dtype=dtype,
+            variant="fp16" if torch.cuda.is_available() else None,
+            use_safetensors=True
+        )
+        model_name = f"SDXL-Refiner + {vae_name}"
+        print(f"✅ Loaded SDXL-Refiner with {vae_name}")
+    else:
+        pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+            "stabilityai/stable-diffusion-xl-refiner-1.0",
+            torch_dtype=dtype,
+            variant="fp16" if torch.cuda.is_available() else None,
+            use_safetensors=True
+        )
+        model_name = "SDXL-Refiner-1.0"
+        print("✅ Loaded SDXL-Refiner with default VAE")
+    
+    pipe.to(device)
+    
+    # Enable memory optimizations
+    print("⚡ Enabling optimizations...")
+    if hasattr(pipe, 'enable_attention_slicing'):
+        pipe.enable_attention_slicing(1)
+    if hasattr(pipe, 'enable_vae_slicing'):
+        pipe.enable_vae_slicing()
+    if hasattr(pipe, 'enable_vae_tiling'):
+        pipe.enable_vae_tiling()
+    
+    # Model offloading for better memory management
+    if device.type == "cpu":
+        print("💻 CPU mode - enabling model offloading")
+        if hasattr(pipe, 'enable_model_cpu_offload'):
+            pipe.enable_model_cpu_offload()
+    else:
+        print("🚀 GPU mode - enabling sequential CPU offload")
+        if hasattr(pipe, 'enable_sequential_cpu_offload'):
+            pipe.enable_sequential_cpu_offload()
+    
+    print(f"✅ Model ready: {model_name}")
+    
+except Exception as e:
+    print(f"❌ Error loading model: {e}")
+
+MAX_SEED = np.iinfo(np.int32).max
+
+# Enhanced restoration presets optimized for SDXL Refiner
+RESTORATION_PRESETS = {
+    "ลงสีภาพขาวดำ (Colorize)": {
+        "prompt": "perfectly colorized vintage photograph, realistic natural colors, vibrant hues, accurate skin tones, professional photo restoration, historical accuracy, film photography aesthetic",
+        "negative": "black and white, grayscale, monochrome, sepia, desaturated, washed out, artificial colors, oversaturated, cartoon",
+        "strength": 0.70,
+        "aesthetic_score": 7.0
+    },
+    "ซ่อมแซมเต็มรูปแบบ (Full Restore)": {
+        "prompt": "pristine restored photograph, museum quality restoration, no damage, crystal clear details, vibrant accurate colors, professional archival restoration, perfect condition",
+        "negative": "scratches, tears, stains, faded, damaged, low quality, blurry, artifacts, deteriorated, aged",
+        "strength": 0.65,
+        "aesthetic_score": 7.5
+    },
+    "ลบรอยขีดข่วน (Remove Damage)": {
+        "prompt": "clean photograph, flawless surface, no scratches, no tears, no stains, professionally restored, smooth perfect condition",
+        "negative": "scratches, damage, tears, cracks, stains, spots, defects, deterioration, wear",
+        "strength": 0.55,
+        "aesthetic_score": 6.5
+    },
+    "เพิ่มสีที่เลือนหาย (Enhance Faded)": {
+        "prompt": "vibrant saturated colors, rich deep tones, enhanced contrast, vivid hues, color-corrected, professional color grading, cinema quality",
+        "negative": "faded, washed out, pale, dull colors, desaturated, flat, low contrast, muddy colors",
+        "strength": 0.50,
+        "aesthetic_score": 7.0
+    },
+    "ลดสัญญาณรบกวน (Denoise)": {
+        "prompt": "crystal clear photograph, ultra sharp details, noise-free, clean, high resolution 8k, professional quality, perfect clarity, no grain",
+        "negative": "noisy, grainy, blurry, pixelated, low quality, compression artifacts, jpeg artifacts, soft",
+        "strength": 0.45,
+        "aesthetic_score": 6.5
+    },
+    "เพิ่มความคมชัด (Sharpen)": {
+        "prompt": "ultra sharp photograph, crisp details, high definition, tack sharp focus, professional photography, 8k resolution, perfect clarity",
+        "negative": "blurry, soft focus, out of focus, low resolution, hazy, unclear, motion blur",
+        "strength": 0.40,
+        "aesthetic_score": 6.5
+    },
+    "สไตล์โปรเฟสชันนัล (Pro Photo)": {
+        "prompt": "professional studio photography, cinematic lighting, commercial quality, advertising photography, high-end fashion photography, vogue magazine style",
+        "negative": "amateur, snapshot, casual photo, low quality, poor lighting, flat",
+        "strength": 0.60,
+        "aesthetic_score": 8.0
+    },
+    "สไตล์ภาพยนตร์ (Cinematic)": {
+        "prompt": "cinematic film still, movie quality, dramatic lighting, film grain, anamorphic lens, color graded, Hollywood production",
+        "negative": "digital, flat, video game, animated, low budget, poor quality",
+        "strength": 0.65,
+        "aesthetic_score": 7.5
+    },
+    "เปลี่ยนเป็นอนิเมะ (Anime Style)": {
+        "prompt": "beautiful anime art style, high quality manga illustration, studio ghibli quality, cel shaded, vibrant anime colors, detailed anime character",
+        "negative": "realistic, photographic, 3d render, western cartoon, ugly, bad anatomy",
+        "strength": 0.85,
+        "aesthetic_score": 7.0
+    },
+    "ภาพวาดสีน้ำ (Watercolor)": {
+        "prompt": "beautiful watercolor painting, soft edges, artistic brushwork, traditional art medium, colorful artistic interpretation, fine art quality",
+        "negative": "photographic, digital art, sharp edges, realistic, computer generated",
+        "strength": 0.80,
+        "aesthetic_score": 7.0
+    },
+    "ภาพวาดน้ำมัน (Oil Painting)": {
+        "prompt": "classical oil painting, renaissance style, visible brush strokes, artistic masterpiece, museum quality fine art, old master technique",
+        "negative": "photographic, digital, modern, flat colors, amateur, low quality",
+        "strength": 0.80,
+        "aesthetic_score": 7.5
+    },
+    "กำหนดเอง (Custom)": {
+        "prompt": "",
+        "negative": "",
+        "strength": 0.65,
+        "aesthetic_score": 6.0
+    }
+}
+
+def preprocess_image(image, enhance_quality=True):
+    """Pre-process image for better results"""
+    if enhance_quality:
+        # Auto enhance
+        enhancer = ImageEnhance.Sharpness(image)
+        image = enhancer.enhance(1.2)
+        
+        enhancer = ImageEnhance.Contrast(image)
+        image = enhancer.enhance(1.1)
+    
+    return image
+
+def restore_photo(
+    input_image,
+    preset_name,
+    custom_prompt,
+    custom_negative,
+    strength,
+    seed,
+    randomize_seed,
+    guidance_scale,
+    steps,
+    enhance_input,
+    use_aesthetic_score,
+    progress=gr.Progress(track_tqdm=True)
+):
+    """Enhanced image-to-image restoration with SDXL Refiner"""
+    if input_image is None:
+        raise gr.Error("กรุณาอัพโหลดภาพ")
+    
+    if pipe is None:
+        raise gr.Error("โมเดลยังไม่พร้อมใช้งาน")
+
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+
+    generator = torch.Generator(device=device).manual_seed(seed)
+    
+    # Get preset settings
+    preset = RESTORATION_PRESETS[preset_name]
+    
+    # Use custom prompt if provided
+    if custom_prompt.strip():
+        prompt = custom_prompt
+        negative_prompt = custom_negative if custom_negative.strip() else preset["negative"]
+        denoising_strength = strength
+        aesthetic_score = preset.get("aesthetic_score", 6.0) if use_aesthetic_score else 6.0
+    else:
+        prompt = preset["prompt"]
+        negative_prompt = preset["negative"]
+        denoising_strength = preset["strength"]
+        aesthetic_score = preset.get("aesthetic_score", 6.0) if use_aesthetic_score else 6.0
+    
+    # Pre-process image
+    original_image = input_image.convert("RGB")
+    if enhance_input:
+        original_image = preprocess_image(original_image)
+    
+    # Optimize size for processing
+    width, height = original_image.size
+    
+    # SDXL Refiner works best at 1024x1024, but we'll optimize for available hardware
+    if device.type == "cpu":
+        max_size = 768  # Higher than before, Refiner can handle it better
+    else:
+        max_size = 1024  # Native SDXL resolution
+    
+    if width > max_size or height > max_size:
+        if width > height:
+            new_width = max_size
+            new_height = int(height * (max_size / width))
+        else:
+            new_height = max_size
+            new_width = int(width * (max_size / height))
+        
+        # Make dimensions divisible by 8
+        new_width = (new_width // 8) * 8
+        new_height = (new_height // 8) * 8
+        original_image = original_image.resize((new_width, new_height), Image.LANCZOS)
+    
+    try:
+        # SDXL Refiner specific parameters
+        result = pipe(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            image=original_image,
+            strength=denoising_strength,
+            guidance_scale=guidance_scale,
+            num_inference_steps=steps,
+            generator=generator,
+            aesthetic_score=aesthetic_score,  # SDXL Refiner specific
+            negative_aesthetic_score=2.5,  # Avoid low quality outputs
+        ).images[0]
+        
+        # Clean up memory
+        if device.type == "cuda":
+            torch.cuda.empty_cache()
+        gc.collect()
+        
+        return result, seed
+    
+    except Exception as e:
+        raise gr.Error(f"เกิดข้อผิดพลาด: {str(e)}")
+
+def update_preset_settings(preset_name):
+    """Update UI when preset changes"""
+    preset = RESTORATION_PRESETS[preset_name]
+    return (
+        preset["prompt"],
+        preset["negative"],
+        preset["strength"]
+    )
+
+css="""
+#col-container {
+    margin: 0 auto;
+    max-width: 1400px;
+}
+#main-title h1 {
+    font-size: 2.5em !important; 
+    text-align: center;
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    background-clip: text;
+}
+.feature-box {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    color: white;
+    border-radius: 12px;
+    padding: 25px;
+    margin: 20px 0;
+    box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+}
+.tip-box {
+    background: #f0f7ff;
+    border-left: 4px solid #4682B4;
+    padding: 15px;
+    margin: 15px 0;
+    border-radius: 8px;
+}
+"""
+
+with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown("# 📸 Professional Photo Restoration", elem_id="main-title")
+        gr.Markdown(f"### ✨ SDXL Refiner 1.0 + Enhanced VAE • Premium Quality")
+        
+        gr.HTML(f"""
+        <div class="feature-box">
+            <h3>🚀 Professional Image-to-Image Restoration</h3>
+            <table style="width:100%; color:white; font-size:1.05em;">
+                <tr>
+                    <td style="padding:8px;">⚡ <strong>Model:</strong></td>
+                    <td>SDXL Refiner 1.0 - Premium Quality</td>
+                </tr>
+                <tr>
+                    <td style="padding:8px;">🎯 <strong>VAE:</strong></td>
+                    <td>Enhanced FP16-optimized VAE</td>
+                </tr>
+                <tr>
+                    <td style="padding:8px;">💎 <strong>Quality:</strong></td>
+                    <td>Professional restoration & enhancement</td>
+                </tr>
+                <tr>
+                    <td style="padding:8px;">🎨 <strong>Features:</strong></td>
+                    <td>Better detail preservation, accurate colors</td>
+                </tr>
+                <tr>
+                    <td style="padding:8px;">💾 <strong>Memory:</strong></td>
+                    <td>Optimized with VAE tiling and attention slicing</td>
+                </tr>
+            </table>
+            <div style="margin-top:15px; padding:10px; background:rgba(255,255,255,0.1); border-radius:8px;">
+                <strong>🔬 Enhanced VAE Benefits:</strong><br>
+                ✨ Better FP16/BF16 precision handling<br>
+                🎨 Reduced VAE artifacts<br>
+                📐 Improved detail preservation<br>
+                💎 No NaN issues on mixed precision
+            </div>
+        </div>
+        """)
+        
+        with gr.Row(equal_height=True):
+            with gr.Column(scale=1):
+                input_image = gr.Image(
+                    label="📤 อัพโหลดภาพถ่ายเก่า", 
+                    type="pil", 
+                    height=420
+                )
+                
+                gr.Markdown("### 🎯 เลือกรูปแบบการซ่อมแซม")
+                preset = gr.Dropdown(
+                    choices=list(RESTORATION_PRESETS.keys()),
+                    label="เลือกพรีเซ็ต",
+                    value="ลงสีภาพขาวดำ (Colorize)",
+                    interactive=True
+                )
+                
+                custom_prompt = gr.Textbox(
+                    label="💬 Prompt (คำสั่งบวก)",
+                    placeholder="อธิบายผลลัพธ์ที่ต้องการ...",
+                    lines=2,
+                    value=RESTORATION_PRESETS["ลงสีภาพขาวดำ (Colorize)"]["prompt"]
+                )
+                
+                custom_negative = gr.Textbox(
+                    label="🚫 Negative Prompt (สิ่งที่ไม่ต้องการ)",
+                    placeholder="อธิบายสิ่งที่ไม่ต้องการ...",
+                    lines=2,
+                    value=RESTORATION_PRESETS["ลงสีภาพขาวดำ (Colorize)"]["negative"]
+                )
+                
+                preset.change(
+                    fn=update_preset_settings,
+                    inputs=[preset],
+                    outputs=[custom_prompt, custom_negative, gr.Slider()]
+                )
+
+                run_button = gr.Button("✨ เริ่มซ่อมแซมภาพ", variant="primary", size="lg")
+
+            with gr.Column(scale=1):
+                output_image = gr.Image(
+                    label="✨ ภาพที่ซ่อมแซมแล้ว", 
+                    interactive=False, 
+                    format="png", 
+                    height=540
+                )
+                
+                with gr.Accordion("⚙️ การตั้งค่าขั้นสูง", open=True):
+                    strength = gr.Slider(
+                        label="💪 Denoising Strength",
+                        minimum=0.1,
+                        maximum=1.0,
+                        step=0.05,
+                        value=0.70,
+                        info="0.3-0.5: เปลี่ยนน้อย | 0.6-0.7: สมดุล | 0.8+: เปลี่ยนมาก"
+                    )
+                    
+                    seed = gr.Slider(
+                        label="🎲 Seed", 
+                        minimum=0, 
+                        maximum=MAX_SEED, 
+                        step=1, 
+                        value=42,
+                        info="ใช้ seed เดิมได้ผลลัพธ์เหมือนเดิม"
+                    )
+                    
+                    randomize_seed = gr.Checkbox(
+                        label="🔀 สุ่ม Seed ทุกครั้ง", 
+                        value=True
+                    )
+                    
+                    guidance_scale = gr.Slider(
+                        label="💬 Guidance Scale", 
+                        minimum=1.0, 
+                        maximum=15.0, 
+                        step=0.5, 
+                        value=7.5,
+                        info="ยิ่งสูง = ทำตามคำสั่งมากขึ้น"
+                    )
+                    
+                    steps = gr.Slider(
+                        label="🔢 Inference Steps", 
+                        minimum=10, 
+                        maximum=50, 
+                        step=5, 
+                        value=30,
+                        info="SDXL Refiner: 20-30 แนะนำ | 40-50 คุณภาพสูงสุด"
+                    )
+                    
+                    enhance_input = gr.Checkbox(
+                        label="✨ ปรับปรุงภาพอินพุตอัตโนมัติ",
+                        value=True,
+                        info="เพิ่มความคมชัดและคอนทราสต์"
+                    )
+                    
+                    use_aesthetic_score = gr.Checkbox(
+                        label="🎨 ใช้ Aesthetic Score (SDXL Refiner)",
+                        value=True,
+                        info="ปรับคุณภาพภาพให้สวยงามยิ่งขึ้น"
+                    )
+                    
+                    gr.HTML("""
+                    <div class="tip-box">
+                        <strong>💡 คำแนะนำ SDXL Refiner:</strong><br>
+                        • <strong>Strength 0.3-0.5:</strong> ซ่อมแซมเล็กน้อย<br>
+                        • <strong>Strength 0.6-0.7:</strong> ซ่อมแซมปานกลาง ✅<br>
+                        • <strong>Strength 0.8+:</strong> เปลี่ยนแปลงมาก<br><br>
+                        • <strong>Guidance 7-9:</strong> สมดุล ✅<br>
+                        • <strong>Steps 20-30:</strong> คุณภาพดี<br>
+                        • <strong>Steps 40-50:</strong> คุณภาพสูงสุด
+                    </div>
+                    """)
+        
+        # Update strength when preset changes
+        preset.change(
+            fn=lambda x: RESTORATION_PRESETS[x]["strength"],
+            inputs=[preset],
+            outputs=[strength]
+        )
+        
+        run_button.click(
+            fn=restore_photo,
+            inputs=[
+                input_image, preset, custom_prompt, custom_negative,
+                strength, seed, randomize_seed, guidance_scale, 
+                steps, enhance_input, use_aesthetic_score
+            ],
+            outputs=[output_image, seed]
+        )
+
+    gr.Markdown(f"""
+    ---
+    ### 📚 คู่มือการใช้งาน SDXL Refiner Professional
+    
+    #### 🚀 **ทำไมต้อง SDXL Refiner 1.0?**
+    
+    **จุดเด่น:**
+    - ✅ **คุณภาพสูงสุด** - ออกแบบมาเพื่อปรับปรุงและซ่อมแซมภาพโดยเฉพาะ
+    - ✅ **Aesthetic Score** - ควบคุมความสวยงามของภาพได้
+    - ✅ **รองรับ 1024px** - คุณภาพภาพสูงกว่าโมเดลอื่น
+    - ✅ **Detail Enhancement** - เพิ่มรายละเอียดได้ดีเยี่ยม
+    - ✅ **Color Accuracy** - สีสมจริงและแม่นยำ
+    - ✅ **Enhanced VAE** - ลด artifacts, ป้องกัน NaN issues
+    
+    **เมื่อเทียบกับโมเดลอื่น:**
+    - **SDXL Refiner**: คุณภาพสูงสุด, เหมาะกับงานคุณภาพ ⭐⭐⭐⭐⭐
+    - **SDXL Turbo**: เร็วที่สุด, คุณภาพดี ⭐⭐⭐⭐
+    - **SD 1.5**: เบา, คุณภาพปานกลาง ⭐⭐⭐
+    
+    #### 🎯 **วิธีใช้งาน:**
+    
+    1. **อัพโหลดภาพ** ที่ต้องการซ่อมแซม
+    2. **เลือกพรีเซ็ต** - มี 11 รูปแบบให้เลือก
+    3. **ปรับ Strength** ตามความต้องการ
+    4. **เปิด Aesthetic Score** สำหรับคุณภาพสูงสุด
+    5. **เปิด Input Enhancement** สำหรับผลลัพธ์ดีขึ้น
+    6. **กดปุ่ม "เริ่มซ่อมแซมภาพ"**
+    7. **รอ 1-5 นาที** (ขึ้นกับฮาร์ดแวร์)
+    
+    #### ⚙️ **การตั้งค่าที่แนะนำ:**
+    
+    | งาน | Strength | Guidance | Steps | Aesthetic |
+    |-----|----------|----------|-------|-----------|
+    | ลงสีภาพขาวดำ | 0.70 | 7-8 | 30 | 7.0 |
+    | ซ่อมแซมรอยขีดข่วน | 0.55 | 6-7 | 25-30 | 6.5 |
+    | เพิ่มความคมชัด | 0.40 | 5-6 | 20-25 | 6.5 |
+    | เปลี่ยนสไตล์ | 0.80-0.85 | 8-10 | 30-40 | 7.0-7.5 |
+    | คุณภาพสูงสุด | 0.65 | 7-9 | 40-50 | 7.5-8.0 |
+    
+    #### 💡 **เคล็ดลับ:**
+    
+    **สำหรับ CPU:**
+    - ใช้ Steps 20-25
+    - Guidance 6-7
+    - Strength 0.6-0.7
+    - ภาพไม่เกิน 768x768px
+    - เปิด Model CPU Offload
+    - Expected: 3-5 นาที/ภาพ
+    
+    **สำหรับ GPU (8GB+):**
+    - ใช้ Steps 30-40
+    - Guidance 7-9
+    - Strength ตามต้องการ
+    - ภาพได้ถึง 1024x1024px
+    - เปิด Aesthetic Score
+    - Expected: 45-90 วินาที/ภาพ
+    
+    **สำหรับ GPU (16GB+):**
+    - ใช้ Steps 40-50
+    - Guidance 8-10
+    - ความละเอียดสูงสุด
+    - Aesthetic Score 7.5-8.0
+    - Expected: 30-60 วินาที/ภาพ
+    
+    #### 🎨 **Aesthetic Score คืออะไร?**
+    
+    Aesthetic Score ช่วยควบคุมคุณภาพและความสวยงามของภาพที่สร้าง:
+    - **6.0-6.5**: คุณภาพมาตรฐาน
+    - **7.0-7.5**: คุณภาพสูง (แนะนำ) ✅
+    - **8.0+**: คุณภาพพรีเมี่ยม
+    
+    #### 📊 **พรีเซ็ตพิเศษ:**
+    
+    - 🎨 **ลงสีภาพขาวดำ** - แปลงภาพขาวดำเป็นสีสมจริง
+    - 🔧 **ซ่อมแซมเต็มรูปแบบ** - ลบรอยชำรุด ปรับปรุงคุณภาพ
+    - 🧹 **ลบรอยขีดข่วน** - กำจัดรอยเสี��หายต่างๆ
+    - 💎 **เพิ่มสีที่เลือนหาย** - ทำให้สีสดใสขึ้น
+    - 📐 **ลดสัญญาณรบกวน** - ลดเกรน เพิ่มความคมชัด
+    - ✨ **เพิ่มความคมชัด** - ทำให้ภาพคมชัดขึ้น
+    - 📸 **สไตล์โปรเฟสชันนัล** - แปลงเป็นภาพถ่ายสตูดิโอ
+    - 🎬 **สไตล์ภาพยนตร์** - ให้ความรู้สึกแบบหนัง
+    - 🎭 **เปลี่ยนเป็นอนิเมะ** - แปลงเป็นสไตล์อนิเมะ
+    - 🖼️ **ภาพวาดสีน้ำ** - แปลงเป็นภาพวาดสีน้ำ
+    - 🎨 **ภาพวาดน้ำมัน** - แปลงเป็นภาพวาดน้ำมัน
+    
+    #### 🔬 **Enhanced VAE:**
+    
+    **ทำไมต้องใช้ Enhanced VAE?**
+    - ✅ **FP16 Optimization**: ทำงานได้ดีกับ Mixed Precision
+    - ✅ **No NaN Issues**: ไม่เกิดปัญหา NaN values
+    - ✅ **Better Colors**: สีแม่นยำและสมจริงกว่า
+    - ✅ **Less Artifacts**: ลด compression artifacts
+    - ✅ **Stable**: เสถียรกว่า VAE มาตรฐาน
+    
+    **VAE ทำอะไร?**
+    ```
+    Input Image
+        ↓
+    VAE Encoder (ภาพ → Latent Space)
+        ↓
+    Diffusion Model (ปรับปรุง Latent)
+        ↓
+    VAE Decoder (Latent → ภาพคุณภาพสูง)
+        ↓
+    Output Image
+    ```
+    
+    #### ⚡ **ประสิทธิภาพ:**
+    
+    **SDXL Refiner Performance:**
+    - **CPU (16GB RAM):** 3-5 นาที/ภาพ (768px)
+    - **GPU (8GB VRAM):** 45-90 วินาที/ภาพ (1024px)
+    - **GPU (12GB VRAM):** 40-75 วินาที/ภาพ (1024px)
+    - **GPU (16GB+ VRAM):** 30-60 วินาที/ภาพ (1024px)
+    
+    **ขนาดโมเดล:**
+    - SDXL Refiner: ~6GB
+    - Enhanced VAE: ~335MB
+    - Total: ~6.4GB
+    
+    #### 🎯 **Use Cases:**
+    
+    ✅ **เหมาะสำหรับ:**
+    - ภาพถ่ายโบราณที่ต้องการซ่อมแซม
+    - ภาพที่ต้องการลงสีอย่างแม่นยำ
+    - ภาพที่มีรายละเอียดเยอะ
+    - งานที่ต้องการคุณภาพสูงสุด
+    - ภาพที่จะนำไปพิมพ์
+    - งาน Professional/Commercial
+    
+    ⚠️ **ไม่เหมาะกับ:**
+    - งานที่ต้องการความเร็ว (ใช้ SDXL Turbo)
+    - เครื่องที่ RAM/VRAM น้อยกว่า 8GB
+    - การทดลองแบบรวดเร็วหลายๆ รูป
+    
+    #### 💡 **Tips & Tricks:**
+    
+    **1. การเลือก Strength:**
+    - 0.3-0.5: เปลี่ยนแปลงเล็กน้อย (ลดสัญญาณรบกวน)
+    - 0.6-0.7: สมดุลดี (แนะนำสำหรับส่วนใหญ่) ✅
+    - 0.8-0.9: เปลี่ยนแปลงมาก (เปลี่ยนสไตล์)
+    
+    **2. ปรับปรุงคุณภาพ:**
+    - เปิด "Input Enhancement" เสมอ
+    - ใช้ Aesthetic Score 7.0-7.5
+    - เพิ่ม Steps สำหรับงานสำคัญ (40-50)
+    - Guidance ไม่ควรเกิน 10
+    
+    **3. ประหยัดเวลา:**
+    - ใช้ Steps 20-25 สำหรับทดลอง
+    - Resize ภาพก่อนอัพโหลด
+    - ลด Guidance เป็น 6-7
+    - ปิด Aesthetic Score ถ้าไม่จำเป็น
+    
+    **4. แก้ปัญหา:**
+    - **ภาพเบลอ**: เพิ่ม Steps, ลด Strength
+    - **สีไม่ถูก**: ปรับ Guidance, Aesthetic Score
+    - **Detail หาย**: ลด Strength เหลือ 0.5-0.6
+    - **Out of Memory**: ลดขนาดภาพ, เปิด VAE tiling
+    - **ใช้เวลานาน**: ลด Steps, ลด resolution
+    
+    #### 🔄 **Comparison Table:**
+    
+    | Feature | SDXL Refiner | SDXL Turbo | SD 1.5 |
+    |---------|--------------|------------|--------|
+    | Quality | ⭐⭐⭐⭐⭐ | ⭐⭐⭐⭐ | ⭐⭐⭐ |
+    | Speed | ⭐⭐⭐ | ⭐⭐⭐⭐⭐ | ⭐⭐⭐⭐ |
+    | Memory | ⭐⭐ | ⭐⭐⭐ | ⭐⭐⭐⭐ |
+    | Resolution | 1024px | 768px | 512px |
+    | Best For | Quality | Speed | Low-end |
+    
+    ---
+    
+    <div style="text-align:center; color:#666; padding:20px;">
+        <strong>🌟 SDXL Refiner 1.0 Professional</strong><br>
+        Premium Photo Restoration System<br>
+        <em>Highest Quality • Enhanced VAE • Aesthetic Control</em><br><br>
+        <div style="font-size:0.9em; margin-top:10px;">
+            <strong>Powered by:</strong><br>
+            Stability AI (SDXL Refiner) + Enhanced VAE
+        </div>
+    </div>
+    """)
+
+demo.launch(server_name="0.0.0.0", server_port=7860, share=False)