From ef17f7574c42e6987765ce54b461fabb7bf73a17 Mon Sep 17 00:00:00 2001
From: keisuke-umezawa <keisuke.umezawa@gmail.com>
Date: Thu, 6 Jun 2024 11:23:16 +0900
Subject: [PATCH] Add error handling

---
 .../ts/components/Artifact/TableArtifactViewer.tsx   | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/optuna_dashboard/ts/components/Artifact/TableArtifactViewer.tsx b/optuna_dashboard/ts/components/Artifact/TableArtifactViewer.tsx
index e810536f..c0338259 100644
--- a/optuna_dashboard/ts/components/Artifact/TableArtifactViewer.tsx
+++ b/optuna_dashboard/ts/components/Artifact/TableArtifactViewer.tsx
@@ -1,6 +1,7 @@
 import ClearIcon from "@mui/icons-material/Clear"
 import { Box, Modal, useTheme } from "@mui/material"
 import IconButton from "@mui/material/IconButton"
+import { useSnackbar } from "notistack"
 import Papa from "papaparse"
 import React, { useState, useEffect, ReactNode } from "react"
 import { DataGrid } from "../DataGrid"
@@ -24,11 +25,18 @@ export const TableArtifactViewer: React.FC<TableArtifactViewerProps> = (
   props
 ) => {
   const [data, setData] = useState<Data[]>([])
+  const { enqueueSnackbar } = useSnackbar()
 
   useEffect(() => {
     const handleFileChange = async () => {
-      const loadedData = await loadCSV(props)
-      setData(loadedData)
+      try {
+        const loadedData = await loadCSV(props)
+        setData(loadedData)
+      } catch (error: unknown) {
+        enqueueSnackbar("Failed to load the csv file.", {
+          variant: "error",
+        })
+      }
     }
     handleFileChange()
   }, [props])