import {
  AgentThinkingBadge,
  AgentsThinkingBadge,
  PALETTES,
} from "@/components/ui/grok-agent-thinking-indicator";

export default function Demo() {
  return (
    <div className="flex min-h-[320px] w-full flex-col items-start justify-center gap-4 bg-background p-8 pl-16 md:pl-32">
      {/* Default single-agent badge */}
      <AgentThinkingBadge />

      {/* Custom labels and palettes */}
      <AgentThinkingBadge label="Drafting reply" palette={PALETTES[1]} />
      <AgentThinkingBadge label="Searching codebase" palette={PALETTES[2]} />

      {/* Multi-agent variants */}
      <AgentsThinkingBadge />
      <AgentsThinkingBadge label="3 agents working" count={3} />
    </div>
  );
}