aiqtech/PoseMaker2
0
1from fastapi import FastAPI, Request, Response2 3filenames = ["js/poseMaker.js"]4contents = '\n'.join([f"<script type='text/javascript' src='{x}'></script>" for x in filenames])5 6app = FastAPI()7 8@app.middleware("http")9async def insert_js(request: Request, call_next):10 path = request.scope['path'] # get the request route11 response = await call_next(request)12 13 if path == "/":14 response_body = ""15 async for chunk in response.body_iterator:16 response_body += chunk.decode()17 18 response_body = response_body.replace("</body>", contents + "</body>")19 20 del response.headers["content-length"]21 22 return Response(23 content=response_body,24 status_code=response.status_code, 25 headers=dict(response.headers),26 media_type=response.media_type27 )28 29 return response