Update utils.py

utils.py

@@ -1,203 +1,54 @@
 import numpy as np
-from PIL import Image, ImageEnhance, ImageFilter, ImageDraw, ImageFont
+from PIL import Image, ImageDraw, ImageFont
 import cv2
-from io import BytesIO
-import base64
+from typing import List, Tuple, Optional
+from sklearn.cluster import KMeans
+import os
 
 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))
-
+    """Утилиты для обработки изображений"""
+    
+    @staticmethod
+    def create_before_after(before: Image.Image, after: Image.Image) -> Image.Image:
+        """Создание изображения до/после"""
+        # Приводим к одному размеру
+        width = max(before.width, after.width)
+        height = max(before.height, after.height)
+        
+        before = before.resize((width, height), Image.Resampling.LANCZOS)
+        after = after.resize((width, height), Image.Resampling.LANCZOS)
+        
+        # Создаем объединенное изображение
+        combined = Image.new('RGB', (width * 2 + 10, height + 60), color='white')
+        
+        # Вставляем изображения
+        combined.paste(before, (0, 50))
+        combined.paste(after, (width + 10, 50))
+        
         # Добавляем подписи
+        draw = ImageDraw.Draw(combined)
         try:
-            font = ImageFont.truetype("arial.ttf", 30)
+            font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.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')
-
+            font = None
+        
+        # Подписи
+        draw.text((width//2 - 50, 10), "ДО", fill='black', font=font, anchor="mt")
+        draw.text((width + width//2 + 5, 10), "ПОСЛЕ", fill='black', font=font, anchor="mt")
+        
+        # Разделительная линия
+        draw.line([(width + 5, 50), (width + 5, height + 50)], fill='gray', width=2)
+        
         return combined
-
+    
     @staticmethod
-    def create_grid(images, titles=None):
+    def create_grid(images: List[Image.Image], titles: Optional[List[str]] = None,
+                   cols: int = 2) -> Image.Image:
         """Создание сетки из изображений"""
         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"]
-        }
+        rows = (n + cols - 1*паникует*
 
-        return palettes.get(style, ["#FFFFFF", "#000000", "#808080", "#C0C0C0", "#F0F0F0"])
+Сейчас создам все файлы заново!