import type { FC } from 'react'
|
import React, { useState } from 'react'
|
import { ArrowUpRightIcon } from '@heroicons/react/24/outline'
|
import { useTranslation } from 'react-i18next'
|
import {
|
RiDeleteBinLine,
|
} from '@remixicon/react'
|
import { StatusItem } from '../../list'
|
import style from '../../style.module.css'
|
import s from './style.module.css'
|
import { SegmentIndexTag } from './common/segment-index-tag'
|
import cn from '@/utils/classnames'
|
import Confirm from '@/app/components/base/confirm'
|
import Switch from '@/app/components/base/switch'
|
import Divider from '@/app/components/base/divider'
|
import Indicator from '@/app/components/header/indicator'
|
import { formatNumber } from '@/utils/format'
|
import type { SegmentDetailModel } from '@/models/datasets'
|
|
const ProgressBar: FC<{ percent: number; loading: boolean }> = ({ percent, loading }) => {
|
return (
|
<div className={s.progressWrapper}>
|
<div className={cn(s.progress, loading ? s.progressLoading : '')}>
|
<div
|
className={s.progressInner}
|
style={{ width: `${loading ? 0 : (Math.min(percent, 1) * 100).toFixed(2)}%` }}
|
/>
|
</div>
|
<div className={loading ? s.progressTextLoading : s.progressText}>{loading ? null : percent.toFixed(2)}</div>
|
</div>
|
)
|
}
|
|
type DocumentTitleProps = {
|
extension?: string
|
name?: string
|
iconCls?: string
|
textCls?: string
|
wrapperCls?: string
|
}
|
|
const DocumentTitle: FC<DocumentTitleProps> = ({ extension, name, iconCls, textCls, wrapperCls }) => {
|
const localExtension = extension?.toLowerCase() || name?.split('.')?.pop()?.toLowerCase()
|
return <div className={cn('flex items-center justify-start flex-1', wrapperCls)}>
|
<div className={cn(s[`${localExtension || 'txt'}Icon`], style.titleIcon, iconCls)}></div>
|
<span className={cn('font-semibold text-lg text-gray-900 ml-1', textCls)}> {name || '--'}</span>
|
</div>
|
}
|
|
export type UsageScene = 'doc' | 'hitTesting'
|
|
type ISegmentCardProps = {
|
loading: boolean
|
detail?: SegmentDetailModel & { document: { name: string } }
|
contentExternal?: string
|
refSource?: {
|
title: string
|
uri: string
|
}
|
isExternal?: boolean
|
score?: number
|
onClick?: () => void
|
onChangeSwitch?: (segId: string, enabled: boolean) => Promise<void>
|
onDelete?: (segId: string) => Promise<void>
|
scene?: UsageScene
|
className?: string
|
archived?: boolean
|
embeddingAvailable?: boolean
|
}
|
|
const SegmentCard: FC<ISegmentCardProps> = ({
|
detail = {},
|
contentExternal,
|
isExternal,
|
refSource,
|
score,
|
onClick,
|
onChangeSwitch,
|
onDelete,
|
loading = true,
|
scene = 'doc',
|
className = '',
|
archived,
|
embeddingAvailable,
|
}) => {
|
const { t } = useTranslation()
|
const {
|
id,
|
position,
|
enabled,
|
content,
|
word_count,
|
hit_count,
|
index_node_hash,
|
answer,
|
} = detail as Required<ISegmentCardProps>['detail']
|
const isDocScene = scene === 'doc'
|
const [showModal, setShowModal] = useState(false)
|
|
const renderContent = () => {
|
if (answer) {
|
return (
|
<>
|
<div className='flex mb-2'>
|
<div className='mr-2 text-[13px] font-semibold text-gray-400'>Q</div>
|
<div className='text-[13px]'>{content}</div>
|
</div>
|
<div className='flex'>
|
<div className='mr-2 text-[13px] font-semibold text-gray-400'>A</div>
|
<div className='text-[13px]'>{answer}</div>
|
</div>
|
</>
|
)
|
}
|
|
if (contentExternal)
|
return contentExternal
|
|
return content
|
}
|
|
return (
|
<div
|
className={cn(
|
s.segWrapper,
|
(isDocScene && !enabled) ? 'bg-gray-25' : '',
|
'group',
|
!loading ? 'pb-4 hover:pb-[10px]' : '',
|
className,
|
)}
|
onClick={() => onClick?.()}
|
>
|
<div className={s.segTitleWrapper}>
|
{isDocScene
|
? <>
|
<SegmentIndexTag positionId={position} className={cn('w-fit group-hover:opacity-100', (isDocScene && !enabled) ? 'opacity-50' : '')} />
|
<div className={s.segStatusWrapper}>
|
{loading
|
? (
|
<Indicator
|
color="gray"
|
className="bg-gray-200 border-gray-300 shadow-none"
|
/>
|
)
|
: (
|
<>
|
<StatusItem status={enabled ? 'enabled' : 'disabled'} reverse textCls="text-gray-500 text-xs" />
|
{embeddingAvailable && (
|
<div className="hidden group-hover:inline-flex items-center">
|
<Divider type="vertical" className="!h-2" />
|
<div
|
onClick={(e: React.MouseEvent<HTMLDivElement, MouseEvent>) =>
|
e.stopPropagation()
|
}
|
className="inline-flex items-center"
|
>
|
<Switch
|
size='md'
|
disabled={archived || detail.status !== 'completed'}
|
defaultValue={enabled}
|
onChange={async (val) => {
|
await onChangeSwitch?.(id, val)
|
}}
|
/>
|
</div>
|
</div>
|
)}
|
</>
|
)}
|
</div>
|
</>
|
: (
|
score !== null
|
? (
|
<div className={s.hitTitleWrapper}>
|
<div className={cn(s.commonIcon, s.targetIcon, loading ? '!bg-gray-300' : '', '!w-3.5 !h-3.5')} />
|
<ProgressBar percent={score ?? 0} loading={loading} />
|
</div>
|
)
|
: null
|
)}
|
</div>
|
{loading
|
? (
|
<div className={cn(s.cardLoadingWrapper, s.cardLoadingIcon)}>
|
<div className={cn(s.cardLoadingBg)} />
|
</div>
|
)
|
: (
|
isDocScene
|
? <>
|
<div
|
className={cn(
|
s.segContent,
|
enabled ? '' : 'opacity-50',
|
'group-hover:text-transparent group-hover:bg-clip-text group-hover:bg-gradient-to-b',
|
)}
|
>
|
{renderContent()}
|
</div>
|
<div className={cn('group-hover:flex', s.segData)}>
|
<div className="flex items-center mr-6">
|
<div className={cn(s.commonIcon, s.typeSquareIcon)}></div>
|
<div className={s.segDataText}>{formatNumber(word_count)}</div>
|
</div>
|
<div className="flex items-center mr-6">
|
<div className={cn(s.commonIcon, s.targetIcon)} />
|
<div className={s.segDataText}>{formatNumber(hit_count)}</div>
|
</div>
|
<div className="grow flex items-center">
|
<div className={cn(s.commonIcon, s.bezierCurveIcon)} />
|
<div className={s.segDataText}>{index_node_hash}</div>
|
</div>
|
{!archived && embeddingAvailable && (
|
<div className='shrink-0 w-6 h-6 flex items-center justify-center rounded-md hover:bg-red-100 hover:text-red-600 cursor-pointer group/delete' onClick={(e) => {
|
e.stopPropagation()
|
setShowModal(true)
|
}}>
|
<RiDeleteBinLine className='w-[14px] h-[14px] text-gray-500 group-hover/delete:text-red-600' />
|
</div>
|
)}
|
</div>
|
</>
|
: <>
|
<div className="h-[140px] overflow-hidden text-ellipsis text-sm font-normal text-gray-800">
|
{renderContent()}
|
</div>
|
<div className={cn('w-full bg-gray-50 group-hover:bg-white')}>
|
<Divider />
|
<div className="relative flex items-center w-full pb-1">
|
<DocumentTitle
|
name={detail?.document?.name || refSource?.title || ''}
|
extension={(detail?.document?.name || refSource?.title || '').split('.').pop() || 'txt'}
|
wrapperCls='w-full'
|
iconCls="!h-4 !w-4 !bg-contain"
|
textCls="text-xs text-gray-700 !font-normal overflow-hidden whitespace-nowrap text-ellipsis"
|
/>
|
<div className={cn(s.chartLinkText, 'group-hover:inline-flex')}>
|
{isExternal ? t('datasetHitTesting.viewDetail') : t('datasetHitTesting.viewChart')}
|
<ArrowUpRightIcon className="w-3 h-3 ml-1 stroke-current stroke-2" />
|
</div>
|
</div>
|
</div>
|
</>
|
)}
|
{showModal
|
&& <Confirm
|
isShow={showModal}
|
title={t('datasetDocuments.segment.delete')}
|
confirmText={t('common.operation.sure')}
|
onConfirm={async () => { await onDelete?.(id) }}
|
onCancel={() => setShowModal(false)}
|
/>
|
}
|
</div>
|
)
|
}
|
|
export default SegmentCard
|
