client/src/components/Input/Generations/Button.tsx

import { cn, removeFocusOutlines } from '~/utils/';

export default function Button({
  type = 'regenerate',
  children,
  onClick,
  className = '',
}: {
  type?: 'regenerate' | 'continue' | 'stop';
  children: React.ReactNode;
  onClick: (e: React.MouseEvent<HTMLButtonElement>) => void;
  className?: string;
}) {
  return (
    <button
      data-testid={`${type}-generation-button`}
      className={cn(
        'custom-btn btn-neutral relative -z-0 whitespace-nowrap border-0 md:border',
        removeFocusOutlines,
        className,
      )}
      onClick={onClick}
    >
      <div className="flex w-full items-center justify-center gap-2">{children}</div>
    </button>
  );
}