fix: hub button download (#4742)

* fix: hub button download and use

* chore: fix linter

* fix: remove the button download on top of model page

web/containers/ModelDownloadButton/index.tsx

@@ -15,6 +15,7 @@ import ModalCancelDownload from '../ModalCancelDownload'
 import { mainViewStateAtom } from '@/helpers/atoms/App.atom'
 import { assistantsAtom } from '@/helpers/atoms/Assistant.atom'
 
+import { serverEnabledAtom } from '@/helpers/atoms/LocalServer.atom'
 import {
   downloadedModelsAtom,
   getDownloadingModelAtom,
@@ -25,10 +26,12 @@ interface Props {
   theme?: 'primary' | 'ghost' | 'icon' | 'destructive' | undefined
   variant?: 'solid' | 'soft' | 'outline' | undefined
   className?: string
+  hideProgress?: boolean
 }
-const ModelDownloadButton = ({ id, theme, variant, className }: Props) => {
+const ModelDownloadButton = ({ id, theme, className, hideProgress }: Props) => {
   const { downloadModel } = useDownloadModel()
   const downloadingModels = useAtomValue(getDownloadingModelAtom)
+  const serverEnabled = useAtomValue(serverEnabledAtom)
   const downloadedModels = useAtomValue(downloadedModelsAtom)
   const assistants = useAtomValue(assistantsAtom)
   const setMainViewState = useSetAtom(mainViewStateAtom)
@@ -62,7 +65,6 @@ const ModelDownloadButton = ({ id, theme, variant, className }: Props) => {
   const defaultButton = (
     <Button
       theme={theme ? theme : 'primary'}
-      // variant={variant ? variant : 'solid'}
       className={twMerge('min-w-[70px]', className)}
       onClick={(e) => {
         e.stopPropagation()
@@ -72,7 +74,10 @@ const ModelDownloadButton = ({ id, theme, variant, className }: Props) => {
       Download
     </Button>
   )
-  const downloadingButton = <ModalCancelDownload modelId={id} />
+  const downloadingButton = !hideProgress && (
+    <ModalCancelDownload modelId={id} />
+  )
+
   const downloadedButton = (
     <Tooltip
       trigger={
@@ -82,11 +87,13 @@ const ModelDownloadButton = ({ id, theme, variant, className }: Props) => {
           variant="outline"
           theme="ghost"
           className="min-w-[70px]"
+          disabled={serverEnabled}
         >
           Use
         </Button>
       }
       content="Threads are disabled while the server is running"
+      disabled={!serverEnabled}
     />
   )
   return (

web/screens/Hub/ModelPage/index.tsx

@@ -70,9 +70,7 @@ const ModelPage = ({ model, onGoBack }: Props) => {
                   : model.metadata.id}
               </span>
               <div className="inline-flex items-center space-x-2">
-                {model.type !== 'cloud' ? (
-                  <ModelDownloadButton id={model.models?.[0].id} />
-                ) : (
+                {model.type === 'cloud' && (
                   <>
                     {!model.metadata?.apiKey?.length ? (
                       <Button