utils/utils.py

def identify_triggers(outputs):
    # Initialize a set to store identified triggers
    identified_triggers = set()

    # Iterate over the model outputs
    for response_text in outputs:
        # Check if the response contains a positive indication of the category
        if "yes" in response_text.lower():
            # Extract the category from the response_text
            category = extract_category(response_text)
            if category:
                identified_triggers.add(category)
    
    # Convert the set of identified triggers to a list and return it
    return list(identified_triggers)

def extract_category(response_text):
    # Define trigger categories
    trigger_categories = [
        "Violence",
        "Self-Harm",
        "Death",
        "Substance Use"
        "Sexual Content"
        "Sexual Abuse",
        "Gun Use",
        "Gore",
        "Vomit", 
        "Mental Health Issues"
        "Animal Cruelty"
    ]
    
    # Check if any category is present in the response_text
    for category in trigger_categories:
        if category.lower() in response_text.lower():
            return category
    # Return None if no category is found
    return None