src/utils/compress.py

from PIL import Image
import os
from typing import Literal, Optional
from io import BytesIO

def compress_image_file(
    input_path: str,
    output_path: str,
    quality: int = 85,
    format: Literal["JPEG", "PNG", "WEBP"] = "JPEG",
    max_width: Optional[int] = None,
    max_height: Optional[int] = None
) -> str:
    """
    Compress an image file from disk.
    """
    try:
        if not os.path.splitext(output_path)[1]:
            extension_map = {"JPEG": ".jpg", "PNG": ".png", "WEBP": ".webp"}
            output_path = output_path + extension_map[format]
        
        with Image.open(input_path) as img:
            if format == "JPEG" and img.mode in ("RGBA", "P"):
                img = img.convert("RGB")
            
            if max_width or max_height:
                img.thumbnail((max_width or img.width, max_height or img.height), Image.Resampling.LANCZOS)
            
            save_kwargs = {"format": format, "optimize": True}
            if format in ["JPEG", "WEBP"]:
                save_kwargs["quality"] = quality
            
            img.save(output_path, **save_kwargs)

        original_size = os.path.getsize(input_path) / 1024 / 1024  
        compressed_size = os.path.getsize(output_path) / 1024 / 1024 
        reduction = (1 - compressed_size/original_size) * 100
        
        return f"✅ Compressed successfully!\nOriginal: {original_size:.2f}MB → Compressed: {compressed_size:.2f}MB\nReduction: {reduction:.1f}%"
    
    except Exception as e:
        return f"❌ Error: {str(e)}"

def compress_image_memory(image: Image.Image, quality: int = 80, format: str = "JPEG") -> Image.Image:
    """
    Compress an image in memory and return the compressed image.
    """
    if format == "JPEG" and image.mode in ("RGBA", "P"):
        image = image.convert("RGB")
    
    output = BytesIO()
    save_kwargs = {"format": format, "optimize": True}
    
    if format in ["JPEG", "WEBP"]:
        save_kwargs["quality"] = quality
    
    image.save(output, **save_kwargs)
    output.seek(0)
    
    return Image.open(output)