dify/web/app/components/datasets/documents/detail/completed/segment-card/chunk-content.tsx

import React, { type FC } from 'react'
import cn from '@/utils/classnames'
import { useSegmentListContext } from '..'
import { Markdown } from '@/app/components/base/markdown'

type ChunkContentProps = {
  detail: {
    answer?: string
    content: string
    sign_content: string
  }
  isFullDocMode: boolean
  className?: string
}

const ChunkContent: FC<ChunkContentProps> = ({
  detail,
  isFullDocMode,
  className,
}) => {
  const { answer, content, sign_content } = detail
  const isCollapsed = useSegmentListContext(s => s.isCollapsed)

  if (answer) {
    return (
      <div className={className}>
        <div className='flex gap-x-1'>
          <div className='w-4 shrink-0 text-[13px] font-medium leading-[20px] text-text-tertiary'>Q</div>
          <div
            className={cn('body-md-regular text-text-secondary',
              isCollapsed ? 'line-clamp-2' : 'line-clamp-20',
            )}>
            {content}
          </div>
        </div>
        <div className='flex gap-x-1'>
          <div className='w-4 shrink-0 text-[13px] font-medium leading-[20px] text-text-tertiary'>A</div>
          <div className={cn('body-md-regular text-text-secondary',
            isCollapsed ? 'line-clamp-2' : 'line-clamp-20',
          )}>
            {answer}
          </div>
        </div>
      </div>
    )
  }
  return <Markdown
    className={cn('!mt-0.5 !text-text-secondary',
      isFullDocMode ? 'line-clamp-3' : isCollapsed ? 'line-clamp-2' : 'line-clamp-20',
      className,
    )}
    content={sign_content || content || ''}
    customDisallowedElements={['input']}
  />
}

export default React.memo(ChunkContent)