import type { FC, PropsWithChildren } from 'react'
import Image from 'next/image'
import SelectionMod from './assets/selection-mod-nocolor.svg'
import type { QA } from '@/models/datasets'

export type ChunkLabelProps = {
  label: string
  characterCount: number
}

export const ChunkLabel: FC<ChunkLabelProps> = (props) => {
  const { label, characterCount } = props
  return <div className='flex items-center text-text-tertiary text-xs font-medium'>
    <Image src={SelectionMod} alt="Selection Mod" width={10} height={10} />
    <p className='flex gap-2 ml-0.5'><span>
      {label}
    </span>
    <span>
      ·
    </span>
    <span>
      {`${characterCount} characters`}
    </span></p>
  </div>
}

export type ChunkContainerProps = ChunkLabelProps & PropsWithChildren

export const ChunkContainer: FC<ChunkContainerProps> = (props) => {
  const { label, characterCount, children } = props
  return <div className='space-y-2'>
    <ChunkLabel label={label} characterCount={characterCount} />
    <p className='text-text-secondary text-sm tracking-[-0.0005em]'>
      {children}
    </p>
  </div>
}

export type QAPreviewProps = {
  qa: QA
}

export const QAPreview: FC<QAPreviewProps> = (props) => {
  const { qa } = props
  return <div className='space-y-2'>
    <div className='flex gap-1 items-start'>
      <label className='text-text-tertiary text-[13px] font-medium'>Q</label>
      <p className='text-text-secondary tracking-[-0.0005em]'>{qa.question}</p>
    </div>
    <div className='flex gap-1 items-start'>
      <label className='text-text-tertiary text-[13px] font-medium'>A</label>
      <p className='text-text-secondary tracking-[-0.0005em]'>{qa.answer}</p>
    </div>
  </div>
}