Added middleware

webrify2.py

@@ -1,5 +1,5 @@
 # scrape.py
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request, Response
 from pydantic import BaseModel
 from typing import Optional
 import base64
@@ -61,6 +61,24 @@ 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)
+
+    for header in [
+        "link", 
+        "x-proxied-host", 
+        "x-proxied-path", 
+        "x-proxied-replica", 
+        "server"
+    ]:
+        response.headers.pop(header, None)
+
+    response.headers["server"] = "Webrify-Secure-Gateway"
+    return response
+
+
 @app.get("/metadata", response_model=MetadataResponse)
 async def get_metadata(url: str):
     page, browser, pw = await get_page(url)