Upload utils.py

utils.py

@@ -0,0 +1,203 @@
+import numpy as np
+from PIL import Image, ImageEnhance, ImageFilter, ImageDraw, ImageFont
+import cv2
+from io import BytesIO
+import base64
+
+class ImageProcessor:
+    @staticmethod
+    def resize_image(image, max_size=1024):
+        """Изменение размера с сохранением пропорций"""
+        width, height = image.size
+
+        if width > max_size or height > max_size:
+            if width > height:
+                new_width = max_size
+                new_height = int(height * (max_size / width))
+            else:
+                new_height = max_size
+                new_width = int(width * (max_size / height))
+
+            image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+
+        return image
+
+    @staticmethod
+    def create_before_after(original, processed):
+        """Создание сравнения до/после"""
+        width = original.width + processed.width + 20
+        height = max(original.height, processed.height)
+
+        combined = Image.new('RGB', (width, height), 'white')
+
+        # Вставляем оригинал
+        combined.paste(original, (0, 0))
+
+        # Линия разделитель
+        draw = ImageDraw.Draw(combined)
+        line_x = original.width + 10
+        draw.line([(line_x, 0), (line_x, height)], fill='gray', width=2)
+
+        # Вставляем обработанное
+        combined.paste(processed, (original.width + 20, 0))
+
+        # Добавляем подписи
+        try:
+            font = ImageFont.truetype("arial.ttf", 30)
+        except:
+            font = ImageFont.load_default()
+
+        draw.text((20, 20), "До", fill='white', font=font, stroke_width=2, stroke_fill='black')
+        draw.text((original.width + 40, 20), "После", fill='white', font=font, stroke_width=2, stroke_fill='black')
+
+        return combined
+
+    @staticmethod
+    def create_grid(images, titles=None):
+        """Создание сетки из изображений"""
+        if not images:
+            return None
+
+        # Определяем размер сетки
+        n = len(images)
+        cols = min(2, n)
+        rows = (n + cols - 1) // cols
+
+        # Размер ячейки
+        cell_width = images[0].width
+        cell_height = images[0].height
+
+        # Создаем холст
+        grid_width = cols * cell_width + (cols - 1) * 10
+        grid_height = rows * cell_height + (rows - 1) * 10
+
+        if titles:
+            grid_height += 40  # Место для заголовков
+
+        grid = Image.new('RGB', (grid_width, grid_height), 'white')
+        draw = ImageDraw.Draw(grid)
+
+        try:
+            font = ImageFont.truetype("arial.ttf", 20)
+        except:
+            font = ImageFont.load_default()
+
+        # Размещаем изображения
+        for i, img in enumerate(images):
+            row = i // cols
+            col = i % cols
+
+            x = col * (cell_width + 10)
+            y = row * (cell_height + 10)
+
+            if titles:
+                y += 40
+                # Добавляем заголовок
+                title_x = x + cell_width // 2
+                draw.text((title_x, y - 30), titles[i], fill='black', font=font, anchor='mm')
+
+            grid.paste(img, (x, y))
+
+        return grid
+
+    @staticmethod
+    def adjust_brightness(image, factor=1.2):
+        """Регулировка яркости"""
+        enhancer = ImageEnhance.Brightness(image)
+        return enhancer.enhance(factor)
+
+    @staticmethod
+    def adjust_contrast(image, factor=1.2):
+        """Регулировка контраста"""
+        enhancer = ImageEnhance.Contrast(image)
+        return enhancer.enhance(factor)
+
+    @staticmethod
+    def apply_filter(image, filter_type="none"):
+        """Применение фильтров"""
+        if filter_type == "blur":
+            return image.filter(ImageFilter.GaussianBlur(radius=2))
+        elif filter_type == "sharpen":
+            return image.filter(ImageFilter.SHARPEN)
+        elif filter_type == "edge_enhance":
+            return image.filter(ImageFilter.EDGE_ENHANCE)
+        else:
+            return image
+
+    @staticmethod
+    def detect_room_type(image):
+        """Автоопределение типа комнаты"""
+        # Здесь должна быть модель классификации
+        # Пока возвращаем заглушку
+        return "Гостиная"
+
+    @staticmethod
+    def image_to_base64(image):
+        """Конвертация изображения в base64"""
+        buffered = BytesIO()
+        image.save(buffered, format="PNG")
+        return base64.b64encode(buffered.getvalue()).decode()
+
+    @staticmethod
+    def create_mask(image, area):
+        """Создание маски для определенной области"""
+        mask = Image.new('L', image.size, 0)
+        draw = ImageDraw.Draw(mask)
+
+        # area должна быть в формате [(x1,y1), (x2,y2), ...]
+        if area:
+            draw.polygon(area, fill=255)
+
+        return mask
+
+class ColorPalette:
+    """Работа с цветовыми палитрами"""
+
+    @staticmethod
+    def extract_colors(image, n_colors=5):
+        """Извлечение основных цветов из изображения"""
+        # Уменьшаем изображение для скорости
+        small_image = image.resize((150, 150))
+
+        # Конвертируем в numpy array
+        pixels = np.array(small_image)
+        pixels = pixels.reshape(-1, 3)
+
+        # K-means для поиска основных цветов
+        from sklearn.cluster import KMeans
+        kmeans = KMeans(n_clusters=n_colors, random_state=42)
+        kmeans.fit(pixels)
+
+        colors = kmeans.cluster_centers_.astype(int)
+
+        # Создаем палитру
+        palette_width = 300
+        palette_height = 60
+        palette = Image.new('RGB', (palette_width, palette_height))
+        draw = ImageDraw.Draw(palette)
+
+        color_width = palette_width // n_colors
+
+        for i, color in enumerate(colors):
+            x1 = i * color_width
+            x2 = (i + 1) * color_width
+            color_tuple = tuple(color)
+            draw.rectangle([x1, 0, x2, palette_height], fill=color_tuple)
+
+        return palette, colors
+
+    @staticmethod
+    def suggest_palette(style):
+        """Предложение цветовой палитры для стиля"""
+        palettes = {
+            "Современный минимализм": ["#FFFFFF", "#F5F5F5", "#333333", "#000000", "#D4D4D4"],
+            "Скандинавский": ["#FFFFFF", "#F0F0F0", "#E8DED2", "#A8B5A6", "#5C5C5C"],
+            "Индустриальный": ["#2B2B2B", "#5C5C5C", "#8B7355", "#CD853F", "#1C1C1C"],
+            "Бохо": ["#D2691E", "#8B4513", "#DEB887", "#F4A460", "#FF6347"],
+            "Японский": ["#F5F5DC", "#D2B48C", "#8B7355", "#2F4F4F", "#F0E68C"],
+            "Ар-деко": ["#FFD700", "#000000", "#800020", "#2F4F4F", "#FFFFFF"],
+            "Прованс": ["#E6E6FA", "#DDA0DD", "#F0E68C", "#F5F5DC", "#98FB98"],
+            "Хай-тек": ["#C0C0C0", "#000000", "#FFFFFF", "#4169E1", "#696969"]
+        }
+
+        return palettes.get(style, ["#FFFFFF", "#000000", "#808080", "#C0C0C0", "#F0F0F0"])