import type { FC } from 'react'
import React, { useState } from 'react'
import { useTranslation } from 'react-i18next'
import TopKItem from '@/app/components/base/param-item/top-k-item'
import ScoreThresholdItem from '@/app/components/base/param-item/score-threshold-item'

type RetrievalSettingsProps = {
  topK: number
  scoreThreshold: number
  onChange: (data: { top_k?: number; score_threshold?: number }) => void
}

const RetrievalSettings: FC<RetrievalSettingsProps> = ({ topK, scoreThreshold, onChange }) => {
  const [scoreThresholdEnabled, setScoreThresholdEnabled] = useState(false)
  const { t } = useTranslation()
  return (
    <div className='flex flex-col gap-2 self-stretch'>
      <div className='flex h-7 pt-1 flex-col gap-2 self-stretch'>
        <label className='text-text-secondary system-sm-semibold'>{t('dataset.retrievalSettings')}</label>
      </div>
      <div className='flex gap-4 self-stretch'>
        <div className='flex flex-col gap-1 flex-grow'>
          <TopKItem
            className='grow'
            value={topK}
            onChange={(_key, v) => onChange({ top_k: v })}
            enable={true}
          />
        </div>
        <div className='flex flex-col gap-1 flex-grow'>
          <ScoreThresholdItem
            className='grow'
            value={scoreThreshold}
            onChange={(_key, v) => onChange({ score_threshold: v })}
            enable={scoreThresholdEnabled}
            hasSwitch={true}
            onSwitchChange={(_key, v) => setScoreThresholdEnabled(v)}
          />
        </div>
      </div>
    </div>
  )
}

export default RetrievalSettings
