src/utils/compress.py

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

def compress_image(
    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.
    
    Args:
        input_path: Path to input image
        output_path: Path for compressed output
        quality: Compression quality 1-95 (for JPEG/WEBP)
        format: Output format
        max_width: Maximum width (optional)
        max_height: Maximum height (optional)
    """
    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)}"