.ai-badge {
    background: purple;
    border-radius: 5px;
    border: 1px solid purple;
    color: white;
    font-weight: bold;
    padding: 1px 5px;
}