Update main.py

main.py

@@ -1,6 +1,5 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import Response
-from urllib.parse import urljoin
 import httpx
 
 app = FastAPI()
@@ -23,22 +22,18 @@ 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
+        # Process the Location header if present in the response
+        response_headers = dict(response.headers)
+        if "Location" in response_headers:
+            # Replace the target URL with the proxy URL
+            response_headers["Location"] = response_headers["Location"].replace("http://www.medicalgpt.club", request.base_url)
 
         return Response(
             content=response.content,
             status_code=response.status_code,
-            headers=dict(response.headers)
+            headers=response_headers
         )
 
 if __name__ == "__main__":
     import os
-    os.system("uvicorn main:app --host 0.0.0.0 --port 7860")
+    os.system("uvicorn main:app --host 0.0.0.0 --port 7860")