src/utils/change_format.py

from PIL import Image
from io import BytesIO
import requests
import base64
from typing import Union

def change_format(image: Union[str, BytesIO], target_format: str) -> str:
    """
    Change the format of an image from a URL to the specified target format.
    
    Args:
        image_url: The URL of the input image.
        target_format: The desired output format (e.g., 'JPEG', 'PNG').
        
    Returns:
        The image converted to the target format as a base64-encoded string.
    """
    
    if not isinstance(image, BytesIO):
        response = requests.get(image, timeout=30)
        response.raise_for_status()
    
        # Open the image from bytes
        img = Image.open(BytesIO(response.content))
    
    # Convert the image to the target format
        output = BytesIO()
        img.save(output, format=target_format)
        output.seek(0)
        
        # Convert to base64 string for JSON serialization
        encoded_image = base64.b64encode(output.getvalue()).decode('utf-8')
        
        return encoded_image  # Return base64 encoded string that can be serialized to JSON
    else: 
        img = Image.open(image)
        
        output = BytesIO()
        img.save(output, format=target_format)
        output.seek(0)
        
        # Convert to base64 string for JSON serialization
        encoded_image = base64.b64encode(output.getvalue()).decode('utf-8')
        
        return encoded_image