Update app.py

app.py

@@ -49,6 +49,60 @@ except Exception as e:
 
 # --- Helper Functions ---
 
+# JS script to fix ZeroGPU headers propagation
+zerogpu_fix_js = """
+<script>
+    (function() {
+        const is_zerogpu = typeof window !== "undefined" && window.parent !== window;
+        if (!is_zerogpu) return;
+
+        console.log("ZeroGPU Fix: Detected iframe, attempting to fetch headers.");
+
+        function post_message(message) {
+            return new Promise((resolve, reject) => {
+                const channel = new MessageChannel();
+                channel.port1.onmessage = (event) => {
+                    channel.port1.close();
+                    resolve(event.data);
+                };
+                window.parent.postMessage(message, "*", [channel.port2]);
+            });
+        }
+
+        async function initZeroGPU() {
+            try {
+                const headers = await post_message("zerogpu-headers");
+                if (headers) {
+                    console.log("ZeroGPU Fix: Received headers", headers);
+                    
+                    const originalFetch = window.fetch;
+                    window.fetch = async function(url, options) {
+                        if (!options) options = {};
+                        if (!options.headers) options.headers = {};
+                        
+                        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 originalFetch(url, options);
+                    };
+                }
+            } catch (e) {
+                console.error("ZeroGPU Fix: Error fetching headers", e);
+            }
+        }
+        
+        initZeroGPU();
+    })();
+</script>
+"""
+
 def get_zerogpu_headers(request_headers):
     """
     Extracts ZeroGPU specific headers from the incoming request headers.
@@ -126,7 +180,7 @@ def generate_image_handler(x, negative_prompt, fine_edge, fix_perspective, grow_
 
 # --- Gradio UI ---
 
-with gr.Blocks(title="MagicQuill V2") as demo:
+with gr.Blocks(title="MagicQuill V2", head=zerogpu_fix_js) as demo:
     with gr.Row(elem_classes="row"):
         text = gr.Markdown(
         """