Function compress_image_if_needed

RealChart2Code_eval/evaluate_task2.py:452–509

(image_path: Path, max_size_bytes: int = 5 * 1024 * 1024)

def compress_image_if_needed(image_path: Path, max_size_bytes: int = 5 * 1024 * 1024) -> bytes:
    image_path = Path(image_path)
    actual_target = int(max_size_bytes * 0.75)
    original_size = image_path.stat().st_size
    
    with Image.open(image_path) as img:
        if img.mode != 'RGB':
            img = img.convert('RGB')
            width, height = img.size
            if width > JPEG_MAX_DIMENSION or height > JPEG_MAX_DIMENSION:
                logging.warning(
                    f"Image '{image_path}' has dimensions ({width}x{height}) "
                    f"exceeding JPEG limit. Resizing it down."
                )
                aspect_ratio = width / height
                if width > height:
                    new_width = JPEG_MAX_DIMENSION
                    new_height = int(new_width / aspect_ratio)
                else:
                    new_height = JPEG_MAX_DIMENSION
                    new_width = int(new_height * aspect_ratio)
                
                
                img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
                logging.info(f"Resized to ({img.width}x{img.height})")
        
        
        if original_size <= actual_target:
            with open(image_path, 'rb') as f:
                return f.read()
        
        
        for quality in [85, 75, 65, 55, 45, 35, 25]:
            buffer = io.BytesIO()
            img.save(buffer, format='JPEG', quality=quality, optimize=True)
            compressed_data = buffer.getvalue()
            
            if len(compressed_data) <= actual_target:
                return compressed_data
        
        
        current_width, current_height = img.size
        for scale in [0.8, 0.6, 0.4, 0.2]:
            new_width = int(current_width * scale)
            new_height = int(current_height * scale)
            resized_img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
            
            buffer = io.BytesIO()
            resized_img.save(buffer, format='JPEG', quality=25, optimize=True)
            compressed_data = buffer.getvalue()
            
            if len(compressed_data) <= actual_target:
                return compressed_data
        
        
        buffer = io.BytesIO()
        resized_img.save(buffer, format='JPEG', quality=10, optimize=True)
        return buffer.getvalue()