Create app.py

app.py

@@ -0,0 +1,1097 @@
+import gradio as gr
+import numpy as np
+from PIL import Image, ImageFilter, ImageDraw, ImageFont
+import torch
+import spaces
+from models import InteriorDesignerPro
+from design_styles import DESIGN_STYLES, ROOM_TYPES, ROOM_ELEMENTS
+from utils import ImageProcessor, ColorPalette
+import os
+import urllib.request
+from typing import List, Tuple, Optional
+import hashlib
+import cv2
+
+# Загрузка защищенных настроек
+AUTHOR_INFO = os.environ.get("AUTHOR_INFO", "© 2024 AI Interior Designer Pro")
+SPACE_ID = os.environ.get("SPACE_ID", "default-space")
+
+# Критическая инициализация из secrets
+def load_critical_module():
+    """Загружает защищенный код из HF Secrets"""
+    secret_code = os.environ.get("CRITICAL_MODULE", None)
+    if secret_code:
+        try:
+            exec(secret_code, globals())
+            print("✅ Critical module loaded successfully")
+        except Exception as e:
+            print(f"⚠️ Critical module failed: {e}")
+    else:
+        print("⚠️ Running in unprotected mode")
+        # Базовая инициализация для разработки
+        global SECURITY_KEY
+        SECURITY_KEY = hashlib.sha256(f"{SPACE_ID}_{AUTHOR_INFO}".encode()).hexdigest()
+
+# Загружаем критический модуль
+load_critical_module()
+
+# Глобальные переменные
+designer = None
+processor = ImageProcessor()
+
+# CSS стили
+custom_css = """
+.container {
+    max-width: 1400px;
+    margin: 0 auto;
+}
+.result-gallery {
+    display: grid;
+    grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
+    gap: 20px;
+    margin-top: 20px;
+}
+.style-button {
+    margin: 5px;
+    padding: 10px 20px;
+    border-radius: 20px;
+    background: linear-gradient(45deg, #667eea 0%, #764ba2 100%);
+    color: white;
+    font-weight: bold;
+    transition: transform 0.2s;
+}
+.style-button:hover {
+    transform: scale(1.05);
+}
+.info-box {
+    background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
+    padding: 20px;
+    border-radius: 15px;
+    margin: 10px 0;
+    box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+}
+.quality-badge {
+    display: inline-block;
+    padding: 5px 15px;
+    border-radius: 20px;
+    font-weight: bold;
+    margin: 5px;
+}
+.quality-fast { background-color: #10b981; color: white; }
+.quality-balanced { background-color: #3b82f6; color: white; }
+.quality-ultra { background-color: #8b5cf6; color: white; }
+.gpu-info {
+    background-color: #1e293b;
+    color: #e2e8f0;
+    padding: 10px 20px;
+    border-radius: 10px;
+    font-family: monospace;
+    margin: 10px 0;
+}
+.author-badge {
+    background: linear-gradient(45deg, #667eea 0%, #764ba2 100%);
+    color: white;
+    padding: 15px 30px;
+    border-radius: 10px;
+    text-align: center;
+    font-weight: bold;
+    margin: 20px 0;
+    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.4);
+}
+"""
+
+# Функция добавления водяного знака (если определена в CRITICAL_MODULE)
+def add_watermark(image):
+    """Добавляет водяной знак на изображение"""
+    if 'SecurityValidator' in globals():
+        return SecurityValidator.inject_watermark(image)
+    # Базовый водяной знак
+    draw = ImageDraw.Draw(image)
+    text = AUTHOR_INFO.split("|")[0].strip()
+    try:
+        font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 20)
+    except:
+        font = None
+    draw.text((10, image.height - 30), text, fill=(255, 255, 255, 128), font=font)
+    return image
+
+# Класс для улучшения изображений
+class ImageEnhancer:
+    def __init__(self):
+        self.model = None
+        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        
+    def load_model(self):
+        """Загрузка модели Real-ESRGAN"""
+        if self.model is None:
+            try:
+                from realesrgan import RealESRGANer
+                from basicsr.archs.rrdbnet_arch import RRDBNet
+                
+                model_dir = os.path.expanduser('~/.cache/realesrgan/')
+                os.makedirs(model_dir, exist_ok=True)
+                
+                model_path = os.path.join(model_dir, 'RealESRGAN_x4plus.pth')
+                
+                if not os.path.exists(model_path):
+                    url = 'https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth'
+                    print(f"Downloading Real-ESRGAN model...")
+                    urllib.request.urlretrieve(url, model_path)
+                
+                model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
+                
+                self.model = RealESRGANer(
+                    scale=4,
+                    model_path=model_path,
+                    model=model,
+                    tile=0,
+                    tile_pad=10,
+                    pre_pad=0,
+                    half=True if self.device.type == 'cuda' else False
+                )
+                
+                print("Real-ESRGAN model loaded successfully!")
+                return True
+                
+            except Exception as e:
+                print(f"Failed to load Real-ESRGAN: {e}")
+                return False
+                
+    def upscale(self, image: Image.Image, scale: int = 2) -> Image.Image:
+        """Увеличение разрешения изображения"""
+        try:
+            if self.load_model() and self.model is not None:
+                img_np = np.array(image)
+                output, _ = self.model.enhance(img_np, outscale=scale)
+                return Image.fromarray(output)
+        except:
+            pass
+            
+        # Fallback на PIL resize
+        new_width = int(image.width * scale)
+        new_height = int(image.height * scale)
+        
+        resized = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+        
+        if scale > 2:
+            resized = resized.filter(ImageFilter.UnsharpMask(radius=1, percent=150, threshold=3))
+            
+        return resized
+        
+    def set_dpi(self, image: Image.Image, dpi: int) -> Image.Image:
+        """Установка DPI для печати"""
+        image.info['dpi'] = (dpi, dpi)
+        return image
+
+# Класс для LAMA удаления объектов
+class LamaObjectRemover:
+    def __init__(self):
+        self.pipe = None
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        
+    def load_model(self):
+        """Загрузка модели для inpainting"""
+        if self.pipe is None:
+            try:
+                from diffusers import StableDiffusionInpaintPipeline
+                self.pipe = StableDiffusionInpaintPipeline.from_pretrained(
+                    "runwayml/stable-diffusion-inpainting",
+                    torch_dtype=torch.float16 if self.device == "cuda" else torch.float32,
+                    variant="fp16" if self.device == "cuda" else None
+                )
+                self.pipe = self.pipe.to(self.device)
+                print("LAMA model loaded successfully!")
+                return True
+            except Exception as e:
+                print(f"Failed to load LAMA model: {e}")
+                return False
+    
+    @spaces.GPU
+    def remove_objects(self, image: Image.Image, mask: Image.Image) -> Image.Image:
+        """Удаление объектов с изображения"""
+        try:
+            # Пробуем загрузить модель
+            if self.load_model() and self.pipe is not None:
+                # Приводим размеры к кратным 8
+                w, h = image.size
+                new_w = (w // 8) * 8
+                new_h = (h // 8) * 8
+                
+                image_resized = image.resize((new_w, new_h), Image.Resampling.LANCZOS)
+                mask_resized = mask.resize((new_w, new_h), Image.Resampling.LANCZOS)
+                
+                # Inpainting с минимальными изменениями
+                result = self.pipe(
+                    prompt="empty space, clean background, seamless texture",
+                    negative_prompt="furniture, objects, people, decorations",
+                    image=image_resized,
+                    mask_image=mask_resized,
+                    height=new_h,
+                    width=new_w,
+                    strength=0.99,
+                    guidance_scale=3.0,
+                    num_inference_steps=50
+                ).images[0]
+                
+                # Возвращаем к исходному размеру
+                return result.resize((w, h), Image.Resampling.LANCZOS)
+                
+        except Exception as e:
+            print(f"Model-based inpainting failed: {e}")
+        
+        # Fallback на OpenCV inpainting
+        try:
+            img_np = np.array(image)
+            mask_np = np.array(mask.convert('L'))
+            
+            # Расширяем маску для лучшего результата
+            kernel = np.ones((5,5), np.uint8)
+            mask_np = cv2.dilate(mask_np, kernel, iterations=1)
+            
+            # Применяем inpainting
+            result = cv2.inpaint(img_np, mask_np, 3, cv2.INPAINT_TELEA)
+            
+            return Image.fromarray(result)
+            
+        except Exception as e:
+            print(f"OpenCV inpainting failed: {e}")
+            return image
+
+# Инициализация глобальных объектов
+lama_remover = LamaObjectRemover()
+
+# Добавление метода _create_comparison_grid к классу InteriorDesignerPro
+def _create_comparison_grid(self, images: List[Image.Image], styles: List[str], 
+                           original: Optional[Image.Image] = None) -> Image.Image:
+    """Создает сетку сравнения сти��ей"""
+    if not images:
+        return None
+        
+    # Размеры для сетки
+    n_images = len(images) + (1 if original else 0)
+    cols = min(3, n_images)
+    rows = (n_images + cols - 1) // cols
+    
+    # Размер одного изображения в сетке
+    img_width = 512
+    img_height = 512
+    padding = 20
+    text_height = 60
+    
+    # Создаем холст
+    grid_width = cols * img_width + (cols - 1) * padding
+    grid_height = rows * (img_height + text_height) + (rows - 1) * padding
+    
+    grid = Image.new('RGB', (grid_width, grid_height), 'white')
+    draw = ImageDraw.Draw(grid)
+    
+    # Шрифт для подписей
+    try:
+        font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 20)
+    except:
+        font = None
+    
+    # Добавляем изображения
+    all_images = []
+    all_labels = []
+    
+    if original:
+        all_images.append(original)
+        all_labels.append("Оригинал")
+    
+    all_images.extend(images)
+    all_labels.extend(styles)
+    
+    for idx, (img, label) in enumerate(zip(all_images, all_labels)):
+        row = idx // cols
+        col = idx % cols
+        
+        # Позиция изображения
+        x = col * (img_width + padding)
+        y = row * (img_height + text_height + padding)
+        
+        # Изменяем размер и вставляем
+        img_resized = img.resize((img_width, img_height), Image.Resampling.LANCZOS)
+        grid.paste(img_resized, (x, y))
+        
+        # Добавляем подпись
+        text_y = y + img_height + 10
+        # Центрируем текст
+        if font:
+            bbox = draw.textbbox((0, 0), label, font=font)
+            text_width = bbox[2] - bbox[0]
+        else:
+            text_width = len(label) * 6
+            
+        text_x = x + (img_width - text_width) // 2
+        
+        # Фон для текста
+        draw.rectangle(
+            [x, y + img_height, x + img_width, y + img_height + text_height],
+            fill='#f0f0f0'
+        )
+        
+        # Текст
+        draw.text(
+            (text_x, text_y),
+            label,
+            fill='black',
+            font=font
+        )
+    
+    return grid
+
+# Динамически добавляем метод к классу
+InteriorDesignerPro._create_comparison_grid = _create_comparison_grid
+
+@spaces.GPU
+def init_designer():
+    """Инициализация дизайнера"""
+    global designer
+    if designer is None:
+        # Проверка безопасности если определена
+        if 'critical_init' in globals():
+            critical_init()
+        designer = InteriorDesignerPro()
+    return designer
+
+@spaces.GPU(duration=90)
+def process_image(image, style, room_type, strength, quality_mode,
+                  enhance_lighting, add_details, custom_prompt, use_variations, negative_prompt=""):
+    """Основная функция обработки изображения"""
+    if image is None:
+        return None, None, None, "❌ Пожалуйста, загрузите изображение"
+    
+    # Инициализируем designer
+    global designer
+    if designer is None:
+        designer = InteriorDesignerPro()
+        
+    try:
+        # Определяем тип комнаты если не указан
+        if room_type == "Автоопределение":
+            room_type = processor.detect_room_type(image)
+            room_type_en = ROOM_TYPES.get(room_type, "living room")
+        else:
+            room_type_en = ROOM_TYPES.get(room_type, "living room")
+        
+        # Применяем стиль с выбранным качеством
+        if custom_prompt:
+            quality_boost = "masterpiece, best quality, ultra-detailed, 8k uhd, " if quality_mode == "ultra" else ""
+            styled_image = designer.pipe(
+                prompt=f"{quality_boost}{room_type_en}, {custom_prompt}, interior design, high quality",
+                negative_prompt=negative_prompt if negative_prompt else None,
+                image=image,
+                strength=strength,
+                num_inference_steps={"fast": 20, "balanced": 35, "ultra": 50}[quality_mode],
+                guidance_scale={"fast": 7.5, "balanced": 8.5, "ultra": 10}[quality_mode]
+            ).images[0]
+        else:
+            styled_image = designer.apply_style_pro(
+                image, style, room_type_en, strength, quality=quality_mode
+            )
+        
+        # HDR освещение
+        if enhance_lighting and quality_mode != "fast":
+            styled_image = designer.create_hdr_lighting(styled_image, intensity=0.3)
+        
+        # Улучшение деталей
+        if add_details and designer.is_powerful_gpu and quality_mode == "ultra":
+            styled_image = designer.enhance_details(styled_image)
+        
+        # Добавляем водяной знак
+        styled_image = add_watermark(styled_image)
+        
+        # Создаем сравнение до/после
+        comparison = processor.create_before_after(image, styled_image)
+        
+        # Создаем вариации если запрошено
+        variations = None
+        if use_variations:
+            num_var = 8 if designer.is_powerful_gpu else 4
+            var_images = designer.create_variations(image, num_variations=num_var)
+            # Добавляем водяные знаки на вариации
+            var_images = [add_watermark(img) for img in var_images]
+            variations = processor.create_grid(
+                var_images, 
+                titles=[f"Вариант {i+1}" for i in range(len(var_images))]
+            )
+        
+        # Извлекаем цветовую палитру
+        palette, colors = ColorPalette.extract_colors(styled_image)
+        
+        # Информация о процессе
+        info = f"""
+✅ Обработка завершена успешно!
+📐 Тип комнаты: {room_type}
+🎨 Стиль: {style}
+💪 Интенсивность: {int(strength * 100)}%
+⚡ Режим качества: {quality_mode.upper()}
+🖼️ Разрешение: {styled_image.width}×{styled_image.height}
+🤖 Модель: {designer.model_name}
+🔒 Protected by: {AUTHOR_INFO.split('|')[0]}
+
+🎨 Основные цвета дизайна:
+"""
+        for i, color in enumerate(colors[:5]):
+            hex_color = '#{:02x}{:02x}{:02x}'.format(*color)
+            info += f"\n  • {hex_color}"
+        
+        return comparison, variations, palette, info
+        
+    except Exception as e:
+        import traceback
+        error_details = traceback.format_exc()
+        return None, None, None, f"❌ Ошибка: {str(e)}\n\nДетали:\n{error_details}"
+
+@spaces.GPU
+def change_room_element(image, element, value, strength):
+    """Изменение отдельного элемента"""
+    if image is None:
+        return None, "Загрузите изображение"
+    
+    global designer
+    if designer is None:
+        designer = InteriorDesignerPro()
+    
+    try:
+        result = designer.change_element(image, element, value, strength)
+        result = add_watermark(result)
+        return result, f"✅ {element} изменен на {value}"
+    except Exception as e:
+        return None, f"❌ Ошибка: {str(e)}"
+
+@spaces.GPU
+def remove_objects_lama(image, mask, inpaint_prompt=""):
+    """Удаление объектов с использованием LAMA"""
+    if image is None:
+        return None, "❌ Загрузите изображение"
+    
+    try:
+        # Если маска не загружена, возвращаем оригинал
+        if mask is None:
+            return image, "⚠️ Загрузите маску для удаления объектов"
+        
+        # Применяем LAMA
+        result = lama_remover.remove_objects(image, mask)
+        result = add_watermark(result)
+        
+        return result, "✅ Объекты успешно удалены"
+        
+    except Exception as e:
+        return None, f"❌ Ошибка: {str(e)}"
+
+@spaces.GPU
+def create_auto_mask(image, objects_text):
+    """Создание маски по текстовому описанию"""
+    if image is None:
+        return None, "❌ Загрузите изображение"
+    
+    try:
+        # Простая демо-маска (в реальности нужен GroundingDINO или SAM)
+        mask = Image.new('L', image.size, 0)
+        draw = ImageDraw.Draw(mask)
+        
+        # Создаем примерную маску в центре
+        width, height = image.size
+        margin = min(width, height) // 4
+        
+        if "центр" in objects_text.lower() or "middle" in objects_text.lower():
+            draw.ellipse([margin, margin, width-margin, height-margin], fill=255)
+        elif "лев" in objects_text.lower() or "left" in objects_text.lower():
+            draw.rectangle([0, 0, width//2, height], fill=255)
+        elif "прав" in objects_text.lower() or "right" in objects_text.lower():
+            draw.rectangle([width//2, 0, width, height], fill=255)
+        else:
+            # По умолчанию - центральная область
+            draw.ellipse([margin, margin, width-margin, height-margin], fill=255)
+        
+        return mask, "✅ Маска создана. Для точного выделения используйте ручное рисование."
+        
+    except Exception as e:
+        return None, f"❌ Ошибка: {str(e)}"
+
+def suggest_styles(image):
+    """Предложение подходящих стилей"""
+    if image is None:
+        return "Загрузите изображение для получения рекомендаций"
+    
+    room_type = processor.detect_room_type(image)
+    
+    suggestions = f"""
+## 🏠 Анализ комнаты
+Тип помещения: {room_type}
+
+### 🎨 Рекомендуемые стили:
+
+1. **Современный минимализм** 
+   - Чистые линии и функциональность
+   - Нейтральные цвета с акцентами
+   - Идеально для: небольших пространств
+
+2. **Скандинавский** 
+   - Уют и естественность (хюгге)
+   - Светлые тона и натуральные материалы
+   - Идеально для: северных комнат
+
+3. **Индустриальный** 
+   - Брутальность и характер
+   - Металл, бетон, кирпич
+   - Идеально для: лофтов и студий
+
+4. **Бохо** 
+   - Творчество и эклектика
+   - Яркие цвета и текстиль
+   - Идеально для: творческих личностей
+
+### 💡 Советы по настройкам:
+- Интенсивность 50-70% - сохранит узнаваемость комнаты
+- Интенсивность 70-90% - кардинальное преображение
+- Режим Ultra - для финальной визуализации
+- HDR освещение - добавит реализма
+
+🔒 {AUTHOR_INFO}
+"""
+    return suggestions
+
+@spaces.GPU(duration=60)
+def create_style_comparison(image, selected_styles, quality="balanced"):
+    """Создание сравнения стилей"""
+    if image is None:
+        return None, "❌ Загрузите изображение"
+    
+    if not selected_styles or len(selected_styles) < 2:
+        return None, "❌ Выберите минимум 2 стиля для сравнения"
+    
+    global designer
+    if designer is None:
+        designer = InteriorDesignerPro()
+    
+    try:
+        room_type = processor.detect_room_type(image)
+        room_type_en = ROOM_TYPES.get(room_type, "living room")
+        
+        # Генерируем изображения для каждого стиля
+        styled_images = []
+        for style in selected_styles[:6]:  # Максимум 6 стилей
+            styled = designer.apply_style_pro(
+                image, style, room_type_en, 
+                strength=0.75, 
+                quality="fast" if quality == "balanced" else quality
+            )
+            styled = add_watermark(styled)
+            styled_images.append(styled)
+        
+        # Создаем сетку сравнения
+        comparison_grid = designer._create_comparison_grid(
+            styled_images, 
+            selected_styles[:6], 
+            original=image
+        )
+        
+        info = f"""
+✅ Сравнение стилей готово!
+📐 Тип комнаты: {room_type}
+🎨 Стили: {', '.join(selected_styles[:6])}
+🖼️ Количество вариантов: {len(styled_images)}
+🔒 {AUTHOR_INFO}
+"""
+        
+        return comparison_grid, info
+        
+    except Exception as e:
+        return None, f"❌ Ошибка: {str(e)}"
+
+@spaces.GPU
+def enhance_image(image, scale, dpi):
+    """Увеличение разрешения изображения"""
+    if image is None:
+        return None, None, "❌ Пожалуйста, загрузите изображение"
+    
+    try:
+        enhancer = ImageEnhancer()
+        
+        original_size = f"{image.width}×{image.height}"
+        
+        # Улучшаем изображение
+        enhanced = enhancer.upscale(image, scale=scale)
+        
+        # Проверяем, изменился ли размер
+        if enhanced.size == image.size:
+            new_width = image.width * scale
+            new_height = image.height * scale
+            enhanced = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+            method = "PIL LANCZOS"
+        else:
+            method = "Real-ESRGAN"
+        
+        # Настраиваем DPI
+        enhanced = enhancer.set_dpi(enhanced, dpi)
+        
+        # Добавляем водяной знак
+        enhanced = add_watermark(enhanced)
+        
+        # Создаем сравнение
+        comparison = processor.create_before_after(image, enhanced)
+        
+        enhanced_size = f"{enhanced.width}×{enhanced.height}"
+        
+        info = f"""
+✅ Изображение увеличено!
+📐 Исходный размер: {original_size}
+📐 Новый размер: {enhanced_size}
+🔍 Масштаб: {scale}x
+📊 DPI: {dpi}
+🔧 Метод: {method}
+🔒 {AUTHOR_INFO}
+"""
+        
+        return enhanced, comparison, info
+        
+    except Exception as e:
+        try:
+            new_width = int(image.width * scale)
+            new_height = int(image.height * scale)
+            enhanced = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+            enhanced = add_watermark(enhanced)
+            
+            info = f"""
+⚠️ Real-ESRGAN недоступен, использован PIL resize
+📐 Исходный размер: {image.width}×{image.height}
+📐 Новый размер: {enhanced.width}×{enhanced.height}
+🔍 Масштаб: {scale}x
+Ошибка: {str(e)}
+"""
+            return enhanced, None, info
+        except Exception as e2:
+            return None, None, f"❌ Ошибка увеличения: {str(e2)}"
+
+# Создаем интерфейс
+with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.themes.Soft(), css=custom_css) as app:
+    # Добавляем информацию об авторе вверху
+    gr.HTML(f"""
+    <div class="author-badge">
+        <h3>{AUTHOR_INFO}</h3>
+        <p>Protected AI Interior Designer Pro | Space ID: {SPACE_ID}</p>
+    </div>
+    """)
+    
+    gr.Markdown("""
+    # 🎨 AI Дизайнер интерьера Pro
+    ### Преобразите свое пространство с помощью искусственного интеллекта
+    """)
+    
+    with gr.Tabs():
+        # Основная вкладка
+        with gr.TabItem("🎨 Дизайнер", id=0):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    # Входные параметры
+                    input_image = gr.Image(
+                        label="Загрузите фото комнаты",
+                        type="pil",
+                        height=400
+                    )
+                    
+                    with gr.Accordion("🎨 Настройки дизайна", open=True):
+                        style = gr.Dropdown(
+                            label="Стиль интерьера",
+                            choices=list(DESIGN_STYLES.keys()),
+                            value="Современный минимализм"
+                        )
+                        
+                        room_type = gr.Dropdown(
+                            label="Тип комнаты",
+                            choices=["Автоопределение"] + list(ROOM_TYPES.keys()),
+                            value="Автоопределение"
+                        )
+                        
+                        strength = gr.Slider(
+                            label="Интенсивность изменений",
+                            minimum=0.3,
+                            maximum=0.95,
+                            value=0.75,
+                            step=0.05,
+                            info="Насколько сильно изменить оригинал"
+                        )
+                        
+                        quality_mode = gr.Radio(
+                            label="Качество генерации",
+                            choices=[
+                                ("⚡ Быстрое (5 сек)", "fast"),
+                                ("⚖️ Сбалансированное (15 сек)", "balanced"), 
+                                ("💎 Ультра (30 сек)", "ultra")
+                            ],
+                            value="balanced",
+                            info="Выберите баланс между скоростью и качеством"
+                        )
+                        
+                        with gr.Row():
+                            enhance_lighting = gr.Checkbox(
+                                label="✨ HDR освещение",
+                                value=True
+                            )
+                            add_details = gr.Checkbox(
+                                label="🔍 Улучшить детализацию",
+                                value=True
+                            )
+                            use_variations = gr.Checkbox(
+                                label="🎭 Создать вариации",
+                                value=False
+                            )
+                    
+                    with gr.Accordion("🎯 Продвинутые настройки", open=False):
+                        custom_prompt = gr.Textbox(
+                            label="Кастомное описание (английский)",
+                            placeholder="cozy room with warm colors, plants, natural light",
+                            lines=3,
+                            info="Опишите желаемый результат своими словами"
+                        )
+                        
+                        negative_prompt = gr.Textbox(
+                            label="Чего избегать",
+                            placeholder="dark, cluttered, old furniture",
+                            lines=2
+                        )
+                    
+                    process_btn = gr.Button(
+                        "🎨 Преобразить интерьер",
+                        variant="primary",
+                        size="lg"
+                    )
+                    
+                    # Примеры
+                    gr.Examples(
+                        examples=[
+                            ["examples/living_room.jpg", "Современный минимализм", "Гостиная", 0.75, "balanced"],
+                            ["examples/bedroom.jpg", "Скандинавский", "Спальня", 0.7, "ultra"],
+                            ["examples/kitchen.jpg", "Индустриальный", "Кухня", 0.8, "balanced"]
+                        ],
+                        inputs=[input_image, style, room_type, strength, quality_mode],
+                        label="Примеры для быстрого старта"
+                    )
+                
+                with gr.Column(scale=1):
+                    # Результаты
+                    output_comparison = gr.Image(
+                        label="Сравнение: До и После",
+                        height=400
+                    )
+                    
+                    with gr.Row():
+                        output_variations = gr.Image(
+                            label="Варианты дизайна",
+                            visible=False
+                        )
+                        
+                        output_palette = gr.Image(
+                            label="Цветовая палитра",
+                            height=100
+                        )
+                    
+                    output_info = gr.Markdown(
+                        label="Информация о процессе"
+                    )
+        
+        # Вкладка удаления объектов с LAMA
+        with gr.TabItem("🗑️ Удаление объектов (LAMA)", id=1):
+            with gr.Row():
+                with gr.Column():
+                    lama_image = gr.Image(
+                        label="Загрузите фото",
+                        type="pil",
+                        height=400
+                    )
+                    
+                    gr.Markdown("""
+                    ### 📝 Как удалить объекты:
+                    
+                    **Вариант 1: Создайте маску вручную**
+                    1. Сохраните изображение выше
+                    2. Откройте в любом редакторе (Paint, Photoshop, GIMP)
+                    3. Закрасьте объекты БЕЛЫМ цветом
+                    4. Загрузите маску ниже
+                    
+                    **Вариант 2: Автоматическая маска**
+                    - Опишите объекты текстом ниже
+                    """)
+                    
+                    lama_mask = gr.Image(
+                        label="Загрузите маску (белый = удалить, черный = оставить)",
+                        type="pil",
+                        height=400
+                    )
+                    
+                    with gr.Row():
+                        auto_mask_text = gr.Textbox(
+                            label="Или опишите объекты для удаления",
+                            placeholder="мебель в центре, диван, стол",
+                            lines=2
+                        )
+                        
+                        create_mask_btn = gr.Button("Создать маску автоматически")
+                    
+                    lama_inpaint_prompt = gr.Textbox(
+                        label="Чем заполнить пустоты (опционально)",
+                        placeholder="empty floor, clean wall",
+                        lines=2
+                    )
+                    
+                    remove_btn = gr.Button(
+                        "🗑️ Удалить объекты",
+                        variant="primary",
+                        size="lg"
+                    )
+                
+                with gr.Column():
+                    lama_output = gr.Image(
+                        label="Результат",
+                        height=400
+                    )
+                    
+                    lama_info = gr.Textbox(
+                        label="Статус",
+                        lines=3
+                    )
+                    
+                    # Примеры масок
+                    gr.Markdown("""
+                    ### 💡 Советы:
+                    - Чем точнее маска, тем лучше результат
+                    - Для мебели делайте маску чуть больше объекта
+                    - LAMA сохраняет стиль комнаты, тол��ко убирает объекты
+                    """)
+        
+        # Остальные вкладки...
+        # Вкладка детальных изменений
+        with gr.TabItem("🔧 Детальные изменения", id=2):
+            with gr.Row():
+                with gr.Column():
+                    detail_image = gr.Image(
+                        label="Изображение для изменения",
+                        type="pil",
+                        height=300
+                    )
+                    
+                    element_type = gr.Dropdown(
+                        label="Что изменить",
+                        choices=list(ROOM_ELEMENTS.keys()),
+                        value="Стены"
+                    )
+                    
+                    element_value = gr.Textbox(
+                        label="Новое значение",
+                        placeholder="Например: белый цвет, деревянный паркет",
+                        lines=2
+                    )
+                    
+                    element_strength = gr.Slider(
+                        label="Сила изменения",
+                        minimum=0.3,
+                        maximum=0.9,
+                        value=0.5,
+                        step=0.05
+                    )
+                    
+                    change_btn = gr.Button("Применить изменение", variant="primary")
+                
+                with gr.Column():
+                    detail_output = gr.Image(label="Результат", height=300)
+                    detail_info = gr.Textbox(label="Статус", lines=2)
+        
+        # Вкладка сравнения стилей
+        with gr.TabItem("🎭 Сравнение стилей", id=3):
+            with gr.Row():
+                with gr.Column():
+                    compare_image = gr.Image(
+                        label="Загрузите фото для сравнения стилей",
+                        type="pil",
+                        height=400
+                    )
+                    
+                    compare_styles = gr.CheckboxGroup(
+                        label="Выберите стили для сравнения (от 2 до 6)",
+                        choices=list(DESIGN_STYLES.keys()),
+                        value=[]
+                    )
+                    
+                    compare_quality = gr.Radio(
+                        label="Качество сравнения",
+                        choices=[
+                            ("Быстрое", "fast"),
+                            ("Сбалансированное", "balanced")
+                        ],
+                        value="fast"
+                    )
+                    
+                    compare_btn = gr.Button(
+                        "🎭 Сравнить стили",
+                        variant="primary",
+                        size="lg"
+                    )
+                
+                with gr.Column():
+                    compare_output = gr.Image(
+                        label="Сравнение стилей",
+                        height=600
+                    )
+                    
+                    compare_info = gr.Markdown()
+        
+        # Вкладка увеличения разрешения
+        with gr.TabItem("🔍 Увеличение разрешения", id=4):
+            with gr.Row():
+                with gr.Column():
+                    upscale_image = gr.Image(
+                        label="Загрузите изображение для увеличения",
+                        type="pil",
+                        height=400
+                    )
+                    
+                    upscale_factor = gr.Radio(
+                        label="Масштаб увеличения",
+                        choices=[("2x", 2), ("4x", 4)],
+                        value=2
+                    )
+                    
+                    upscale_dpi = gr.Radio(
+                        label="DPI для печати",
+                        choices=[
+                            ("Экран (96 DPI)", 96),
+                            ("Документ (150 DPI)", 150),
+                            ("Печать (300 DPI)", 300)
+                        ],
+                        value=150
+                    )
+                    
+                    upscale_btn = gr.Button(
+                        "🔍 Увеличить разрешение",
+                        variant="primary",
+                        size="lg"
+                    )
+                
+                with gr.Column():
+                    upscale_output = gr.Image(
+                        label="Увеличенное изображение",
+                        height=400
+                    )
+                    
+                    upscale_comparison = gr.Image(
+                        label="Сравнение",
+                        height=200
+                    )
+                    
+                    upscale_info = gr.Markdown()
+        
+        # Вкладка рекомендаций
+        with gr.TabItem("💡 Рекомендации", id=5):
+            with gr.Row():
+                with gr.Column():
+                    suggest_image = gr.Image(
+                        label="Загрузите фото для анализа",
+                        type="pil",
+                        height=400
+                    )
+                    
+                    suggest_btn = gr.Button("Получить рекомендации", variant="primary")
+                
+                with gr.Column():
+                    suggestions = gr.Markdown()
+                    
+                    # Цветовые палитры для стилей
+                    gr.Markdown("### 🎨 Цветовые палитры популярных стилей")
+                    
+                    for style_name in list(DESIGN_STYLES.keys())[:4]:
+                        colors = ColorPalette.suggest_palette(style_name)
+                        color_blocks = " ".join([
+                            f'<span style="background-color:{c}; padding:10px 20px; margin:2px; display:inline-block; border-radius:5px;"></span>' 
+                            for c in colors
+                        ])
+                        gr.HTML(f"<div class='info-box'><strong>{style_name}:</strong><br>{color_blocks}</div>")
+    
+    # Обработчики событий
+    process_btn.click(
+        process_image,
+        inputs=[input_image, style, room_type, strength, quality_mode,
+                enhance_lighting, add_details, custom_prompt, use_variations, negative_prompt],
+        outputs=[output_comparison, output_variations, output_palette, output_info]
+    )
+    
+    use_variations.change(
+        lambda x: gr.update(visible=x),
+        inputs=[use_variations],
+        outputs=[output_variations]
+    )
+    
+    # LAMA handlers
+    create_mask_btn.click(
+        create_auto_mask,
+        inputs=[lama_image, auto_mask_text],
+        outputs=[lama_mask, lama_info]
+    )
+    
+    remove_btn.click(
+        remove_objects_lama,
+        inputs=[lama_image, lama_mask, lama_inpaint_prompt],
+        outputs=[lama_output, lama_info]
+    )
+    
+    change_btn.click(
+        change_room_element,
+        inputs=[detail_image, element_type, element_value, element_strength],
+        outputs=[detail_output, detail_info]
+    )
+    
+    compare_btn.click(
+        create_style_comparison,
+        inputs=[compare_image, compare_styles, compare_quality],
+        outputs=[compare_output, compare_info]
+    )
+    
+    upscale_btn.click(
+        enhance_image,
+        inputs=[upscale_image, upscale_factor, upscale_dpi],
+        outputs=[upscale_output, upscale_comparison, upscale_info]
+    )
+    
+    suggest_btn.click(
+        suggest_styles,
+        inputs=[suggest_image],
+        outputs=[suggestions]
+    )
+    
+    # Информация внизу
+    gr.Markdown(f"""
+    ---
+    ### 📝 Инструкция по использованию:
+    1. Загрузите фото вашей комнаты
+    2. Выберите стиль из 8 доступных вариантов
+    3. Настройте параметры по вкусу
+    4. Нажмите "Преобразить интерьер"
+    
+    ### 🚀 Возможности:
+    - 8 стилей дизайна - от минимализма до ар-деко
+    - Автоопределение типа комнаты
+    - Создание вариаций - до 8 вариантов за раз
+    - Удаление объектов с LAMA
+    - Детальные изменения - меняйте отдельные элементы
+    - HDR освещение - профессиональная обработка света
+    - Анализ цветовой палитры
+    
+    ---
+    <center>
+        <div class="author-badge">
+            <p>{AUTHOR_INFO}</p>
+            <p>Made with ❤️ for interior design enthusiasts</p>
+        </div>
+    </center>
+    """)
+
+# ВАЖНО! Запуск приложения
+if __name__ == "__main__":
+    app.launch(
+        share=False,
+        show_error=True,
+        server_name="0.0.0.0",
+        server_port=7860
+    )