Update app/routers/screenshot/screenshot.py

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, StreamingResponse
+from fastapi.responses import Response
 
 router = APIRouter()
 playwright_context = AsyncPlaywrightContext()
@@ -20,7 +20,7 @@ async def screenshot(
     async with playwright_context as sc:
         try:
             response = await sc.screenshot(screenshot_model=screenshot_model, page_model=page_model)
-            return StreamingResponse(content=response, media_type="image/png")
+            return Response(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}"