import pandas as pd
import joblib
import config

def load_model():
    return joblib.load(config.MODEL_PATH)

def load_ppp_rpp():
    ppp = pd.read_csv(config.PPP_PATH, index_col=0)
    rpp = pd.read_csv(config.RPP_PATH, index_col=0)
    return ppp, rpp

def load_lambdas():
    df = pd.read_csv(config.LAMBDAS_PATH)
    return dict(zip(df['key'], df['value']))

def load_locations():
    countries = pd.read_csv(config.COUNTRIES_PATH)
    states = pd.read_csv(config.STATES_PATH)

    countries_dict = dict(zip(countries['alpha-2'], countries['name']))
    states_dict = dict(zip(states['Abbreviation'], states['State']))

    return countries_dict, states_dict