Delete app.py

app.py

@@ -1,578 +0,0 @@
-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("🎨 LongCat Image-Edit Photo Restoration")
-print("=" * 50)
-print("Using device:", device)
-print("=" * 50)
-
-from diffusers import StableDiffusionXLImg2ImgPipeline
-import gc
-
-# Initialize model
-pipe = None
-model_name = None
-
-try:
-    print("🔄 Loading LongCat-Image-Edit model...")
-    
-    # Load LongCat Image Edit model
-    pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
-        "meituan-longcat/LongCat-Image-Edit",
-        torch_dtype=dtype,
-        variant="fp16" if torch.cuda.is_available() else None,
-        use_safetensors=True
-    )
-    
-    model_name = "LongCat-Image-Edit"
-    print(f"✅ Loaded {model_name}")
-    
-    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 LongCat-Image-Edit
-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.75
-    },
-    "ซ่อมแซมเต็มรูปแบบ (Full Restore)": {
-        "prompt": "pristine restored photograph, museum quality restoration, no damage, crystal clear details, vibrant accurate colors, professional archival restoration, perfect condition, high quality",
-        "negative": "scratches, tears, stains, faded, damaged, low quality, blurry, artifacts, deteriorated, aged, noise",
-        "strength": 0.70
-    },
-    "ลบรอยขีดข่วน (Remove Damage)": {
-        "prompt": "clean photograph, flawless surface, no scratches, no tears, no stains, professionally restored, smooth perfect condition, pristine",
-        "negative": "scratches, damage, tears, cracks, stains, spots, defects, deterioration, wear, marks",
-        "strength": 0.60
-    },
-    "เพิ่มสีที่เลือนหาย (Enhance Faded)": {
-        "prompt": "vibrant saturated colors, rich deep tones, enhanced contrast, vivid hues, color-corrected, professional color grading, cinema quality, perfect exposure",
-        "negative": "faded, washed out, pale, dull colors, desaturated, flat, low contrast, muddy colors, underexposed",
-        "strength": 0.55
-    },
-    "ลดสัญญาณรบกวน (Denoise)": {
-        "prompt": "crystal clear photograph, ultra sharp details, noise-free, clean, high resolution 8k, professional quality, perfect clarity, no grain, smooth",
-        "negative": "noisy, grainy, blurry, pixelated, low quality, compression artifacts, jpeg artifacts, soft, unclear",
-        "strength": 0.50
-    },
-    "เพิ่มความคมชัด (Sharpen)": {
-        "prompt": "ultra sharp photograph, crisp details, high definition, tack sharp focus, professional photography, 8k resolution, perfect clarity, detailed",
-        "negative": "blurry, soft focus, out of focus, low resolution, hazy, unclear, motion blur, fuzzy",
-        "strength": 0.45
-    },
-    "สไตล์โปรเฟสชันนัล (Pro Photo)": {
-        "prompt": "professional studio photography, cinematic lighting, commercial quality, advertising photography, high-end fashion photography, vogue magazine style, perfect composition",
-        "negative": "amateur, snapshot, casual photo, low quality, poor lighting, flat, unprofessional",
-        "strength": 0.65
-    },
-    "สไตล์ภาพยนตร์ (Cinematic)": {
-        "prompt": "cinematic film still, movie quality, dramatic lighting, film grain aesthetic, anamorphic lens, color graded, Hollywood production, cinematic look",
-        "negative": "digital, flat, video game, animated, low budget, poor quality, amateur",
-        "strength": 0.70
-    },
-    "เปลี่ยนเป็นอนิเมะ (Anime Style)": {
-        "prompt": "beautiful anime art style, high quality manga illustration, studio ghibli quality, cel shaded, vibrant anime colors, detailed anime character, professional anime",
-        "negative": "realistic, photographic, 3d render, western cartoon, ugly, bad anatomy, distorted",
-        "strength": 0.85
-    },
-    "ภาพวาดสีน้ำ (Watercolor)": {
-        "prompt": "beautiful watercolor painting, soft edges, artistic brushwork, traditional art medium, colorful artistic interpretation, fine art quality, watercolor texture",
-        "negative": "photographic, digital art, sharp edges, realistic, computer generated, 3d",
-        "strength": 0.80
-    },
-    "ภาพวาดน้ำมัน (Oil Painting)": {
-        "prompt": "classical oil painting, renaissance style, visible brush strokes, artistic masterpiece, museum quality fine art, old master technique, rich colors",
-        "negative": "photographic, digital, modern, flat colors, amateur, low quality, cartoon",
-        "strength": 0.80
-    },
-    "กำหนดเอง (Custom)": {
-        "prompt": "",
-        "negative": "",
-        "strength": 0.70
-    }
-}
-
-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,
-    progress=gr.Progress(track_tqdm=True)
-):
-    """Enhanced image-to-image restoration with LongCat-Image-Edit"""
-    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
-    else:
-        prompt = preset["prompt"]
-        negative_prompt = preset["negative"]
-        denoising_strength = preset["strength"]
-    
-    # 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
-    
-    # LongCat works best at 1024x1024
-    if device.type == "cpu":
-        max_size = 768
-    else:
-        max_size = 1024
-    
-    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:
-        # LongCat-Image-Edit generation
-        result = pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            image=original_image,
-            strength=denoising_strength,
-            guidance_scale=guidance_scale,
-            num_inference_steps=steps,
-            generator=generator,
-        ).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"### ✨ LongCat-Image-Edit • Professional Image Editing")
-        
-        gr.HTML(f"""
-        <div class="feature-box">
-            <h3>🚀 LongCat-Image-Edit Professional System</h3>
-            <table style="width:100%; color:white; font-size:1.05em;">
-                <tr>
-                    <td style="padding:8px;">⚡ <strong>Model:</strong></td>
-                    <td>LongCat-Image-Edit by Meituan</td>
-                </tr>
-                <tr>
-                    <td style="padding:8px;">🎯 <strong>Specialty:</strong></td>
-                    <td>Professional Image Editing & Restoration</td>
-                </tr>
-                <tr>
-                    <td style="padding:8px;">💎 <strong>Quality:</strong></td>
-                    <td>SDXL-based, optimized for editing tasks</td>
-                </tr>
-                <tr>
-                    <td style="padding:8px;">🎨 <strong>Features:</strong></td>
-                    <td>Better prompt following, precise edits</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>🔬 LongCat Benefits:</strong><br>
-                ✨ Better instruction following<br>
-                🎨 More accurate color restoration<br>
-                📐 Superior detail preservation<br>
-                💎 Specialized for image editing tasks
-            </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.75,
-                        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="LongCat: 25-35 แนะนำ | 40-50 คุณภาพสูงสุด"
-                    )
-                    
-                    enhance_input = gr.Checkbox(
-                        label="✨ ปรับปรุงภาพอินพุตอัตโนมัติ",
-                        value=True,
-                        info="เพิ่มความคมชัดและคอนทราสต์"
-                    )
-                    
-                    gr.HTML("""
-                    <div class="tip-box">
-                        <strong>💡 คำแนะนำ LongCat-Image-Edit:</strong><br>
-                        • <strong>Strength 0.4-0.6:</strong> แก้ไขเล็กน้อย<br>
-                        • <strong>Strength 0.7-0.8:</strong> แก้ไขปานกลาง ✅<br>
-                        • <strong>Strength 0.85+:</strong> เปลี่ยนแปลงมาก<br><br>
-                        • <strong>Guidance 7-9:</strong> สมดุลดี ✅<br>
-                        • <strong>Steps 25-35:</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
-            ],
-            outputs=[output_image, seed]
-        )
-
-    gr.Markdown(f"""
-    ---
-    ### 📚 คู่มือการใช้งาน LongCat-Image-Edit Professional
-    
-    #### 🚀 **ทำไมต้อง LongCat-Image-Edit?**
-    
-    **จุดเด่น:**
-    - ✅ **ออกแบบเฉพาะทาง** - เน้นการแก้ไขและปรับปรุงภาพโดยเฉพาะ
-    - ✅ **SDXL-based** - ใช้พื้นฐาน SDXL คุณภาพสูง
-    - ✅ **Better Following** - ทำตามคำสั่งได้แม่นยำกว่า
-    - ✅ **Precise Editing** - แก้ไขจุดเฉพาะได้ดีเยี่ยม
-    - ✅ **Natural Results** - ผลลัพธ์ดูเป็นธรรมชาติ
-    - ✅ **Flexible Strength** - ปรับระดับการแก้ไขได้ละเอียด
-    
-    **เหมาะกับ:**
-    - 🎨 การลงสีภาพขาวดำ
-    - 🔧 การซ่อมแซมภาพเก่า
-    - ✨ การปรับปรุงคุณภาพภาพ
-    - 🎭 การเปลี่ยนสไตล์ภาพ
-    - 📸 การทำ Photo Enhancement
-    
-    #### 🎯 **วิธีใช้งาน:**
-    
-    1. **อัพโหลดภาพ** ที่ต้องการแก้ไข
-    2. **เลือกพรีเซ็ต** - มี 12 รูปแบบให้เลือก
-    3. **ปรับ Strength** ตามความต้องการ (0.7-0.8 แนะนำ)
-    4. **เปิด Input Enhancement** สำหรับผลลัพธ์ดีขึ้น
-    5. **กดปุ่ม "เริ่มซ่อมแซมภาพ"**
-    6. **รอ 1-3 นาที** (ขึ้นกับฮาร์ดแวร์)
-    
-    #### ⚙️ **การตั้งค่าที่แนะนำ:**
-    
-    | งาน | Strength | Guidance | Steps |
-    |-----|----------|----------|-------|
-    | ลงสีภาพขาวดำ | 0.75 | 7-8 | 30 |
-    | ซ่อมแซมรอยขีดข่วน | 0.60 | 6-7 | 25-30 |
-    | เพิ่มความคมชัด | 0.45 | 5-6 | 25 |
-    | เปลี่ยนสไตล์ | 0.80-0.85 | 8-10 | 30-40 |
-    | คุณภาพสูงสุด | 0.70 | 7-9 | 40-50 |
-    
-    #### 💡 **เคล็ดลับ:**
-    
-    **สำหรับ CPU:**
-    - ใช้ Steps 20-25
-    - Guidance 6-7
-    - Strength 0.6-0.7
-    - ภาพไม่เกิน 768x768px
-    - Expected: 2-4 นาที/ภาพ
-    
-    **สำหรับ GPU (8GB+):**
-    - ใช้ Steps 30-40
-    - Guidance 7-9
-    - ภาพได้ถึง 1024x1024px
-    - Expected: 30-60 วินาที/ภาพ
-    
-    **สำหรับ GPU (16GB+):**
-    - ใช้ Steps 40-50
-    - Guidance 8-10
-    - ความละเอียดสูงสุด
-    - Expected: 20-45 วินาที/ภาพ
-    
-    #### 📊 **พรีเซ็ตทั้งหมด:**
-    
-    - 🎨 **ลงสีภาพขาวดำ** - แปลงภาพขาวดำเป็นสีสมจริง
-    - 🔧 **ซ่อมแซมเต็มรูปแบบ** - ลบรอยชำรุด ปรับปรุงคุณภาพ
-    - 🧹 **ลบรอยขีดข่วน** - กำจัดรอยเสียหาย
-    - 💎 **เพิ่มสีที่เลือนหาย** - ทำให้สีสดใสขึ้น
-    - 📐 **ลดสัญญาณรบกวน** - ลดเกรน เพิ่มความคมชัด
-    - ✨ **เพิ่มความคมชัด** - ทำให้ภาพคมชัดขึ้น
-    - 📸 **สไตล์โปรเฟสชันนัล** - แปลงเป็นภาพถ่ายสตูดิโอ
-    - 🎬 **สไตล์ภาพยนตร์** - ให้ความรู้สึกแบบหนัง
-    - 🎭 **เปลี่ยนเป็นอนิเมะ** - แปลงเป็นสไตล์อนิเมะ
-    - 🖼️ **ภาพวาดสีน้ำ** - แปลงเป็นภาพวาดสีน้ำ
-    - 🎨 **ภาพวาดน้ำมัน** - แปลงเป็นภาพวาดน้ำมัน
-    - ⚙️ **กำหนดเอง** - ปรับแต่งเองได้ทุกอย่าง
-    
-    #### ⚡ **ประสิทธิภาพ:**
-    
-    **LongCat-Image-Edit Performance:**
-    - **CPU (16GB RAM):** 2-4 นาที/ภาพ (768px)
-    - **GPU (8GB VRAM):** 30-60 วินาที/ภาพ (1024px)
-    - **GPU (12GB VRAM):** 25-50 วินาที/ภาพ (1024px)
-    - **GPU (16GB+ VRAM):** 20-45 วินาที/ภาพ (1024px)
-    
-    **ขนาดโมเดล:**
-    - LongCat-Image-Edit: ~6.5GB
-    - Total: ~6.5GB
-    
-    #### 🎯 **Use Cases:**
-    
-    ✅ **เหมาะสำหรับ:**
-    - ภาพถ่ายโบราณที่ต้องการซ่อมแซม
-    - ภาพที่ต้องการลงสี
-    - ภาพที่ต้องการแก้ไขเฉพาะจุด
-    - การปรับปรุงคุณภาพภาพ
-    - การเปลี่ยนสไตล์ภาพ
-    - งาน Professional Photo Editing
-    
-    ⚠️ **ข้อควรระวัง:**
-    - ต้องการ RAM/VRAM อย่างน้อย 8GB
-    - Strength สูงเกินไปอาจทำให้ภาพเปลี่ยนมาก
-    - ควร resize ภาพก่อนถ้าขนาดใหญ่มาก
-    
-    #### 💡 **Tips & Tricks:**
-    
-    **1. การเลือก Strength:**
-    - 0.4-0.6: แก้ไขเล็กน้อย (ลดสัญญาณรบกวน)
-    - 0.7-0.8: แก้ไขปานกลาง (แนะนำ) ✅
-    - 0.85-0.95: เปลี่ยนแปลงมาก (เปลี่ยนสไตล์)
-    
-    **2. ปรับปรุงคุณภาพ:**
-    - เปิ
-    
-    ---
-    """)
-
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)