jan/web/screens/Settings/ImportingModelModal/ImportingModelItem.tsx
2024-08-05 14:31:31 +07:00

53 lines
1.5 KiB
TypeScript

import { useCallback, useMemo } from 'react'
import { ImportingModel } from '@janhq/core'
import { AlertCircle } from 'lucide-react'
import { toGibibytes } from '@/utils/converter'
import ImportInProgressIcon from '../ImportInProgressIcon'
import ImportSuccessIcon from '../ImportSuccessIcon'
type Props = {
model: ImportingModel
}
const ImportingModelItem = ({ model }: Props) => {
const onEditModelInfoClick = useCallback(() => {
// setEditingModelId(model.importId)
// setImportModelStage('EDIT_MODEL_INFO')
}, [])
const onDeleteModelClick = useCallback(() => {}, [])
const displayStatus = useMemo(() => {
if (model.status === 'FAILED') {
return 'Failed'
} else {
return toGibibytes(model.size)
}
}, [model.status, model.size])
return (
<div className="flex w-full flex-row items-center space-x-3 rounded-lg border border-[hsla(var(--app-border))] px-4 py-3">
<p className="line-clamp-1 flex-1 font-semibold text-[hsla(var(--text-secondary))]">
{model.name}
</p>
<p className="text-[hsla(var(--text-secondary))]">{displayStatus}</p>
{model.status === 'IMPORTED' && (
<ImportSuccessIcon onEditModelClick={onEditModelInfoClick} />
)}
{(model.status === 'IMPORTING' || model.status === 'PREPARING') && (
<ImportInProgressIcon
percentage={model.percentage ?? 0}
onDeleteModelClick={onDeleteModelClick}
/>
)}
{model.status === 'FAILED' && <AlertCircle size={24} />}
</div>
)
}
export default ImportingModelItem