"""
    utils.py - Utility functions for the project.
"""

def postprocess(text:str):
    """
    postprocess - remove common values in scraped dataset

    Args:
        text (str): the text to postprocess
    """

    replacements = {
        "ENA":"<COMPANY>",
        "Enron":"<COMPANY>",
        "Enron Corporation":"<COMPANY>",
        "Sony Pictures Entertainment":"<COMPANY>",
        "Columbia Pictures":"<COMPANY>",
        "Sony":"<COMPANY>",
        "Columbia":"<COMPANY>",
        "Hillary":"Jane",
        "Clinton":"Smith",
        "Amy":"Jane",
        "Sara":"Jane",
        "Harambe":"Jane",
        "Pascal":"<PERSON>",
    }

    # replace common values, also check lowercase
    for k, v in replacements.items():
        text = text.replace(k, v)
        text = text.replace(k.lower(), v)

    return text

def clear(**kwargs):
   """for use with buttons"""

   return ""