Update app.py

app.py

@@ -446,7 +446,7 @@ def enhance_image(image, scale, dpi):
 
 @spaces.GPU(duration=30)
 def remove_objects_by_text(image, objects_text, mask_precision):
-    """Удаление объектов - оптимизированная версия"""
+    """Удаление объектов - улучшенная версия"""
     if image is None:
         return None, "❌ Загрузите изображение"
 
@@ -455,51 +455,97 @@ def remove_objects_by_text(image, objects_text, mask_precision):
         if designer is None:
             designer = InteriorDesignerPro()
 
-        # Ресайзим для скорости если слишком большое
+        # Ресайзим для скорости
         original_size = image.size
         if image.width > 768 or image.height > 768:
             image.thumbnail((768, 768), Image.Resampling.LANCZOS)
 
-        # Проверяем наличие object_remover
-        if not hasattr(designer, 'object_remover'):
-            from models import ObjectRemover
-            designer.object_remover = ObjectRemover(designer.inpaint_pipe)
-
-        # Простая маска
+        # Создаем маску на основе описания
         width, height = image.size
         mask = Image.new('L', (width, height), 0)
         draw = ImageDraw.Draw(mask)
 
-        # Быстрая генерация маски по центру
-        margin = int(min(width, height) * (0.5 - mask_precision/2))
-        draw.ellipse([margin, margin, width-margin, height-margin], fill=255)
-        mask = mask.filter(ImageFilter.GaussianBlur(radius=5))
-
-        # Быстрый inpainting
+        # Улучшенная логика создания маски
+        objects_lower = objects_text.lower()
+
+        # Проверяем ключевые слова для позиции
+        if any(word in objects_lower for word in ["весь", "все", "полностью"]):
+            # Маска на все изображение
+            draw.rectangle([0, 0, width, height], fill=255)
+        elif any(word in objects_lower for word in ["центр", "середина", "диван", "стол", "кресло"]):
+            # Центральная область (для мебели обычно в центре)
+            margin_x = int(width * 0.2)
+            margin_y = int(height * 0.25)
+            draw.rectangle([margin_x, margin_y, width-margin_x, height-margin_y], fill=255)
+        elif any(word in objects_lower for word in ["лев", "слева"]):
+            # Левая часть
+            draw.rectangle([0, 0, int(width * 0.4), height], fill=255)
+        elif any(word in objects_lower for word in ["прав", "справа"]):
+            # Правая часть
+            draw.rectangle([int(width * 0.6), 0, width, height], fill=255)
+        elif any(word in objects_lower for word in ["верх", "потолок", "люстра"]):
+            # Верхняя часть
+            draw.rectangle([0, 0, width, int(height * 0.4)], fill=255)
+        elif any(word in objects_lower for word in ["низ", "пол", "ковер", "коврик"]):
+            # Нижняя часть
+            draw.rectangle([0, int(height * 0.6), width, height], fill=255)
+        elif any(word in objects_lower for word in ["картин", "рам", "постер"]):
+            # Области на стенах (обычно в верхней половине)
+            # Левая стена
+            draw.rectangle([0, int(height * 0.2), int(width * 0.2), int(height * 0.6)], fill=255)
+            # Правая стена
+            draw.rectangle([int(width * 0.8), int(height * 0.2), width, int(height * 0.6)], fill=255)
+        elif any(word in objects_lower for word in ["угол", "углу"]):
+            # Углы
+            corner_size = int(min(width, height) * 0.3)
+            if "лев" in objects_lower:
+                draw.rectangle([0, 0, corner_size, corner_size], fill=255)
+            elif "прав" in objects_lower:
+                draw.rectangle([width-corner_size, 0, width, corner_size], fill=255)
+            else:
+                # Все углы
+                draw.rectangle([0, 0, corner_size, corner_size], fill=255)
+                draw.rectangle([width-corner_size, 0, width, corner_size], fill=255)
+        else:
+            # По умолчанию - центральная область с учетом precision
+            margin = int(min(width, height) * (0.5 - mask_precision))
+            draw.ellipse([margin, margin, width-margin, height-margin], fill=255)
+
+        # Размываем маску для плавных краев
+        mask = mask.filter(ImageFilter.GaussianBlur(radius=10))
+
+        # Усиливаем inpainting промпт на основе контекста
+        room_type = "interior"
+        if any(word in objects_lower for word in ["гостин", "зал"]):
+            room_type = "living room"
+        elif any(word in objects_lower for word in ["спальн", "кроват"]):
+            room_type = "bedroom"
+        elif any(word in objects_lower for word in ["кухн", "кухон"]):
+            room_type = "kitchen"
+
+        # Проверяем наличие inpaint_pipe
         if designer.inpaint_pipe:
             result = designer.inpaint_pipe(
-                prompt="empty clean background",
+                prompt=f"empty {room_type}, clean walls, seamless floor, no furniture",
+                negative_prompt=f"{objects_text}, furniture, objects, people",
                 image=image,
                 mask_image=mask,
-                strength=0.95,
-                num_inference_steps=25,
-                guidance_scale=5.0
+                strength=0.99,  # Максимальная сила для полного удаления
+                num_inference_steps=30,  # Немного больше шагов для качества
+                guidance_scale=7.5  # Выше для лучшего следования промпту
             ).images[0]
         else:
-            # Fallback на OpenCV
-            result = cv2.inpaint(
-                np.array(image), 
-                np.array(mask.convert('L')), 
-                3, 
-                cv2.INPAINT_TELEA
-            )
-            result = Image.fromarray(result)
+            # Fallback
+            from models import ObjectRemover
+            if not hasattr(designer, 'object_remover'):
+                designer.object_remover = ObjectRemover(None)
+            result = designer.object_remover.simple_inpaint(image, mask)
 
         # Возвращаем к оригинальному размеру
         if result.size != original_size:
             result = result.resize(original_size, Image.Resampling.LANCZOS)
 
-        return result, f"✅ Обработано за ~25 сек"
+        return result, f"✅ Удалено: {objects_text}"
 
     except Exception as e:
         import traceback
@@ -690,8 +736,84 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
                     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
+                    )
+
+                    selected_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")
+
+                with gr.Column():
+                    compare_output = gr.Image(label="Сравнение стилей", height=600)
+                    compare_info = gr.Textbox(label="Статус", lines=2)
+
+        # Вкладка увеличения разрешения
+        with gr.TabItem("🔍 Увеличение разрешения", id=4):
+            with gr.Row():
+                with gr.Column():
+                    upscale_image = gr.Image(
+                        label="Изображение для увеличения",
+                        type="pil",
+                        height=400
+                    )
+
+                    scale_factor = gr.Radio(
+                        label="Масштаб увеличения",
+                        choices=[("2x", 2), ("4x", 4)],
+                        value=2
+                    )
+
+                    dpi_setting = gr.Radio(
+                        label="DPI для печати",
+                        choices=[("Экран (96)", 96), ("Печать (150)", 150), ("Высокое качество (300)", 300)],
+                        value=96
+                    )
+
+                    upscale_btn = gr.Button("🔍 Увеличить разрешение", variant="primary")
+
+                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 = get_style_colors(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(
@@ -719,6 +841,24 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
         outputs=[remove_output, remove_info]
     )
 
+    compare_btn.click(
+        create_style_comparison,
+        inputs=[compare_image, selected_styles, compare_quality],
+        outputs=[compare_output, compare_info]
+    )
+
+    upscale_btn.click(
+        enhance_image,
+        inputs=[upscale_image, scale_factor, dpi_setting],
+        outputs=[upscale_output, upscale_comparison, upscale_info]
+    )
+
+    suggest_btn.click(
+        suggest_styles,
+        inputs=[suggest_image],
+        outputs=[suggestions]
+    )
+
     # Информация внизу
     author_info = os.environ.get("AUTHOR_INFO", "")
     if author_info:
@@ -736,11 +876,12 @@ with gr.Blocks(title="AI Дизайнер интерьера Pro", theme=gr.them
     4. Нажмите "Преобразить интерьер"
 
     ### 🚀 Возможности:
-    - 20 стилей дизайна 
+    - 20 стилей дизайна - от минимализма до ар-деко
     - Автоопределение типа комнаты
-    - Создание вариаций
-    - Удаление объектов
-    - HDR освещение
+    - Создание вариаций - до 8 вариантов за раз
+    - Детальные изменения - меняйте отдельные элементы
+    - HDR освещение - профессиональная обработка света
+    - Анализ цветовой палитры
 
     ---
     <center>Made with ❤️ for interior design enthusiasts</center>