# preprocessors.py
import pandas as pd

def add_store_age(df):
    df = df.copy()
    df['Store_Age'] = 2025 - df['Store_Established_Year']
    df = df.drop(columns=['Store_Established_Year'])
    return df

def map_ordered_features(X):
    sugar_order_map = {'No Sugar': 0, 'Low Sugar': 1, 'Regular': 2}
    size_order_map = {'Small': 0, 'Medium': 1, 'High': 2}
    city_order_map = {'Tier 3': 0, 'Tier 2': 1, 'Tier 1': 2}

    X = X.copy()
    X['Product_Sugar_Content'] = X['Product_Sugar_Content'].map(sugar_order_map).astype(int)
    X['Store_Size'] = X['Store_Size'].map(size_order_map).astype(int)
    X['Store_Location_City_Type'] = X['Store_Location_City_Type'].map(city_order_map).astype(int)
    return X