Update the error message and automatically download a dataset when a part of a recipe is not found

src/renderer/components/Experiment/Train/ImportRecipeModal.tsx

@@ -110,11 +110,30 @@ export default function ImportRecipeModal({ open, setOpen, mutate }) {
         alert("Warning: This recipe does not have an associated dataset")
       } else {
         let msg = "";
-        if (!response.model.downloaded) {
-          msg += "Download model " + response.model.path
-        }
         if (!response.dataset.downloaded) {
-          msg += "Download dataset " + response.dataset.path
+          msg += "\n Download dataset " + response.dataset.path + ", this will be done automatically";
+          fetch(chatAPI.Endpoints.Dataset.Download(response.dataset.path))
+            .then((response) => {
+            if (!response.ok) {
+                console.log(response);
+                throw new Error(`HTTP Status: ${response.status}`);
+              }
+              return response.json();
+            })
+            .catch((error) => {
+              alert('Download failed:\n' + error);
+            });
+        }
+        if (!response.model.downloaded) {
+          msg += "\n Download model " + response.model.path + ", this will be done automatically";
+          chatAPI.downloadModelFromHuggingFace(response.model.path)
+          .then((response) => {
+            if (response.status == "error") {
+              console.log(response);
+              throw new Error(`${response.message}`);
+            }
+            return response;
+          })
         }
         if (msg) {
           const alert_msg = "Warning: To use this recipe you will need to: " + msg

src/renderer/components/Experiment/Train/TrainLoRA.tsx

@@ -46,6 +46,8 @@ import ImportRecipeModal from './ImportRecipeModal';
 
 import dayjs from 'dayjs';
 import relativeTime from 'dayjs/plugin/relativeTime';
+import CurrentDownloadBox from 'renderer/components/currentDownloadBox';
+import DownloadProgressBox from 'renderer/components/Shared/DownloadProgressBox';
 dayjs.extend(relativeTime);
 var duration = require('dayjs/plugin/duration');
 dayjs.extend(duration);
@@ -118,6 +120,15 @@ export default function TrainLoRA({ experimentInfo }) {
     refreshInterval: 2000,
   });
 
+  const {
+    data: downloadJobs,
+    error: downloadJobsError,
+    isLoading: downloadJobsIsLoading,
+    mutate: downloadJobsMutate,
+  } = useSWR(chatAPI.Endpoints.Jobs.GetJobsOfType('DOWNLOAD_MODEL', 'RUNNING'), fetcher, {
+    refreshInterval: 2000,
+  });
+
   //Fetch available training plugins
   const {
     data: pluginsData,
@@ -173,6 +184,7 @@ export default function TrainLoRA({ experimentInfo }) {
           overflow: 'hidden',
         }}
       >
+        <DownloadProgressBox jobId={downloadJobs[0]?.id} assetName={downloadJobs[0]?.job_data.model}/>
         {/* <Typography level="h1">Train</Typography> */}
         <Stack direction="row" justifyContent="space-between" gap={2}>
           <Typography level="title-md" startDecorator={<GraduationCapIcon />}>

src/renderer/components/Shared/DownloadProgressBox.tsx

@@ -8,6 +8,7 @@ import {
 } from '@mui/joy';
 
 import {
+    clamp,
     formatBytes,
 } from '../../lib/utils';
 import * as chatAPI from '../../lib/transformerlab-api-sdk';