import { EyeOutlined } from '@ant-design/icons'
import { useTranslation } from 'react-i18next'

import type { CustomTagProps } from '../CustomTag'
import CustomTag from '../CustomTag'

type Props = {
  size?: number
  showTooltip?: boolean
  showLabel?: boolean
} & Omit<CustomTagProps, 'size' | 'tooltip' | 'icon' | 'color' | 'children'>

export const VisionTag = ({ size, showTooltip, showLabel, ...restProps }: Props) => {
  const { t } = useTranslation()

  return (
    <CustomTag
      size={size}
      color="#00b96b"
      icon={<EyeOutlined style={{ fontSize: size }} />}
      tooltip={showTooltip ? t('models.type.vision') : undefined}
      {...restProps}>
      {showLabel ? t('models.type.vision') : ''}
    </CustomTag>
  )
}
