ui bug bash fixes (TransformerOptimus#839)

ricky8goo · Jul 21, 2023 · 540509d · 540509d
1 parent 9156985
commit 540509d
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/gui/pages/Content/Agents/ResourceManager.js b/gui/pages/Content/Agents/ResourceManager.js
@@ -14,18 +14,26 @@ export default function ResourceManager({agentId, runs}) {
   const [isDragging, setIsDragging] = useState(false);
   const fileInputRef = useRef(null);
 
-  const handleFileInputChange = (event) => {
-    const files = event.target.files;
+  function handleFile(files) {
     if (files.length > 0) {
-      const fileData = {
-        "file": files[0],
-        "name": files[0].name,
-        "size": files[0].size,
-        "type": files[0].type,
-      };
-      uploadResource(fileData);
+      const sizeInMB = files[0].size / (1024*1024);
+      if (sizeInMB > 5) {
+        toast.error('File size should not exceed 5MB', {autoClose: 1800});
+      } else {
+        const fileData = {
+          "file": files[0],
+          "name": files[0].name,
+          "size": files[0].size,
+          "type": files[0].type,
+        };
+        uploadResource(fileData);
+      }
     }
   };
+  const handleFileInputChange = (event) => {
+    const files = event.target.files;
+    handleFile(files);
+  };
 
   const handleDropAreaClick = () => {
     fileInputRef.current.click();
@@ -48,15 +56,7 @@ export default function ResourceManager({agentId, runs}) {
     event.preventDefault();
     setIsDragging(false);
     const files = event.dataTransfer.files;
-    if (files.length > 0) {
-      const fileData = {
-        "file": files[0],
-        "name": files[0].name,
-        "size": files[0].size,
-        "type": files[0].type,
-      };
-      uploadResource(fileData);
-    }
+    handleFile(files);
   };
 
   useEffect(() => {