Update app.py

app.py

@@ -82,45 +82,44 @@ zerogpu_fix_js = """
                     
                     const headerMap = headers instanceof Map ? headers : new Map(Object.entries(headers));
                     
-                    headerMap.forEach((value, key) => {
-                        if (options.headers instanceof Headers) {
-                            options.headers.append(key, value);
-                        } else {
-                            options.headers[key] = value;
-                        }
-                    });
-                    return options;
-                };
-
-                if (headers) {
-                    console.log("ZeroGPU Fix: Received initial headers", headers);
-                    
-                    const originalFetch = window.fetch;
-                    window.fetch = async function(url, options) {
-                        // Inject headers for ALL requests to be safe, especially queue/join
-                        // We filter out only obviously external or static asset requests if needed, 
-                        // but for debugging, let's be aggressive.
-                        if (typeof url === 'string' && (url.includes('/queue/') || url.includes('/api/') || url.includes('run/predict'))) {
-                            console.log("ZeroGPU Fix: Intercepting request to", url);
-                            
-                            try {
-                                // Attempt to refresh headers
-                                const currentHeaders = await post_message("zerogpu-headers");
-                                if (currentHeaders) {
-                                    options = updateHeaders(options, currentHeaders);
-                                    console.log("ZeroGPU Fix: Injected FRESH headers");
-                                } else {
-                                    options = updateHeaders(options, headers);
-                                    console.log("ZeroGPU Fix: Injected CACHED headers (refresh failed)");
-                                }
-                            } catch (e) {
-                                options = updateHeaders(options, headers);
-                                console.log("ZeroGPU Fix: Injected CACHED headers (error)");
-                            }
-                        }
-                        
-                        return originalFetch(url, options);
-                    };
+                             headerMap.forEach((value, key) => {
+                                 // Force lowercase keys to avoid case mismatch
+                                 key = key.toLowerCase();
+                                 if (options.headers instanceof Headers) {
+                                     options.headers.set(key, value); // Use set instead of append to avoid duplicates
+                                 } else {
+                                     options.headers[key] = value;
+                                 }
+                             });
+                             return options;
+                         };
+         
+                         if (headers) {
+                             console.log("ZeroGPU Fix: Received initial headers", headers);
+                             
+                             const originalFetch = window.fetch;
+                             window.fetch = async function(url, options) {
+                                 console.log("ZeroGPU Fix: Intercepting request to", url);
+                                 
+                                 if (!options) options = {};
+                                 // Inject headers for ALL requests to be safe
+                                 try {
+                                     // Attempt to refresh headers
+                                     const currentHeaders = await post_message("zerogpu-headers");
+                                     if (currentHeaders) {
+                                         options = updateHeaders(options, currentHeaders);
+                                         console.log("ZeroGPU Fix: Injected FRESH headers");
+                                     } else {
+                                         options = updateHeaders(options, headers);
+                                         console.log("ZeroGPU Fix: Injected CACHED headers (refresh failed)");
+                                     }
+                                 } catch (e) {
+                                     options = updateHeaders(options, headers);
+                                     console.log("ZeroGPU Fix: Injected CACHED headers (error)");
+                                 }
+                                 
+                                 return originalFetch(url, options);
+                             };
                     console.log("ZeroGPU Fix: Fetch interceptor installed.");
                 } else {
                     console.warn("ZeroGPU Fix: No headers received from parent.");