added support for .svg

ModelEngineering · Sep 11, 2023 · 335079c · 335079c
1 parent 1be49d1
commit 335079c
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/UI/components/model/modelDataLoader.tsx b/src/UI/components/model/modelDataLoader.tsx
@@ -1,7 +1,7 @@
 import { useEffect } from "react";
 
 // Define the supported file types for the model files
-const SUPPORTED_FILE_TYPES = [".png", ".jpg", ".ant", ".m", ".py", ".jl", ".r", ".txt"];
+const SUPPORTED_FILE_TYPES = [".png", ".svg", ".jpg", ".ant", ".m", ".py", ".jl", ".r", ".txt"];
 
 // Define the ModelDataLoader component
 const ModelDataLoader = ({

diff --git a/src/UI/components/model/modelDisplay.tsx b/src/UI/components/model/modelDisplay.tsx
@@ -4,7 +4,7 @@ import ImageDisplay from "./imageDisplay";
 // Define the supported file types for code files
 const SUPPORTED_FILE_TYPES = [".ant", ".m", ".py", ".jl", ".r", ".txt"];
 // Define the supported image file types
-const SUPPORTED_IMAGE_TYPES = [".png", ".jpg"];
+const SUPPORTED_IMAGE_TYPES = [".png", ".jpg", ".svg"];
 
 // Define the ModelDisplay component
 const ModelDisplay = ({

diff --git a/src/UI/components/model/modelFileLoader.tsx b/src/UI/components/model/modelFileLoader.tsx
@@ -40,7 +40,7 @@ const ModelFileLoader = ({
     };
 
     // Determine whether the file is an image or not
-    if (fileName.endsWith(".png") || fileName.endsWith(".jpg")) {
+    if (fileName.endsWith(".png") || fileName.endsWith(".jpg") || fileName.endsWith(".svg")) {
       fetchImage(); // Fetch and display image files
     } else {
       getFile(); // Fetch and display other types of files