utils/ab_generator.py

from .strategist import rewrite_hook

def generate_variants(segment):

    text = " ".join([w["text"] for w in segment])

    return [
        rewrite_hook(text),
        f"🔥 {text}",
        f"Did you know? {text}",
    ]