chore: added max depth to model files recursion

hotosm · Nov 25, 2024 · a603e6d · a603e6d
1 parent b2e440b
commit a603e6d
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 5 deletions.
diff --git a/frontend/src/features/models/components/dialogs/model-files-dialog.tsx b/frontend/src/features/models/components/dialogs/model-files-dialog.tsx
@@ -20,6 +20,7 @@ const ModelFilesDialog: React.FC<TrainingAreaDialogProps> = ({
       closeDialog={closeDialog}
       label={APP_CONTENT.models.modelsDetailsCard.modelFilesDialog.dialogTitle}
     >
+      <p className="text-dark text-body-2base px-2 mb-4">{APP_CONTENT.models.modelsDetailsCard.modelFilesDialog.dialogDescription}</p>
       {isOpened && (
         <DirectoryTree
           trainingId={trainingId}

diff --git a/frontend/src/features/models/components/directory-tree.tsx b/frontend/src/features/models/components/directory-tree.tsx
@@ -120,28 +120,41 @@ const DirectoryTree: React.FC<DirectoryTreeProps> = ({
 
   const fetchDirectoryRecursive = async (
     currentDirectory: string = "",
+    currentDepth: number = 0,
+    maxDepth: number = 2
   ): Promise<any> => {
+    if (currentDepth >= maxDepth) {
+
+      return {};
+    }
+
     const data = await fetchDirectoryData(currentDirectory);
     if (!data) return {};
 
     const { dir, file } = data;
-    const subdirectories = dir
-      ? await Promise.all(
+
+    const subdirectories =
+      dir && currentDepth < maxDepth
+        ? await Promise.all(
           Object.keys(dir).map(async (key: string) => {
             const fullPath = currentDirectory
               ? `${currentDirectory}/${key}`
               : key;
-            const subDirData = await fetchDirectoryRecursive(fullPath);
+            const subDirData = await fetchDirectoryRecursive(
+              fullPath,
+              currentDepth + 1,
+              maxDepth
+            );
             return {
               [key]: {
                 ...subDirData,
                 size: dir[key]?.size || 0,
                 length: dir[key]?.len || 0,
               },
             };
-          }),
+          })
         )
-      : [];
+        : [];
 
     return {
       dir: Object.assign({}, ...subdirectories),

diff --git a/frontend/src/utils/content.ts b/frontend/src/utils/content.ts
@@ -506,6 +506,7 @@ export const APP_CONTENT = {
       modelFilesDialog: {
         rootDirectory: "Root Directory",
         dialogTitle: "Model Files",
+        dialogDescription: "Click to download each file...",
         error: "Error loading directories.",
       },
       trainingInfoDialog: {