Spaces:
Sleeping
Sleeping
from flask import Flask | |
from starlette.requests import Request | |
from starlette.responses import StreamingResponse | |
from starlette.background import BackgroundTask | |
import httpx | |
client = httpx.AsyncClient(base_url="http://127.0.0.1:8080/") | |
async def _reverse_proxy(request: Request): | |
url = httpx.URL(path=request.url.path, | |
query=request.url.query.encode("utf-8")) | |
rp_req = client.build_request(request.method, url, | |
headers=request.headers.raw, | |
content=await request.body()) | |
rp_resp = await client.send(rp_req, stream=True) | |
return StreamingResponse( | |
rp_resp.aiter_raw(), | |
status_code=rp_resp.status_code, | |
headers=rp_resp.headers, | |
background=BackgroundTask(rp_resp.aclose), | |
) | |
app = Flask(__name__) | |
app.add_route("/netron", _reverse_proxy, methods=["GET", "POST", "PUT", "DELETE", "PATCH", "OPTIONS", "HEAD"]) | |
def base(): | |
return "<p>Netron is running</p>" | |