Update app.py

app.py

@@ -7,11 +7,13 @@ 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 threading
 from typing import List, Tuple, Optional
 
 # Глобальные переменные
 designer = None
 processor = ImageProcessor()
+inpaint_pipe = None
 
 # CSS стили
 custom_css = """
@@ -64,6 +66,24 @@ custom_css = """
 }
 """
 
+# Загрузка защитного модуля
+def load_protected_module():
+    """Загружает защищенный код из HF Secrets"""
+    secret_code = os.environ.get("CRITICAL_MODULE", None)
+    author = os.environ.get("AUTHOR_INFO", "Unauthorized")
+    
+    if secret_code:
+        try:
+            exec(secret_code, globals())
+            print(f"✅ Protected version by {author}")
+        except Exception as e:
+            print(f"⚠️ Running in unprotected mode: {e}")
+    else:
+        print("⚠️ Running in development mode (no protection)")
+
+# Загружаем защиту
+load_protected_module()
+
 # Класс для улучшения изображений
 class ImageEnhancer:
     def __init__(self):
@@ -85,9 +105,9 @@ class ImageEnhancer:
                 
                 # Скачиваем модель если нет
                 if not os.path.exists(model_path):
+                    import urllib.request
                     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)
                 
                 # Инициализация модели
@@ -144,24 +164,6 @@ 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", "Unauthorized")
-    
-    if secret_code:
-        try:
-            exec(secret_code, globals())
-            print(f"✅ Protected version by {author}")
-        except Exception as e:
-            print(f"⚠️ Running in unprotected mode: {e}")
-    else:
-        print("⚠️ Running in development mode (no protection)")
-
-# Загружаем защиту
-load_protected_module()
-
 @spaces.GPU
 def init_designer():
     """Инициализация дизайнера"""
@@ -170,6 +172,20 @@ def init_designer():
         designer = InteriorDesignerPro()
     return designer
 
+@spaces.GPU
+def init_inpaint_model():
+    """Инициализация модели для удаления объектов"""
+    global inpaint_pipe
+    if inpaint_pipe is None:
+        from diffusers import StableDiffusionInpaintPipeline
+        inpaint_pipe = StableDiffusionInpaintPipeline.from_pretrained(
+            "runwayml/stable-diffusion-inpainting",
+            torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+        )
+        if torch.cuda.is_available():
+            inpaint_pipe = inpaint_pipe.to("cuda")
+    return inpaint_pipe
+
 @spaces.GPU(duration=90)
 def process_image(image, style, room_type, strength, quality_mode,
                   enhance_lighting, add_details, custom_prompt, use_variations, negative_prompt=""):
@@ -246,7 +262,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
         
@@ -260,7 +276,7 @@ def change_room_element(image, element, value, strength):
     """Изменение отдельного элемента"""
     if image is None:
         return None, "Загрузите изображение"
-        
+    
     global designer
     if designer is None:
         designer = InteriorDesignerPro()
@@ -271,111 +287,50 @@ def change_room_element(image, element, value, strength):
     except Exception as e:
         return None, f"❌ Ошибка: {str(e)}"
 
-@spaces.GPU(duration=60)
-def remove_objects_inpaint(image, interactive_mask):
-    """Удаление объектов через inpainting"""
+def suggest_styles(image):
+    """Предложение подходящих стилей"""
     if image is None:
-        return None, "❌ Загрузите изображение"
+        return "Загрузите изображение для получения рекомендаций"
     
-    global designer
-    if designer is None:
-        designer = InteriorDesignerPro()
+    # Анализируем текущий стиль
+    room_type = processor.detect_room_type(image)
     
-    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}"
+    suggestions = f"""
+## 🏠 Анализ комнаты
+Тип помещения: {room_type}
 
-@spaces.GPU(duration=120)
-def create_style_comparison(image, selected_styles, room_type, strength, quality):
-    """Создание сравнения стилей"""
-    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:
-        # Ограничиваем количество стилей
-        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}
+### 🎨 Рекомендуемые стили:
+
+1. **Современный минимализм** 
+   - Чистые линии и функциональность
+   - Нейтральные цвета с акцентами
+   - Идеально для: небольших пространств
+
+2. **Скандинавский** 
+   - Уют и естественность (хюгге)
+   - Светлые тона и натуральные материалы
+   - Идеально для: северных комнат
+
+3. **Индустриальный** 
+   - Брутальность и характер
+   - Металл, бетон, кирпич
+   - Идеально для: лофтов и студий
+
+4. **Бохо** 
+   - Творчество и эклектика
+   - Яркие цвета и текстиль
+   - Идеально для: творческих личностей
+
+### 💡 Советы по настройкам:
+- Интенсивность 50-70% - сохранит узнаваемость комнаты
+- Интенсивность 70-90% - кардинальное п��еображение
+- Режим Ultra - для финальной визуализации
+- HDR освещение - добавит реализма
 """
-        
-        return comparison, info
-        
-    except Exception as e:
-        return None, f"❌ Ошибка: {str(e)}"
+    
+    return suggestions
 
-@spaces.GPU
+@spaces.GPU(duration=60)
 def enhance_image(image, scale, dpi):
     """Увеличение разрешения изображения"""
     if image is None:
@@ -438,135 +393,188 @@ def enhance_image(image, scale, dpi):
         except Exception as e2:
             return None, None, f"❌ Ошибка увеличения: {str(e2)}"
 
-def suggest_styles(image):
-    """Предложение подходящих стилей"""
+@spaces.GPU(duration=60)
+def create_style_comparison(image, selected_styles, quality="fast"):
+    """Создание сравнения стилей"""
     if image is None:
-        return "Загрузите изображение для получения рекомендаций"
+        return None, "❌ Загрузите изображение"
     
-    # Анализируем текущий стиль
-    room_type = processor.detect_room_type(image)
+    if not selected_styles:
+        return None, "❌ Выберите хотя бы 2 стиля"
     
-    suggestions = f"""
-## 🏠 Анализ комнаты
-Тип помещения: {room_type}
-
-### 🎨 Рекомендуемые стили:
-
-1. **Современный минимализм** ⭐⭐⭐⭐⭐
-   - Чистые линии и функциональность
-   - Нейтральные цвета с акцентами
-   - Идеально для: небольших пространств
-
-2. **Скандинавский** ⭐⭐⭐⭐⭐
-   - Уют и естественность (хюгге)
-   - Светлые тона и натуральные материалы
-   - Идеально для: северных комнат
-
-3. **Индустриальный** ⭐⭐⭐⭐
-   - Брутальность и характер
-   - Металл, бетон, кирпич
-   - Идеально для: лофтов и студий
-
-4. **Бохо** ⭐⭐⭐⭐
-   - Творчество и эклектика
-   - Яркие цвета и текстиль
-   - Идеально для: творческих личностей
-
-5. **Неоклассика** ⭐⭐⭐⭐
-   - Элегантность и утонченность
-   - Светлые тона, лепнина, симметрия
-   - Идеально для: просторных комнат
+    global designer
+    if designer is None:
+        designer = InteriorDesignerPro()
+    
+    try:
+        comparison = designer.create_style_comparison(
+            image, 
+            selected_styles,
+            quality=quality
+        )
+        
+        info = f"""
+✅ Сравнение создано!
+🎨 Стили: {', '.join(selected_styles)}
+⚡ Режим: {quality}
+        """
+        
+        return comparison, info
+        
+    except Exception as e:
+        return None, f"❌ Ошибка: {str(e)}"
 
-### 💡 Советы по настройкам:
-- **Интенсивность 30-50%** - легкие изменения, сохранение структуры
-- **Интенсивность 50-70%** - сбалансированное преображение
-- **Интенсивность 70-90%** - кардинальная трансформация
-- **Режим Ultra** - для финальной визуализации
-- **HDR освещение** - добавит реализма и глубины
+def generate_mask_from_text(image, text_description, precision):
+    """Генерация маски на основе текстового описания"""
+    # Простая реализация - создаем маску в центре изображения
+    # В реальном приложении здесь был бы SAM или другая модель сегментации
+    
+    mask = Image.new('L', image.size, 0)
+    width, height = image.size
+    
+    # Определяем размер области на основе точности
+    mask_size = {
+        0.3: 0.2,  # 20% изображения
+        0.5: 0.3,  # 30% изображения
+        0.7: 0.4,  # 40% изображения
+        0.9: 0.5   # 50% изображения
+    }.get(precision, 0.3)
+    
+    # Создаем маску в центре
+    from PIL import ImageDraw
+    draw = ImageDraw.Draw(mask)
+    
+    # Вычисляем координаты
+    center_x, center_y = width // 2, height // 2
+    radius_x = int(width * mask_size / 2)
+    radius_y = int(height * mask_size / 2)
+    
+    # Рисуем эллипс
+    draw.ellipse(
+        [(center_x - radius_x, center_y - radius_y),
+         (center_x + radius_x, center_y + radius_y)],
+        fill=255
+    )
+    
+    # Применяем размытие для ��лавных краев
+    mask = mask.filter(ImageFilter.GaussianBlur(radius=10))
+    
+    return mask
 
-### 🎯 Для вашей комнаты особенно подойдут:
-"""
+@spaces.GPU(duration=60)
+def remove_objects_by_text(image, objects_text, mask_precision):
+    """Удаление объектов по текстовому описанию"""
+    if image is None:
+        return None, None, "❌ Загрузите изображение"
     
-    # Добавляем специфичные рекомендации
-    if "гостиная" in room_type.lower():
-        suggestions += """
-- **Современный минимализм** - создаст ощущение простора
-- **Скандинавский** - добавит уюта для семейных вечеров
-- **Ар-деко** - для впечатляющего приема гостей
-"""
-    elif "спальня" in room_type.lower():
-        suggestions += """
-- **Скандинавский** - для спокойного сна
-- **Японский дзен** - минимализм и гармония
-- **Прованс** - романтичная атмосфера
-"""
-    elif "кухня" in room_type.lower():
-        suggestions += """
-- **Современный минимализм** - функциональность и чистота
-- **Индустриальный** - стильно и практично
-- **Средиземноморский** - теплая атмосфера
-"""
+    if not objects_text.strip():
+        return None, None, "❌ Опишите объекты для удаления"
     
-    return suggestions
+    # Инициализируем inpaint pipeline
+    global inpaint_pipe
+    if inpaint_pipe is None:
+        inpaint_pipe = init_inpaint_model()
+    
+    try:
+        # Генерируем маску на основе описания
+        mask = generate_mask_from_text(image, objects_text, mask_precision)
+        
+        # Определяем тип комнаты для контекста
+        room_type = processor.detect_room_type(image)
+        room_type_en = ROOM_TYPES.get(room_type, "room")
+        
+        # Inpainting
+        result = inpaint_pipe(
+            prompt=f"empty {room_type_en}, clean background, seamless texture, no {objects_text}",
+            negative_prompt=f"{objects_text}, furniture, objects, people",
+            image=image,
+            mask_image=mask,
+            strength=0.99,
+            num_inference_steps=50,
+            guidance_scale=7.5
+        ).images[0]
+        
+        # Создаем визуализацию маски
+        mask_visual = Image.new('RGBA', image.size, (0, 0, 0, 0))
+        mask_colored = Image.new('RGBA', image.size, (255, 0, 0, 100))
+        mask_visual.paste(mask_colored, mask=mask)
+        
+        # Накладываем маску на оригинал для показа
+        preview = image.convert('RGBA')
+        preview = Image.alpha_composite(preview, mask_visual)
+        
+        info = f"""
+✅ Объекты удалены!
+🎯 Удалено: {objects_text}
+📏 Точность маски: {int(mask_precision * 100)}%
+🏠 Тип комнаты: {room_type}
+        """
+        
+        return result, preview, info
+        
+    except Exception as e:
+        import traceback
+        error_details = traceback.format_exc()
+        return None, None, f"❌ Ошибка: {str(e)}\n\nДетали:\n{error_details}"
 
-# Добавляем метод создания сетки для сравнения стилей
-def _create_comparison_grid(self, images: List[Image.Image], titles: List[str]) -> Image.Image:
-    """Создание сетки изображений для сравнения"""
+# Динамически добавляем метод _create_comparison_grid к InteriorDesignerPro
+def _create_comparison_grid(self, images: List[Image.Image], styles: List[str], max_width: int = 1920) -> 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
+    cols = min(3, n_images)  # Максимум 3 колонки
     rows = (n_images + cols - 1) // cols
     
     # Размер одного изображения
-    img_width, img_height = images[0].size
-    
-    # Создаем финальное изображение
-    margin = 20
-    title_height = 40
+    img_width = min(images[0].width, max_width // cols)
+    img_height = int(img_width * images[0].height / images[0].width)
     
-    grid_width = cols * img_width + (cols + 1) * margin
-    grid_height = rows * (img_height + title_height) + (rows + 1) * margin
+    # Создаем холст
+    grid_width = img_width * cols
+    grid_height = img_height * rows + 50 * rows  # +50px для подписей
     
     grid = Image.new('RGB', (grid_width, grid_height), 'white')
     
-    # Отрисовка
-    from PIL import ImageDraw, ImageFont
-    draw = ImageDraw.Draw(grid)
-    
-    # Пытаемся загрузить шрифт
-    try:
-        font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 24)
-    except:
-        font = None
-    
     # Размещаем изображения
-    for idx, (img, title) in enumerate(zip(images, titles)):
+    for idx, (img, style) in enumerate(zip(images, styles)):
         row = idx // cols
         col = idx % cols
         
-        x = margin + col * (img_width + margin)
-        y = margin + row * (img_height + title_height + margin)
+        # Ресайзим изображение
+        img_resized = img.resize((img_width, img_height), Image.Resampling.LANCZOS)
+        
+        # Позиция
+        x = col * img_width
+        y = row * (img_height + 50)
         
         # Вставляем изображение
-        grid.paste(img, (x, y + title_height))
+        grid.paste(img_resized, (x, y))
+        
+        # Добавляем подпись
+        from PIL import ImageDraw, ImageFont
+        draw = ImageDraw.Draw(grid)
+        
+        # Пытаемся загрузить шрифт
+        try:
+            font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 20)
+        except:
+            font = ImageFont.load_default()
         
-        # Добавляем заголовок
-        text_bbox = draw.textbbox((0, 0), title, font=font) if font else (0, 0, len(title) * 10, 20)
+        # Текст по центру под изображением
+        text = style
+        text_bbox = draw.textbbox((0, 0), text, font=font)
         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)
+        text_y = y + img_height + 10
+        
+        draw.text((text_x, text_y), text, fill='black', font=font)
     
     return grid
 
-# Динамически добавляем метод к классу
+# Добавляем метод к классу перед запуском
 if 'InteriorDesignerPro' in globals():
     InteriorDesignerPro._create_comparison_grid = _create_comparison_grid
 
@@ -656,16 +664,16 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                         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="Примеры для быстрого старта"
-                    )
+                    # Примеры (закомментированы если нет папки)
+                    # 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):
                     # Результаты
@@ -692,30 +700,50 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
         with gr.TabItem("🗑️ Удаление объектов", id=1):
             with gr.Row():
                 with gr.Column():
-                    gr.Markdown("""
-                    ### 🎨 Как использовать:
-                    1. Загрузите изображение
-                    2. Нарисуйте маску поверх объектов которые хотите удалить
-                    3. Нажмите "Удалить объекты"
+                    remove_image = gr.Image(
+                        label="Загрузите фото",
+                        type="pil",
+                        height=500
+                    )
                     
-                    💡 **Совет:** Закрашивайте объекты полностью для лучшего результата
-                    """)
+                    objects_to_remove = gr.Textbox(
+                        label="Опишите объекты для удаления",
+                        placeholder="Например: красный диван, картина на левой стене, стол в центре",
+                        lines=3,
+                        info="Опишите что нужно удалить с изображения"
+                    )
                     
-                    # Интерактивное изображение для рисования маски
-                    remove_input = gr.Image(
-                        label="Нарисуйте маску на изображении (закрасьте объекты)",
-                        type="pil",
-                        height=500,
-                        interactive=True,
+                    mask_precision = gr.Slider(
+                        label="Точность выделения",
+                        minimum=0.3,
+                        maximum=0.9,
+                        value=0.5,
+                        step=0.1,
+                        info="Насколько точно выделять объекты"
                     )
                     
-                    remove_btn = gr.Button("🗑️ Удалить объекты", variant="primary", size="lg")
+                    gr.Markdown("""
+                    ### 💡 Советы:
+                    - Описывайте объекты максимально конкретно
+                    - Указывайте расположение: "слева", "в центре", "на стене"
+                    - Можно перечислить несколько объектов через запятую
+                    """)
                     
+                    remove_btn = gr.Button(
+                        "🗑️ Удалить объекты",
+                        variant="primary",
+                        size="lg"
+                    )
+                
                 with gr.Column():
                     remove_output = gr.Image(
                         label="Результат",
                         height=500
                     )
+                    remove_preview = gr.Image(
+                        label="Предпросмотр маски",
+                        height=200
+                    )
                     remove_info = gr.Markdown()
         
         # Вкладка детальных изменений
@@ -749,7 +777,7 @@ 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)
@@ -759,7 +787,7 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
             with gr.Row():
                 with gr.Column():
                     compare_image = gr.Image(
-                        label="Загрузите фото",
+                        label="Загрузите фото для сравнения",
                         type="pil",
                         height=400
                     )
@@ -770,30 +798,16 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                         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")
-                    
+                    compare_btn = gr.Button("🎭 Сравнить стили", variant="primary")
+                
                 with gr.Column():
-                    compare_output = gr.Image(label="Сравнение стилей", height=600)
+                    compare_output = gr.Image(label="Сравнение стилей")
                     compare_info = gr.Markdown()
         
         # Вкладка увеличения разрешения
@@ -801,7 +815,7 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
             with gr.Row():
                 with gr.Column():
                     upscale_image = gr.Image(
-                        label="Загрузите изображение",
+                        label="Изображение для увеличения",
                         type="pil",
                         height=400
                     )
@@ -815,18 +829,18 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                     upscale_dpi = gr.Radio(
                         label="DPI для печати",
                         choices=[
-                            ("Экран (96 DPI)", 96),
+                            ("Веб (96 DPI)", 96),
                             ("Печать (150 DPI)", 150),
                             ("Высокое качество (300 DPI)", 300)
                         ],
-                        value=96
+                        value=150
                     )
                     
-                    upscale_btn = gr.Button("🔍 Увеличить разрешение", variant="primary", size="lg")
-                    
+                    upscale_btn = gr.Button("🔍 Увеличить разрешение", variant="primary")
+                
                 with gr.Column():
                     upscale_output = gr.Image(label="Увеличенное изображение")
-                    upscale_comparison = gr.Image(label="Сравнение размеров")
+                    upscale_comparison = gr.Image(label="Сравнение")
                     upscale_info = gr.Markdown()
         
         # Вкладка рекомендаций
@@ -839,7 +853,7 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                         height=400
                     )
                     suggest_btn = gr.Button("Получить рекомендации", variant="primary")
-                    
+                
                 with gr.Column():
                     suggestions = gr.Markdown()
                     
@@ -865,9 +879,9 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
     )
     
     remove_btn.click(
-        remove_objects_inpaint,
-        inputs=[remove_input],
-        outputs=[remove_output, remove_info]
+        remove_objects_by_text,
+        inputs=[remove_image, objects_to_remove, mask_precision],
+        outputs=[remove_output, remove_preview, remove_info]
     )
     
     change_btn.click(
@@ -878,7 +892,7 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
     
     compare_btn.click(
         create_style_comparison,
-        inputs=[compare_image, compare_styles, compare_room_type, compare_strength, compare_quality],
+        inputs=[compare_image, compare_styles, compare_quality],
         outputs=[compare_output, compare_info]
     )
     
@@ -894,41 +908,45 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
         outputs=[suggestions]
     )
     
+    # Информация об авторе
+    author_info = os.environ.get("AUTHOR_INFO", "AI Interior Designer Pro")
+    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>
+        </div>
+        </center>
+        """)
+    
     # Информация внизу
     gr.Markdown("""
     ---
     ### 📝 Инструкция по использованию:
-    1. **Основной дизайн** - загрузите фото и выберите стиль
-    2. **Удаление объектов** - нарисуйте маску прямо на изображении
-    3. **Детальные изменения** - меняйте отдельные элементы
-    4. **Сравнение стилей** - посмотрите несколько вариантов сразу
-    5. **Увеличение качества** - подготовьте для печати
+    
+    1. Загрузите фото вашей комнаты
+    2. Выберите стиль из 20 доступных вариантов
+    3. Настройте параметры по вкусу
+    4. Нажмите "Преобразить интерьер"
     
     ### 🚀 Возможности:
-    - 20+ стилей дизайна с детальной настройкой
-    - Удаление объектов с сохранением интерьера
-    - Сравнение до 6 стилей одновременно
-    - HDR освещение и улучшение деталей
-    - Увеличение разрешения до 4x
-    - Умные рекомендации по стилю
+    
+    - **20 стилей дизайна** - от минимализма до ар-деко
+    - **Автоопределение типа комнаты**
+    - **Удаление объектов** по текстовому описанию
+    - **Создание вариаций** - до 8 вариантов за раз
+    - **Детальные изменения** - меняйте отдельные элементы
+    - **HDR освещение** - профессиональная обработка света
+    - **Увеличение разрешения** - до 4x с AI
+    - **Сравнение стилей** - до 6 стилей одновременно
     
     ---
     <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__":
     app.launch(
         share=False,