'use client'
import type { FC } from 'react'
import type { AgentIteration } from '@/models/log'
import { cn } from '@langgenius/dify-ui/cn'
import { useTranslation } from 'react-i18next'
import Divider from '@/app/components/base/divider'
import ToolCall from './tool-call'

type Props = {
  isFinal: boolean
  index: number
  iterationInfo: AgentIteration
}

const Iteration: FC<Props> = ({ iterationInfo, isFinal, index }) => {
  const { t } = useTranslation()

  return (
    <div className={cn('px-4 py-2')}>
      <div className="flex items-center">
        {isFinal && (
          <div className="mr-3 shrink-0 text-xs leading-[18px] font-semibold text-text-tertiary">{t('agentLogDetail.finalProcessing', { ns: 'appLog' })}</div>
        )}
        {!isFinal && (
          <div className="mr-3 shrink-0 text-xs leading-[18px] font-semibold text-text-tertiary">{`${t('agentLogDetail.iteration', { ns: 'appLog' }).toUpperCase()} ${index}`}</div>
        )}
        <Divider bgStyle="gradient" className="mx-0 h-px grow" />
      </div>
      <ToolCall
        isLLM
        isFinal={isFinal}
        tokens={iterationInfo.tokens}
        observation={iterationInfo.tool_raw.outputs}
        finalAnswer={iterationInfo.thought}
        toolCall={{
          status: 'success',
          tool_icon: null,
        }}
      />
      {iterationInfo.tool_calls.map((toolCall, index) => (
        <ToolCall
          isLLM={false}
          key={index}
          toolCall={toolCall}
        />
      ))}
    </div>
  )
}

export default Iteration