Update components/middleware.py

components/middleware.py

@@ -282,27 +282,28 @@ class x401Kit(BaseHTTPMiddleware):
                         if response.headers.get("content-type") == "application/json":
                             
                             body_bytes = b""
-                        async for chunk in response.body_iterator:
-                            body_bytes += chunk
+                            
+                            async for chunk in response.body_iterator:
+                                body_bytes += chunk
 
-                        try:
-                            data = json.loads(body_bytes.decode())
-                        except json.JSONDecodeError:
-                            return response
+                            try:
+                                data = json.loads(body_bytes.decode())
+                            except json.JSONDecodeError:
+                                return response
 
         
-                        data["address"] = X_401_Addr
+                            data["address"] = X_401_Addr
         
-                        response_headers = dict(response.headers)
-                        response_headers.pop("content-length", None)
-                        print(data) 
+                            response_headers = dict(response.headers)
+                            response_headers.pop("content-length", None)
+                            print(data) 
 
-                        return JSONResponse(
+                            return JSONResponse(
                                     content=data,
                                     status_code=response.status_code,
                                     headers=response_headers
                                     )
-
+                        
 
                         return response