import os
from pathlib import Path
from PIL import Image
from tqdm import tqdm
from concurrent.futures import ThreadPoolExecutor

# Пути к директориям исходных и целевых файлов
source_dir = Path("uploads/WPL/ALL_IMAGE_FOLDERS/")
target_dir = Path("uploads-webp/WPL/ALL_IMAGE_FOLDERS/")
quality = 70  # качество WebP

# Функция для конвертации изображений в одной директории
def convert_directory(folder):
    images = list(folder.rglob("*.jpg")) + list(folder.rglob("*.jpeg"))
    folder_name = folder.relative_to(source_dir)
    
    # Отображаем прогресс для текущей директории
    with tqdm(total=len(images), desc=f"Конвертация {folder_name}", unit="файл") as pbar:
        for img_path in images:
            # Задаем путь для сохранения нового изображения
            relative_path = img_path.relative_to(source_dir)
            new_path = target_dir / relative_path.with_suffix(".webp")
            new_path.parent.mkdir(parents=True, exist_ok=True)

            try:
                # Конвертируем изображение
                with Image.open(img_path) as img:
                    img.save(new_path, "WEBP", quality=quality)
            except (UnidentifiedImageError, OSError) as e:
                print(f"Ошибка при конвертации {img_path}: {e}")
                # Пропускаем текущее изображение и продолжаем обработку
            finally:
                pbar.update(1)

# Основная функция для запуска многопоточного выполнения
def convert_images_to_webp():
    # Получаем первую вложенность директорий
    folders = [f for f in source_dir.iterdir() if f.is_dir()]
    
    # Запускаем пул потоков
    with ThreadPoolExecutor(max_workers=2) as executor:
        for i, _ in enumerate(executor.map(convert_directory, folders), start=1):
            print(f"Обработана директория {i}/{len(folders)}")

if __name__ == "__main__":
    convert_images_to_webp()