Update app.py

app.py

@@ -1,20 +1,17 @@
 import gradio as gr
 import numpy as np
-from PIL import Image, ImageFilter, ImageDraw
+from PIL import Image, ImageFilter
 import torch
 import spaces
 from models import InteriorDesignerPro
 from design_styles import DESIGN_STYLES, ROOM_TYPES, ROOM_ELEMENTS, get_detailed_prompt
 from utils import ImageProcessor, ColorPalette
 import os
-import urllib.request
 from typing import List, Tuple, Optional
-import cv2
 
 # Глобальные переменные
 designer = None
 processor = ImageProcessor()
-lama_remover = None
 
 # CSS стили
 custom_css = """
@@ -67,109 +64,6 @@ custom_css = """
 }
 """
 
-# Класс для удаления объектов через LAMA
-class LamaObjectRemover:
-    def __init__(self):
-        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-        self.pipe = None
-        
-    @spaces.GPU
-    def init_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 torch.cuda.is_available() else torch.float32,
-                    safety_checker=None,
-                    requires_safety_checker=False
-                ).to(self.device)
-                print("✅ LAMA model loaded successfully")
-            except Exception as e:
-                print(f"Failed to load inpainting model: {e}")
-                self.pipe = None
-                
-    def extract_mask_from_drawing(self, image_with_drawing):
-        """Извлекает маску из нарисованного изображения"""
-        if isinstance(image_with_drawing, dict):
-            # Новый формат Gradio
-            original = image_with_drawing.get("background", image_with_drawing.get("image"))
-            composite = image_with_drawing.get("composite", image_with_drawing.get("mask"))
-            
-            if original and composite:
-                # Создаем маску из разницы
-                orig_np = np.array(original)
-                comp_np = np.array(composite)
-                
-                # Находим отличия (где рисовали)
-                diff = np.any(orig_np != comp_np, axis=-1)
-                mask = Image.fromarray((diff * 255).astype(np.uint8))
-                
-                return original, mask
-        else:
-            # Старый формат или просто изображение с рисунком
-            img_array = np.array(image_with_drawing)
-            
-            # Ищем красные области (предполагаем что рисовали красным)
-            red_mask = (img_array[:,:,0] > 200) & (img_array[:,:,1] < 100) & (img_array[:,:,2] < 100)
-            
-            # Очищаем оригинал от красных меток
-            clean_img = img_array.copy()
-            clean_img[red_mask] = [255, 255, 255]  # Заменяем на белый
-            
-            mask = Image.fromarray((red_mask * 255).astype(np.uint8))
-            original = Image.fromarray(clean_img)
-            
-            return original, mask
-            
-    @spaces.GPU(duration=60)
-    def remove_objects(self, image_with_drawing):
-        """Удаляет объекты используя LAMA подход"""
-        if image_with_drawing is None:
-            return None, "❌ Пожалуйста, загрузите изображение"
-            
-        try:
-            # Инициализируем модель если нужно
-            if self.pipe is None:
-                self.init_model()
-                
-            # Извлекаем оригинал и маску
-            original, mask = self.extract_mask_from_drawing(image_with_drawing)
-            
-            # Проверяем что маска не пустая
-            if np.array(mask).sum() == 0:
-                return original, "⚠️ Ничего не выделено для удаления"
-            
-            # Немного расширяем маску для лучшего результата
-            mask_np = np.array(mask)
-            kernel = np.ones((5,5), np.uint8)
-            mask_np = cv2.dilate(mask_np, kernel, iterations=2)
-            mask = Image.fromarray(mask_np)
-            
-            if self.pipe:
-                # Используем inpainting pipeline
-                result = self.pipe(
-                    prompt="empty space, clean background, seamless texture",
-                    negative_prompt="furniture, objects, people, decorations",
-                    image=original,
-                    mask_image=mask,
-                    num_inference_steps=30,
-                    strength=0.99,
-                    guidance_scale=3.0
-                ).images[0]
-            else:
-                # Fallback на OpenCV inpainting
-                img_array = np.array(original)
-                mask_array = np.array(mask)
-                result_array = cv2.inpaint(img_array, mask_array, 3, cv2.INPAINT_TELEA)
-                result = Image.fromarray(result_array)
-                
-            return result, "✅ Объекты успешно удалены!"
-            
-        except Exception as e:
-            return None, f"❌ Ошибка: {str(e)}"
-
 # Класс для улучшения изображений
 class ImageEnhancer:
     def __init__(self):
@@ -193,6 +87,7 @@ class ImageEnhancer:
                 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...")
+                    import urllib.request
                     urllib.request.urlretrieve(url, model_path)
                 
                 # Инициализация модели
@@ -249,11 +144,11 @@ class ImageEnhancer:
         image.info['dpi'] = (dpi, dpi)
         return image
 
-# Загрузка защищенного модуля (если есть)
+# Загрузка защитного модуля
 def load_protected_module():
     """Загружает защищенный код из HF Secrets"""
     secret_code = os.environ.get("CRITICAL_MODULE", None)
-    author = os.environ.get("AUTHOR_INFO", "Development Mode")
+    author = os.environ.get("AUTHOR_INFO", "Unauthorized")
     
     if secret_code:
         try:
@@ -275,15 +170,6 @@ def init_designer():
         designer = InteriorDesignerPro()
     return designer
 
-@spaces.GPU
-def init_lama():
-    """Инициализация LAMA"""
-    global lama_remover
-    if lama_remover is None:
-        lama_remover = LamaObjectRemover()
-        lama_remover.init_model()
-    return lama_remover
-
 @spaces.GPU(duration=90)
 def process_image(image, style, room_type, strength, quality_mode,
                   enhance_lighting, add_details, custom_prompt, use_variations, negative_prompt=""):
@@ -317,16 +203,10 @@ def process_image(image, style, room_type, strength, quality_mode,
                 guidance_scale={"fast": 7.5, "balanced": 8.5, "ultra": 10}[quality_mode]
             ).images[0]
         else:
-            # Используем предустановленный стиль с детальным промптом
-            detailed_prompt = get_detailed_prompt(style, room_type_en)
-            styled_image = designer.pipe(
-                prompt=detailed_prompt,
-                negative_prompt=DESIGN_STYLES[style].get("negative", ""),
-                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]
+            # Используем предустановленный стиль
+            styled_image = designer.apply_style_pro(
+                image, style, room_type_en, strength, quality=quality_mode
+            )
         
         # HDR освещение
         if enhance_lighting and quality_mode != "fast":
@@ -366,7 +246,7 @@ def process_image(image, style, room_type, strength, quality_mode,
 """
         for i, color in enumerate(colors[:5]):
             hex_color = '#{:02x}{:02x}{:02x}'.format(*color)
-            info += f"\n • {hex_color}"
+            info += f"\n  • {hex_color}"
             
         return comparison, variations, palette, info
         
@@ -391,50 +271,111 @@ def change_room_element(image, element, value, strength):
     except Exception as e:
         return None, f"❌ Ошибка: {str(e)}"
 
-def suggest_styles(image):
-    """Предложение подходящих стилей"""
+@spaces.GPU(duration=60)
+def remove_objects_inpaint(image, interactive_mask):
+    """Удаление объектов через inpainting"""
     if image is None:
-        return "Загрузите изображение для получения рекомендаций"
-        
-    # Анализируем текущий стиль
-    room_type = processor.detect_room_type(image)
+        return None, "❌ Загрузите изображение"
     
-    suggestions = f"""
-## 🏠 Анализ комнаты
-Тип помещения: {room_type}
-
-### 🎨 Рекомендуемые стили:
-
-1. **Современный минимализм** 
-   - Чистые линии и функциональность
-   - Нейтральные цвета с акцентами
-   - Идеально для: небольших пространств
-
-2. **Скандинавский** 
-   - Уют и естественность (хюгге)
-   - Светлые тона и натуральные материалы
-   - Идеально для: северных комнат
-
-3. **Индустриальный** 
-   - Брутальность и характер
-   - Металл, бетон, кирпич
-   - Идеально для: лофтов и студий
-
-4. **Бохо** 
-   - Творчество и эклектика
-   - Яркие цвета и текстиль
-   - Идеально для: творческих личностей
+    global designer
+    if designer is None:
+        designer = InteriorDesignerPro()
+    
+    try:
+        # Проверяем тип данных
+        if isinstance(interactive_mask, dict) and 'mask' in interactive_mask:
+            # Новый формат Gradio
+            mask = interactive_mask['mask']
+        elif hasattr(interactive_mask, 'save'):
+            # Это уже изображение маски
+            mask = interactive_mask
+        else:
+            return None, "❌ Не удалось получить маску"
+        
+        # Убеждаемся что маска правильного размера
+        if mask.size != image.size:
+            mask = mask.resize(image.size, Image.Resampling.NEAREST)
+        
+        # Инвертируем маску если нужно (белый = удалить)
+        mask_array = np.array(mask.convert('L'))
+        if mask_array.mean() < 128:  # Если маска в основном черная
+            mask_array = 255 - mask_array
+        mask = Image.fromarray(mask_array)
+        
+        # Используем inpaint pipeline
+        if hasattr(designer, 'inpaint_pipe'):
+            result = designer.inpaint_pipe(
+                prompt="empty space, clean background, seamless texture, interior without objects",
+                negative_prompt="furniture, objects, people, decorations, artifacts",
+                image=image,
+                mask_image=mask,
+                strength=0.99,
+                num_inference_steps=50,
+                guidance_scale=7.5
+            ).images[0]
+        else:
+            # Fallback на обычный pipeline
+            result = designer.pipe(
+                prompt="clean empty interior space, seamless background",
+                image=image,
+                mask_image=mask,
+                strength=0.99,
+                num_inference_steps=50
+            ).images[0]
+        
+        return result, "✅ Объекты удалены!"
+        
+    except Exception as e:
+        import traceback
+        error_details = traceback.format_exc()
+        return None, f"❌ Ошибка: {str(e)}\n\n{error_details}"
 
-### 💡 Советы по настройкам:
-- Интенсивность 50-70% - сохранит узнаваемость комнаты
-- Интенсивность 70-90% - кардинальное преображение
-- Режим Ultra - для финальной визуализации
-- HDR освещение - добавит реализма
-"""
+@spaces.GPU(duration=120)
+def create_style_comparison(image, selected_styles, room_type, strength, quality):
+    """Создание сравнения стилей"""
+    if image is None:
+        return None, "❌ Загрузите изображение"
     
-    return suggestions
+    if not selected_styles or len(selected_styles) < 2:
+        return None, "❌ Выберите минимум 2 стиля для сравнения"
+    
+    global designer
+    if designer is None:
+        designer = InteriorDesignerPro()
+    
+    try:
+        # Ограничиваем количество стилей
+        selected_styles = selected_styles[:6]
+        
+        # Применяем каждый стиль
+        styled_images = []
+        for style in selected_styles:
+            if quality == "fast":
+                # Быстрый режим - уменьшаем разрешение
+                small_image = image.resize((512, 512), Image.Resampling.LANCZOS)
+                styled = designer.apply_style_pro(small_image, style, ROOM_TYPES.get(room_type, "living room"), strength, quality="fast")
+                styled = styled.resize(image.size, Image.Resampling.LANCZOS)
+            else:
+                styled = designer.apply_style_pro(image, style, ROOM_TYPES.get(room_type, "living room"), strength, quality=quality)
+            
+            styled_images.append(styled)
+        
+        # Создаем сетку
+        comparison = designer._create_comparison_grid(styled_images, selected_styles)
+        
+        info = f"""
+✅ Сравнение стилей завершено!
+📊 Стилей: {len(selected_styles)}
+🎨 Выбранные стили: {', '.join(selected_styles)}
+⚡ Режим: {quality}
+"""
+        
+        return comparison, info
+        
+    except Exception as e:
+        return None, f"❌ Ошибка: {str(e)}"
 
-@spaces.GPU(duration=60)
+@spaces.GPU
 def enhance_image(image, scale, dpi):
     """Увеличение разрешения изображения"""
     if image is None:
@@ -497,113 +438,137 @@ def enhance_image(image, scale, dpi):
         except Exception as e2:
             return None, None, f"❌ Ошибка увеличения: {str(e2)}"
 
-# Добавляем метод _create_comparison_grid к InteriorDesignerPro если его нет
-if hasattr(InteriorDesignerPro, 'create_style_comparison') and not hasattr(InteriorDesignerPro, '_create_comparison_grid'):
-    def _create_comparison_grid(self, images: List[Image.Image], styles: List[str], 
-                                cols: int = 3) -> Image.Image:
-        """Создание сетки для сравнения стилей"""
-        if not images:
-            return None
-            
-        # Размеры
-        img_width = 512
-        img_height = 512
-        padding = 20
-        text_height = 40
-        
-        # Изменяем размер всех изображений
-        resized_images = []
-        for img in images:
-            resized = img.resize((img_width, img_height), Image.Resampling.LANCZOS)
-            resized_images.append(resized)
-        
-        # Вычисляем размер итогового изображения
-        rows = (len(images) + cols - 1) // cols
-        total_width = cols * img_width + (cols + 1) * padding
-        total_height = rows * (img_height + text_height) + (rows + 1) * padding
-        
-        # Создаем пустое изображение
-        grid = Image.new('RGB', (total_width, total_height), color='white')
-        draw = ImageDraw.Draw(grid)
-        
-        # Размещаем изображения
-        for idx, (img, style) in enumerate(zip(resized_images, styles)):
-            row = idx // cols
-            col = idx % cols
-            
-            x = col * (img_width + padding) + padding
-            y = row * (img_height + text_height + padding) + padding
-            
-            # Вставляем изображение
-            grid.paste(img, (x, y))
-            
-            # Добавляем подпись
-            text_y = y + img_height + 5
-            draw.text((x + img_width//2, text_y), style, 
-                     fill='black', anchor='mt')
-        
-        return grid
+def suggest_styles(image):
+    """Предложение подходящих стилей"""
+    if image is None:
+        return "Загрузите изображение для получения рекомендаций"
     
-    InteriorDesignerPro._create_comparison_grid = _create_comparison_grid
+    # Анализируем текущий стиль
+    room_type = processor.detect_room_type(image)
+    
+    suggestions = f"""
+## 🏠 Анализ комнаты
+Тип помещения: {room_type}
 
-@spaces.GPU(duration=120)
-def create_style_comparison(image, selected_styles, room_type, strength, fast_mode):
-    """Создание сравнения нескольких стилей"""
-    if image is None:
-        return None, "❌ Пожалуйста, загрузите изображение"
-        
-    if not selected_styles:
-        return None, "❌ Выберите хотя бы 2 стиля для сравнения"
-        
-    global designer
-    if designer is None:
-        designer = InteriorDesignerPro()
-        
+### 🎨 Рекомендуемые стили:
+
+1. **Современный минимализм** ⭐⭐⭐⭐⭐
+   - Чистые линии и функциональность
+   - Нейтральные цвета с акцентами
+   - Идеально для: небольших пространств
+
+2. **Скандинавский** ⭐⭐⭐⭐⭐
+   - Уют и естественность (хюгге)
+   - Светлые тона и натуральные материалы
+   - Идеально для: северных комнат
+
+3. **Индустриальный** ⭐⭐⭐⭐
+   - Брутальность и характер
+   - Металл, бетон, кирпич
+   - Идеально для: лофтов и студий
+
+4. **Бохо** ⭐⭐⭐⭐
+   - Творчество и эклектика
+   - Яркие цвета и текстиль
+   - Идеально для: творческих личностей
+
+5. **Неоклассика** ⭐⭐⭐⭐
+   - Элегантность и утонченность
+   - Светлые тона, лепнина, симметрия
+   - Идеально для: просторных комнат
+
+### 💡 Советы по настройкам:
+- **Интенсивность 30-50%** - легкие изменения, сохранение структуры
+- **Интенсивность 50-70%** - сбалансированное преображение
+- **Интенсивность 70-90%** - кардинальная трансформация
+- **Режим Ultra** - для финальной визуализации
+- **HDR освещение** - добавит реализма и глубины
+
+### 🎯 Для вашей комна��ы особенно подойдут:
+"""
+    
+    # Добавляем специфичные рекомендации
+    if "гостиная" in room_type.lower():
+        suggestions += """
+- **Современный минимализм** - создаст ощущение простора
+- **Скандинавский** - добавит уюта для семейных вечеров
+- **Ар-деко** - для впечатляющего приема гостей
+"""
+    elif "спальня" in room_type.lower():
+        suggestions += """
+- **Скандинавский** - для спокойного сна
+- **Японский дзен** - минимализм и гармония
+- **Прованс** - романтичная атмосфера
+"""
+    elif "кухня" in room_type.lower():
+        suggestions += """
+- **Современный минимализм** - функциональность и чистота
+- **Индустриальный** - стильно и практично
+- **Средиземноморский** - теплая атмосфера
+"""
+    
+    return suggestions
+
+# Добавляем метод создания сетки для сравнения стилей
+def _create_comparison_grid(self, images: List[Image.Image], titles: List[str]) -> Image.Image:
+    """Создание сетки изображений для сравнения"""
+    if not images:
+        return None
+    
+    # Определяем размер сетки
+    n_images = len(images)
+    if n_images <= 2:
+        cols = 2
+    elif n_images <= 4:
+        cols = 2
+    else:
+        cols = 3
+    rows = (n_images + cols - 1) // cols
+    
+    # Размер одного изображения
+    img_width, img_height = images[0].size
+    
+    # Создаем финальное изображение
+    margin = 20
+    title_height = 40
+    
+    grid_width = cols * img_width + (cols + 1) * margin
+    grid_height = rows * (img_height + title_height) + (rows + 1) * margin
+    
+    grid = Image.new('RGB', (grid_width, grid_height), 'white')
+    
+    # Отрисовка
+    from PIL import ImageDraw, ImageFont
+    draw = ImageDraw.Draw(grid)
+    
+    # Пытаемся загрузить шрифт
     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")
-        
-        # Настройки качества для режима сравнения
-        steps = 20 if fast_mode else 30
-        
-        # Генерируем изображения для каждого стиля
-        styled_images = []
-        for style in selected_styles:
-            detailed_prompt = get_detailed_prompt(style, room_type_en)
-            
-            styled = designer.pipe(
-                prompt=detailed_prompt,
-                negative_prompt=DESIGN_STYLES[style].get("negative", ""),
-                image=image,
-                strength=strength,
-                num_inference_steps=steps,
-                guidance_scale=7.5
-            ).images[0]
-            
-            styled_images.append(styled)
-        
-        # Создаем сетку сравнения
-        comparison = designer._create_comparison_grid(
-            styled_images,
-            selected_styles,
-            cols=3 if len(selected_styles) > 4 else 2
-        )
-        
-        info = f"""
-✅ Сравнение стилей завершено!
-📊 Сравнено стилей: {len(selected_styles)}
-🏠 Тип комнаты: {room_type}
-⚡ Режим: {'Быстрый' if fast_mode else 'Качественный'}
-        """
-        
-        return comparison, info
-        
-    except Exception as e:
-        return None, f"❌ Ошибка: {str(e)}"
+        font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 24)
+    except:
+        font = None
+    
+    # Размещаем изображения
+    for idx, (img, title) in enumerate(zip(images, titles)):
+        row = idx // cols
+        col = idx % cols
+        
+        x = margin + col * (img_width + margin)
+        y = margin + row * (img_height + title_height + margin)
+        
+        # Вставляем изображение
+        grid.paste(img, (x, y + title_height))
+        
+        # Добавляем заголовок
+        text_bbox = draw.textbbox((0, 0), title, font=font) if font else (0, 0, len(title) * 10, 20)
+        text_width = text_bbox[2] - text_bbox[0]
+        text_x = x + (img_width - text_width) // 2
+        draw.text((text_x, y + 5), title, fill='black', font=font)
+    
+    return grid
+
+# Динамически добавляем метод к классу
+if 'InteriorDesignerPro' in globals():
+    InteriorDesignerPro._create_comparison_grid = _create_comparison_grid
 
 # Создаем интерфейс
 with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.themes.Soft(), css=custom_css) as app:
@@ -612,12 +577,6 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
     ### Преобразите свое пространство с помощью искусственного интеллекта
     """)
     
-    # Инициализация моделей при запуске
-    with gr.Row(visible=False):
-        init_btn = gr.Button("Init")
-        init_btn.click(fn=init_designer, inputs=[], outputs=[])
-        init_btn.click(fn=init_lama, inputs=[], outputs=[])
-    
     with gr.Tabs():
         # Основная вкладка
         with gr.TabItem("🎨 Дизайнер", id=0):
@@ -684,6 +643,7 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                             lines=3,
                             info="Опишите желаемый результат своими словами"
                         )
+                        
                         negative_prompt = gr.Textbox(
                             label="Чего избегать",
                             placeholder="dark, cluttered, old furniture",
@@ -728,100 +688,39 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                         label="Информация о процессе"
                     )
         
-        # Вкладка сравнения стилей
-        with gr.TabItem("🎭 Сравнение стилей", id=1):
+        # Вкладка удаления объектов
+        with gr.TabItem("🗑️ Удаление объектов", id=1):
             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_room_type = gr.Dropdown(
-                        label="Тип комнаты",
-                        choices=["Автоопределение"] + list(ROOM_TYPES.keys()),
-                        value="Автоопределение"
-                    )
-                    
-                    compare_strength = gr.Slider(
-                        label="Интенсивность",
-                        minimum=0.3,
-                        maximum=0.95,
-                        value=0.7,
-                        step=0.05
-                    )
+                    gr.Markdown("""
+                    ### 🎨 Как использовать:
+                    1. Загрузите изображение
+                    2. Нарисуйте маску поверх объектов которые хотите удалить
+                    3. Нажмите "Удалить объекты"
                     
-                    compare_fast = gr.Checkbox(
-                        label="⚡ Быстрый режим",
-                        value=True,
-                        info="Быстрее, но немного ниже качество"
-                    )
+                    💡 **Совет:** Закрашивайте объекты полностью для лучшего результата
+                    """)
                     
-                    compare_btn = gr.Button(
-                        "🎭 Сравнить стили",
-                        variant="primary"
-                    )
-                
-                with gr.Column():
-                    compare_output = gr.Image(
-                        label="Сравнение стилей",
-                        height=600
-                    )
-                    compare_info = gr.Markdown()
-        
-        # Вкладка удаления объектов с РИСОВАНИЕМ
-        with gr.TabItem("🗑️ Удаление объектов (LAMA)", id=2):
-            with gr.Row():
-                with gr.Column():
-                    # Используем gr.ImageEditor для рисования маски
-                    lama_input = gr.Image(
-                        label="Закрасьте объекты для удаления красным цветом",
-                        source="upload",
-                        tool="color-sketch",  # Позволяет рисовать
+                    # Интерактивное изображение для рисования маски
+                    remove_input = gr.Image(
+                        label="Нарисуйте маску на изображении (закрасьте объекты)",
                         type="pil",
                         height=500,
-                        brush_color="#FF0000",  # Красная кисть
-                        elem_id="lama_canvas"
+                        interactive=True,
+                        tool="sketch"
                     )
                     
-                    gr.Markdown("""
-                    ### 🖌️ Инструкция:
-                    1. Загрузите фото
-                    2. Выберите инструмент рисования (кисть)
-                    3. Закрасьте объекты КРАСНЫМ цветом
-                    4. Нажмите "Удалить объекты"
+                    remove_btn = gr.Button("🗑️ Удалить объекты", variant="primary", size="lg")
                     
-                    💡 Советы:
-                    - Закрашивайте объекты полностью
-                    - Используйте увеличение для точности
-                    - Можно стереть лишнее ластиком
-                    """)
-                    
-                    lama_btn = gr.Button(
-                        "🗑️ Удалить объекты",
-                        variant="primary",
-                        size="lg"
-                    )
-                
                 with gr.Column():
-                    lama_output = gr.Image(
+                    remove_output = gr.Image(
                         label="Результат",
                         height=500
                     )
-                    lama_info = gr.Textbox(
-                        label="Статус",
-                        lines=2
-                    )
+                    remove_info = gr.Markdown()
         
         # Вкладка детальных изменений
-        with gr.TabItem("🔧 Детальные изменения", id=3):
+        with gr.TabItem("🔧 Детальные изменения", id=2):
             with gr.Row():
                 with gr.Column():
                     detail_image = gr.Image(
@@ -851,17 +750,59 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                     )
                     
                     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_room_type = gr.Dropdown(
+                        label="Тип комнаты",
+                        choices=list(ROOM_TYPES.keys()),
+                        value="Гостиная"
+                    )
+                    
+                    compare_strength = gr.Slider(
+                        label="Интенсивность",
+                        minimum=0.3,
+                        maximum=0.9,
+                        value=0.7,
+                        step=0.05
+                    )
+                    
+                    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="Изображение для увеличения",
+                        label="Загрузите изображение",
                         type="pil",
                         height=400
                     )
@@ -875,27 +816,18 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                     upscale_dpi = gr.Radio(
                         label="DPI для печати",
                         choices=[
-                            ("96 DPI (экран)", 96),
-                            ("150 DPI (обычная печать)", 150),
-                            ("300 DPI (высокое качество)", 300)
+                            ("Экран (96 DPI)", 96),
+                            ("Печать (150 DPI)", 150),
+                            ("Высокое качество (300 DPI)", 300)
                         ],
-                        value=150
+                        value=96
                     )
                     
-                    upscale_btn = gr.Button(
-                        "🔍 Увеличить разрешение",
-                        variant="primary"
-                    )
-                
+                    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_output = gr.Image(label="Увеличенное изображение")
+                    upscale_comparison = gr.Image(label="Сравнение размеров")
                     upscale_info = gr.Markdown()
         
         # Вкладка рекомендаций
@@ -908,14 +840,14 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                         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 = DESIGN_STYLES[style_name].get("colors", ["#FFFFFF"] * 6)
+                        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>")
     
@@ -933,17 +865,10 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
         outputs=[output_variations]
     )
     
-    compare_btn.click(
-        create_style_comparison,
-        inputs=[compare_image, compare_styles, compare_room_type, compare_strength, compare_fast],
-        outputs=[compare_output, compare_info]
-    )
-    
-    # Обработчик для LAMA
-    lama_btn.click(
-        lambda img: lama_remover.remove_objects(img) if lama_remover else (None, "❌ LAMA не инициализирован"),
-        inputs=[lama_input],
-        outputs=[lama_output, lama_info]
+    remove_btn.click(
+        remove_objects_inpaint,
+        inputs=[remove_input],
+        outputs=[remove_output, remove_info]
     )
     
     change_btn.click(
@@ -952,6 +877,12 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
         outputs=[detail_output, detail_info]
     )
     
+    compare_btn.click(
+        create_style_comparison,
+        inputs=[compare_image, compare_styles, compare_room_type, compare_strength, compare_quality],
+        outputs=[compare_output, compare_info]
+    )
+    
     upscale_btn.click(
         enhance_image,
         inputs=[upscale_image, upscale_factor, upscale_dpi],
@@ -964,42 +895,39 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
         outputs=[suggestions]
     )
     
-    # Информация о защите
-    author_info = os.environ.get("AUTHOR_INFO", "Development Mode")
-    if "Development" not in author_info:
-        with gr.Row():
-            gr.Markdown(f"""
-            ---
-            <center>
-            <div style='background: linear-gradient(45deg, #667eea 0%, #764ba2 100%); 
-                        padding: 20px; border-radius: 10px; color: white;'>
-            <h3>🔒 {author_info}</h3>
-            <p>Protected AI Interior Designer Pro</p>
-            </div>
-            </center>
-            """)
-    
     # Информация внизу
     gr.Markdown("""
     ---
     ### 📝 Инструкция по использованию:
-    
-    1. **Основной дизайнер** - загрузите фото и выберите стиль
-    2. **Сравнение стилей** - посмотрите как выглядят разные стили на одном фото
-    3. **Удаление объектов** - закрасьте ненужные предметы красным цветом
-    4. **Детальные изменения** - измените отдельные элементы интерьера
-    5. **Увеличение разрешения** - улучшите качество для печати
+    1. **Основной дизайн** - загрузите фото и выберите стиль
+    2. **Удаление объектов** - нарисуйте маску прямо на изображении
+    3. **Детальные изменения** - меняйте отдельные элементы
+    4. **Сравнение стилей** - посмотрите несколько вариантов сразу
+    5. **Увеличение качества** - подготовьте для печати
     
     ### 🚀 Возможности:
-    - 20+ стилей дизайна с детальными настройками
-    - Удаление объектов с помощью LAMA
-    - Создание вариаций дизайна
+    - 20+ стилей дизайна с детальной настройкой
+    - Удаление объектов с сохранением интерьера
+    - Сравнение до 6 стилей одновременно
     - HDR освещение и улучшение деталей
     - Увеличение разрешения до 4x
+    - Умные рекомендации по стилю
     
     ---
     <center>Made with ❤️ for interior design enthusiasts</center>
     """)
+    
+    # Показываем информацию об авторе если есть
+    author_info = os.environ.get("AUTHOR_INFO", None)
+    if author_info:
+        gr.Markdown(f"""
+        <center>
+        <div style='background: linear-gradient(45deg, #667eea 0%, #764ba2 100%); 
+                    padding: 20px; border-radius: 10px; color: white; margin-top: 20px;'>
+        <h3>🔒 {author_info}</h3>
+        </div>
+        </center>
+        """)
 
 # ВАЖНО! Запуск приложения
 if __name__ == "__main__":