From c22202dbe636dc1190553e1d07d90cafb6d479bc Mon Sep 17 00:00:00 2001
From: sumn2u <sumn2u@gmail.com>
Date: Tue, 18 Jun 2024 08:00:46 -0500
Subject: [PATCH] fixed images not showing issue on refresh for multiple
 uploads

---
 client/src/Annotation/index.jsx | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/client/src/Annotation/index.jsx b/client/src/Annotation/index.jsx
index 4965b85..4e84785 100644
--- a/client/src/Annotation/index.jsx
+++ b/client/src/Annotation/index.jsx
@@ -4,6 +4,7 @@ import SetupPage from "../SetupPage";
 import { useSettings } from "../SettingsProvider";
 import {setIn} from "seamless-immutable"
 import config from '../config.js';
+import { useSnackbar } from '../SnackbarContext'
 
 const extractRelevantProps = (region) => ({
   cls: region.cls,
@@ -50,6 +51,7 @@ export default () => {
   const [showLabel, setShowLabel] = useState(false)
   const [imageNames, setImageNames] = useState([])
   const settingsConfig = useSettings()
+  const { showSnackbar } = useSnackbar();
   const [settings, setSettings] =  useState({
     taskDescription: "",
     taskChoice: "image_classification",
@@ -114,6 +116,24 @@ export default () => {
     }
   };
 
+  const fetchImages = async (imageUrls) => {
+    try {
+      const fetchPromises = imageUrls.map(url =>
+        fetch(url.src).then(response => response.blob())
+          .then(blob => ({ ...url, src: URL.createObjectURL(blob) }))
+      );
+      const images = await Promise.all(fetchPromises);
+      setSettings(prevSettings => ({
+        ...prevSettings,
+        images
+      }));
+      setImageNames(images);
+    } catch (error) {
+      showSnackbar(error.message, 'error');
+    } finally {
+      setLoading(false);
+    }
+  }
   
   const getToolSelectionType = (toolName) => {
     const regions = [ {name: "Polygon", value: "create-polygon"}, {name: "Bounding Box", value: "create-box"}, {name: "Point", value: "create-point"}] 
@@ -129,9 +149,14 @@ export default () => {
   }
 
   useEffect(() => {
-    preloadConfiguration()
-    setLoading(false);
-  }, []);
+    preloadConfiguration();
+    if (settings.images.length > 0) {
+      fetchImages(settings.images);
+    } else {
+      setLoading(false);
+    }
+  }, [settingsConfig.settings, showLabel]);
+
 
   return (
     <>