88 lines
3.0 KiB
TypeScript
88 lines
3.0 KiB
TypeScript
import React, { Fragment } from 'react'
|
||
import { useSetAtom } from 'jotai'
|
||
import { Button } from '@uikit'
|
||
import { modelDownloadStateAtom } from '@helpers/atoms/DownloadState.atom'
|
||
import DownloadedModelTable from '@/_components/DownloadedModelTable'
|
||
import ActiveModelTable from '@/_components/ActiveModelTable'
|
||
import DownloadingModelTable from '@/_components/DownloadingModelTable'
|
||
import { useAtomValue } from 'jotai'
|
||
import { useGetDownloadedModels } from '@hooks/useGetDownloadedModels'
|
||
import { formatDownloadPercentage } from '@utils/converter'
|
||
import { LayoutGrid } from 'lucide-react'
|
||
import {
|
||
setMainViewStateAtom,
|
||
MainViewState,
|
||
} from '@helpers/atoms/MainView.atom'
|
||
|
||
const MyModelsScreen = () => {
|
||
const { downloadedModels } = useGetDownloadedModels()
|
||
const setMainViewState = useSetAtom(setMainViewStateAtom)
|
||
const modelDownloadStates = useAtomValue(modelDownloadStateAtom)
|
||
|
||
const downloadStates: DownloadState[] = []
|
||
for (const [, value] of Object.entries(modelDownloadStates)) {
|
||
downloadStates.push(value)
|
||
}
|
||
|
||
const isDownloadingFirstModel = downloadStates.length > 0
|
||
|
||
if (!downloadedModels || downloadedModels.length === 0)
|
||
return (
|
||
<div className="flex h-full items-center justify-center px-4">
|
||
<div className="text-center">
|
||
<LayoutGrid size={32} className="mx-auto text-accent/50" />
|
||
<div className="mt-4">
|
||
{isDownloadingFirstModel ? (
|
||
<div className="relative">
|
||
<div className="mt-4">
|
||
<h1 className="text-2xl font-bold leading-snug">
|
||
Donwloading your first model
|
||
</h1>
|
||
<p className="mt-1 text-muted-foreground">
|
||
{downloadStates[0].fileName} -{' '}
|
||
{formatDownloadPercentage(downloadStates[0].percent)}
|
||
</p>
|
||
</div>
|
||
</div>
|
||
) : (
|
||
<Fragment>
|
||
<h1 className="text-2xl font-bold leading-snug">{`Ups, You don't have a model.`}</h1>
|
||
<p className="mt-1 text-base text-muted-foreground">{`let’s download your first model`}</p>
|
||
<Button
|
||
className="mt-4"
|
||
themes="accent"
|
||
onClick={() => setMainViewState(MainViewState.ExploreModel)}
|
||
>
|
||
Explore Models
|
||
</Button>
|
||
</Fragment>
|
||
)}
|
||
</div>
|
||
</div>
|
||
</div>
|
||
)
|
||
|
||
return (
|
||
<div className="flex h-full w-full overflow-y-auto">
|
||
<div className="w-full p-5">
|
||
<h1
|
||
data-testid="testid-mymodels-header"
|
||
className="text-lg font-semibold"
|
||
>
|
||
My Models
|
||
</h1>
|
||
<p className="mt-2 text-gray-600 dark:text-gray-400">
|
||
You have <span>{downloadedModels.length}</span> models downloaded
|
||
</p>
|
||
<div>
|
||
{/* <ActiveModelTable /> */}
|
||
{/* <DownloadingModelTable /> */}
|
||
<DownloadedModelTable />
|
||
</div>
|
||
</div>
|
||
</div>
|
||
)
|
||
}
|
||
|
||
export default MyModelsScreen
|