Update app/routers/screenshot/screenshot.py

Test file streaming for speed up and reuse instance

app/routers/screenshot/screenshot.py

@@ -5,7 +5,7 @@ import logging
 from core.service.playwright import AsyncPlaywrightContext, TimeoutError
 from core.service.playwright.models import PageModel, ScreenshotModel
 from fastapi import APIRouter, HTTPException
-from fastapi.responses import Response
+from fastapi.responses import Response, StreamingResponse
 
 router = APIRouter()
 playwright_context = AsyncPlaywrightContext()
@@ -20,11 +20,11 @@ async def screenshot(
     async with playwright_context as sc:
         try:
             response = await sc.screenshot(screenshot_model=screenshot_model, page_model=page_model)
-            return Response(content=response, media_type="image/png")
+            return StreamingResponse(content=response, media_type="image/png")
         except TimeoutError as e:
             status_code = 504 if isinstance(e, TimeoutError) else 400
             detail = f"An error occurred while generating the screenshot: {e}"
             raise HTTPException(status_code=status_code, detail=detail) from e
         except Exception as e:
             logging.exception("Unhandled error in screenshot generation")
-            raise HTTPException(status_code=500, detail="An unexpected error occurred") from e
+            raise HTTPException(status_code=500, detail="An unexpected error occurred") from e