utils/persona.py

def predict_audience(words):

    if len(words) < 50:
        return "short_attention_gen_z"

    if any(w["text"].lower() in ["money", "success"] for w in words):
        return "motivated_entrepreneur"

    return "general"