Added middleware

webrify2.py

@@ -60,12 +60,11 @@ async def get_page(url):
 
     return page, browser, pw
 
-
-# 🛡 Middleware to remove platform-identifying headers
 @app.middleware("http")
 async def remove_leaky_headers(request: Request, call_next):
     response: Response = await call_next(request)
 
+    # Safe header removal
     for header in [
         "link", 
         "x-proxied-host", 
@@ -73,8 +72,12 @@ async def remove_leaky_headers(request: Request, call_next):
         "x-proxied-replica", 
         "server"
     ]:
-        response.headers.pop(header, None)
+        try:
+            del response.headers[header]
+        except KeyError:
+            pass  # Header not present
 
+    # Add your own branded header
     response.headers["server"] = "Webrify-Secure-Gateway"
     return response