settings.py

# Format: rgb_generation_setting-rgba-rgba_cutout_name


def parse_rgb_generation_setting(rgb_generation_setting: str, setting_dict: dict):
    if 'centering' not in rgb_generation_setting:
        setting_dict['centering'] = 'False'

    if 'no_exc' in rgb_generation_setting:
        setting_dict['exclude_generic_nouns'] = 'False'

    if 'suffix' in rgb_generation_setting:
        setting_dict['use_suffix'] = 'True'

    return setting_dict



def parse_setting(setting: str):
    setting_dict = {
        'centering': 'True',
        'use_neg_prompt': 'True',
        'exclude_generic_nouns': 'True',
        'cutout_model': 'grabcut',
        'use_suffix': 'False'
    }

    if 'rgba' in setting:
        rgb_generation_setting, rgba_cutout_setting = setting.rsplit('-rgba-')
    else:
        rgb_generation_setting = setting
        rgba_cutout_setting = None
    
    setting_dict = parse_rgb_generation_setting(rgb_generation_setting, setting_dict)
    if rgba_cutout_setting:
        if 'alfie' in rgba_cutout_setting:
            setting_dict['cutout_model'] = 'grabcut'
        elif 'vit_matte' in rgba_cutout_setting:
            setting_dict['cutout_model'] = 'vit-matte'
        else:
            raise ValueError(f"Invalid cutout model: {rgba_cutout_setting}. Options: ['grabcut', 'vit_matte']")

    return setting_dict