From 334dc00372d2d10f7fde5dd480121a8862e10753 Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Thu, 20 Mar 2025 21:15:48 +0530
Subject: [PATCH] add request exception handling and toast notifications for
 error

---
 client/src/Pages/UploadPage.tsx      | 35 +++++++++++++++++-----------
 client/src/components/Toast.tsx      | 26 +++++++++++++++++++++
 client/src/contexts/MediaContext.tsx |  8 ++++++-
 3 files changed, 54 insertions(+), 15 deletions(-)
 create mode 100644 client/src/components/Toast.tsx

diff --git a/client/src/Pages/UploadPage.tsx b/client/src/Pages/UploadPage.tsx
index 855d384..6a8c241 100644
--- a/client/src/Pages/UploadPage.tsx
+++ b/client/src/Pages/UploadPage.tsx
@@ -1,6 +1,7 @@
 import React, { useState, useEffect } from "react";
 import { useNavigate } from "react-router-dom";
 import Logs from "../components/Logs"
+import Toast from "../components/Toast";
 import axios from "axios";
 import {
   Typography,
@@ -23,13 +24,17 @@ function UploadPage() {
   const navigate = useNavigate();
   const theme = useTheme();
   const { socket, isConnected } = useWebSocket();
-  const { setMediaFile, setResponse, setLogs } = useMediaContext();
+  const { setMediaFile, setResponse, setLogs, setToastOpen, setToastMessage } = useMediaContext();
   const [file, setFile] = useState<File | null>(null);
   const [isDragging, setIsDragging] = useState(false);
-  const [fileError, setFileError] = useState("");
   const [upload, setUpload] = useState(false);
   const [inputEnabled, setInputEnabled] = useState(false);
 
+  const showErrorToast = (message: string) => {
+    setToastMessage(message);
+    setToastOpen(true);
+  }
+
   useEffect(() => {
     const startLogs = async () => {
       setLogs([formatLogMessage("Initializing freqsplit")]);
@@ -102,7 +107,6 @@ function UploadPage() {
 
   const handleFileChange = (e: React.ChangeEvent<HTMLInputElement>) => {
     setUpload(false);
-    setFileError("");
     setResponse({
       audio_class: "",
       file_uuid: "",
@@ -115,7 +119,7 @@ function UploadPage() {
     
     if (selectedFile) {
       if (selectedFile.size > maxSize) {
-        setFileError("Max file size is 100MB!");
+        showErrorToast("Max file size is 100MB!");
         setFile(null);
         setUpload(false);
         e.target.value = "";
@@ -128,13 +132,11 @@ function UploadPage() {
   };
 
   const validateAndSetFile = (file: File | null) => {
-    setFileError("");
-
     if (!file) return;
 
     const fileType = file.type;
     if (!fileType.includes("audio") && !fileType.includes("video")) {
-      setFileError("Please upload only audio or video files.");
+      showErrorToast("Please upload only audio or video files.");
       return;
     }
 
@@ -151,7 +153,7 @@ function UploadPage() {
       }); //
       navigate('/preview');
     } else {
-      setFileError("Please upload a file to continue.");
+      showErrorToast("Please upload a file to continue");
     }
   };
 
@@ -198,6 +200,16 @@ function UploadPage() {
         setLogs((prevLogs) => [...prevLogs, formatLogMessage(`freqsplit/input: audio_class: ${res.data.audio_class}`)])
       }
     } catch (error) {
+      showErrorToast("Failed to upload file");
+      setUpload(false);
+      setResponse({
+        audio_class: "",
+        file_uuid: "",
+        sr: 0,
+        spectrogram: "",
+        spec_sr: 0
+      });
+      setFile(null);
       console.error("Upload failed:", error);
     }
   };
@@ -255,12 +267,6 @@ function UploadPage() {
           )}
         </Box>
 
-        {fileError && (
-          <Typography variant="body2" color="error" sx={{ mt: 1 }}>
-            {fileError}
-          </Typography>
-        )}
-
         <Box sx={{ mt: 4, display: "flex", justifyContent: "space-between" }}>
           <Button
             variant="outlined"
@@ -280,6 +286,7 @@ function UploadPage() {
         </Box>
       </Paper>
       <Logs />
+      <Toast />
     </Container>
   );
 }
diff --git a/client/src/components/Toast.tsx b/client/src/components/Toast.tsx
new file mode 100644
index 0000000..9d38961
--- /dev/null
+++ b/client/src/components/Toast.tsx
@@ -0,0 +1,26 @@
+import { Snackbar, Alert } from "@mui/material";
+import { useMediaContext } from "../contexts/MediaContext";
+
+const Toast: React.FC = () => {
+    const { toastOpen, setToastOpen, toastMessage } = useMediaContext();
+    return (
+        <Snackbar
+                open={toastOpen}
+                autoHideDuration={4000}
+                onClose={() => {
+                    setToastOpen(false);
+                }}
+                anchorOrigin={{ vertical: "top", horizontal: "right" }}
+              >
+                <Alert
+                  onClose={() => setToastOpen(false)}
+                  severity="error"
+                  sx={{ width: "100%" }}
+                >
+                  {toastMessage}
+                </Alert>
+              </Snackbar>
+      );
+}
+
+export default Toast
\ No newline at end of file
diff --git a/client/src/contexts/MediaContext.tsx b/client/src/contexts/MediaContext.tsx
index af18d00..2f282a5 100644
--- a/client/src/contexts/MediaContext.tsx
+++ b/client/src/contexts/MediaContext.tsx
@@ -13,6 +13,10 @@ interface MediaContextType {
   setDownloadedFileSpectrogram: (spectrogram: {spectrogram: string, spec_sr: number}) => void;
   logs: string[];
   setLogs: React.Dispatch<React.SetStateAction<string[]>>;
+  toastOpen: boolean;
+  setToastOpen: ( value: boolean) => void;
+  toastMessage: string;
+  setToastMessage: ( message: string) => void;
 }
 
 
@@ -34,9 +38,11 @@ export const MediaProvider: React.FC<{ children: React.ReactNode }> = ({ childre
     spec_sr: 0
   });
   const [logs, setLogs] = useState<MediaContextType["logs"]>([""]);
+  const [toastOpen, setToastOpen] = useState<MediaContextType["toastOpen"]>(false);
+  const [toastMessage, setToastMessage] = useState<MediaContextType["toastMessage"]>("");
 
   return (
-    <MediaContext.Provider value={{ mediaFile, setMediaFile, response, setResponse, extractedFiles, setExtractedFiles, downloadedFileURL, setDownloadedFileURL, downloadedFileSpectrogram, setDownloadedFileSpectrogram, logs, setLogs }}>
+    <MediaContext.Provider value={{ mediaFile, setMediaFile, response, setResponse, extractedFiles, setExtractedFiles, downloadedFileURL, setDownloadedFileURL, downloadedFileSpectrogram, setDownloadedFileSpectrogram, logs, setLogs, toastOpen, setToastOpen, toastMessage, setToastMessage }}>
       {children}
     </MediaContext.Provider>
   );