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