Update main.py

main.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import Response
+from urllib.parse import urljoin
 import httpx
 
 app = FastAPI()
@@ -22,6 +23,16 @@ async def proxy(request: Request, path: str):
             content=content
         )
 
+        # Check if there's a Location header (indicating a redirect)
+        if "Location" in response.headers:
+            # Modify the location to pass through the proxy
+            location = response.headers["Location"]
+            
+            # Construct a new URL that goes through the proxy
+            new_location = urljoin(f"http://{request.url.netloc}", location)
+            
+            response.headers["Location"] = new_location
+
         return Response(
             content=response.content,
             status_code=response.status_code,