Adding reverse proxy for netron

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.9
+FROM python:3.9-alpine3.20
 
 RUN useradd -m -u 1000 user
 USER user
@@ -10,4 +10,5 @@ COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
+
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,8 +1,31 @@
-from fastapi import FastAPI
+from flask import Flask
 
-app = FastAPI()
+from starlette.requests import Request
+from starlette.responses import StreamingResponse
+from starlette.background import BackgroundTask
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World"}
+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"])
+
+@app.get("/")
+def base():
+    return "<p>Netron is running</p>" 

requirements.txt

@@ -1,2 +1,5 @@
 fastapi
 uvicorn[standard]
+flask
+httpx
+netron

wrapper.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+
+# Start the netron server
+netron &
+
+# Start the the web app 
+uvicorn app:app --host 0.0.0.0 --port 7860 &
+
+wait -n
+
+exit $?