from src.cocktails.pipeline.get_affect2affective_cluster import get_affect2affective_cluster
from src.cocktails.pipeline.cocktail2affect import cocktail2affect

def get_cocktail2affective_cluster():
    find_cluster = get_affect2affective_cluster()
    def cocktail2affect_cluster(cocktail_rep):
        affective_coordinates, _ = cocktail2affect(cocktail_rep)
        return find_cluster(affective_coordinates)
    return cocktail2affect_cluster