Update app.py

app.py

@@ -47,95 +47,6 @@ except Exception as e:
     print(f"Failed to connect to SAM client: {e}")
     sam_client = None
 
-# --- Helper Functions ---
-zerogpu_fix_js = """
-<script>
-    (function() {
-        const is_zerogpu = typeof window !== "undefined" && window.parent !== window;
-        if (!is_zerogpu) {
-            console.log("ZeroGPU Fix: Not in iframe, skipping.");
-            return;
-        }
-
-        console.log("ZeroGPU Fix: Detected iframe, initializing...");
-
-        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 {
-                console.log("ZeroGPU Fix: Requesting initial headers...");
-                const headers = await post_message("zerogpu-headers");
-                
-                // Helper to update headers object
-                const updateHeaders = (options, headers) => {
-                    if (!options) options = {};
-                    if (!options.headers) options.headers = {};
-                    
-                    const headerMap = headers instanceof Map ? headers : new Map(Object.entries(headers));
-                    
-                             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");
-                                         console.log(options.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.");
-                }
-            } catch (e) {
-                console.error("ZeroGPU Fix: Error initializing", e);
-            }
-        }
-        
-        // Run immediately
-        initZeroGPU();
-    })();
-</script>
-"""
-
 def get_zerogpu_headers(request_headers):
     """
     Extracts ZeroGPU specific headers from the incoming request headers.
@@ -172,17 +83,10 @@ def generate_image_handler(x, negative_prompt, fine_edge, fix_perspective, grow_
     add_prop_image = x['from_frontend']['add_prop_image']
     positive_prompt = x['from_backend']['prompt']
 
-    # 1. Extract headers from the current user's request
-    print(request.headers)
     forward_headers = get_zerogpu_headers(request.headers)
     
-    # 打印调试信息，确认是否收到了 x-zerogpu-token
     print(f"Debug: Received headers keys: {list(request.headers.keys())}")
     print(forward_headers)
-    if "x-zerogpu-token" in forward_headers:
-        print("Debug: x-zerogpu-token found in request headers.")
-    else:
-        print("Debug: x-zerogpu-token NOT found in request headers.")
 
     try:
         # 2. Instantiate a client specifically for this request with the forwarded headers.