import numpy as np
from image_processor import preprocess_image
import config

def compute_mse(original, mosaic):
    """
    Compute Mean Squared Error between original and mosaic.

    Parameters:
    -----------
    original : np.ndarray
        Original image
    mosaic : np.ndarray
        Mosaic image

    Returns:
    --------
    float : MSE value
    """ 
    # Preprocess image
    original = preprocess_image(original, config.mosaic_size)
    mse = np.mean((original.astype(float) - mosaic.astype(float)) ** 2)
    return mse