修改一些细节

slince-zero · May 9, 2024 · 97663aa · 97663aa
1 parent 841d34f
commit 97663aa
Showing 1 changed file with 4 additions and 14 deletions.
diff --git a/src/context/index.tsx b/src/context/index.tsx
@@ -250,17 +250,6 @@ export default function ImgContextProvider({
     return result
   }
 
-  function blobToDataURL(blob: any) {
-    return new Promise((resolve, reject) => {
-      const reader = new FileReader()
-      reader.onloadend = function () {
-        resolve(reader.result)
-      }
-      reader.onerror = reject
-      reader.readAsDataURL(blob)
-    })
-  }
-
   // 生成AI图片
   function handleGenerateAIImage() {
     try {
@@ -269,9 +258,8 @@ export default function ImgContextProvider({
         inputs: aiValue,
       }
       query(data).then((res) => {
-        blobToDataURL(res).then((url) => {
-          setAiResult(url)
-        })
+        const url = URL.createObjectURL(res)
+        setAiResult(url)
         setLoadingAIImage(false)
       })
     } catch (e) {
@@ -290,6 +278,8 @@ export default function ImgContextProvider({
     })
   }
 
+
+
   return (
     <ImgContext.Provider
       value={{