Refactor caching logic in sw.js and transformers.ts

public/sw.js

@@ -46,21 +46,6 @@ self.addEventListener('fetch', (event) => {
             event.respondWith(new Response(`${error}`))
         }
     }
-    if(path[1] === 'transformers'){
-        event.respondWith((async () => {
-            const cache = await caches.open('risuCache')
-            const res = await cache.match(url)
-            if(res){
-                return res
-            }else{
-                url.host = "https://sv.risuai.xyz"
-                return await fetch(url, {
-                    headers: event.request.headers,
-                    method: event.request.method
-                })
-            }
-        })())
-    }
     if(path[1] === 'tf'){{
         event.respondWith(new Response("Cannot find resource from cache", {
             status: 404

src/ts/process/embedding/transformers.ts

@@ -12,9 +12,20 @@ async function initTransformers(){
     }
     tfCache = await caches.open('transformers')
     env.localModelPath = "/tf/"
-    env.remoteHost = "https://sv.risuai.xyz/transformers/"
+    env.useBrowserCache = false
-    env.customCache = tfCache
+    env.useFSCache = false
+    env.useCustomCache = true
+    env.customCache = {
+        put: async (url:URL|string, response:Response) => {
+            await tfCache.put(url, response)
+        },
+        match: async (url:URL|string) => {
+            console.log('match', url)
+            return await tfCache.match(url)
+        }
+    }
     tfLoaded = true
+    console.log('transformers loaded')
 }
 
 export const runTransformers = async (baseText:string, model:string,config:TextGenerationConfig = {}) => {
@@ -103,7 +114,11 @@ export const runVITS = async (text: string, modelData:string|OnnxModelFiles = 'X
             const files = modelData.files
             const keys = Object.keys(files)
             for(const key of keys){
-                tfCache.put(key, new Response(await loadAsset(files[key])))
+                const ast = new Response(await loadAsset(files[key]))
+                tfCache.put(key,ast.clone())
+                tfCache.put(location.origin + key, ast.clone())
+                console.log('put', key)
+                console.log('put', location.origin + key)
             }
             lastSynth = modelData.id
             synthesizer = await pipeline('text-to-speech', modelData.id);
@@ -157,7 +172,6 @@ export const registerOnnxModel = async ():Promise<OnnxModelFiles> => {
         if(url.startsWith('/')){
             url = url.substring(1)
         }
-        url = '/tf/' + id +'/' + url
         fileIdMapped[url] = fid
     }