Render mockups from a FastAPI app

import os
from contextlib import asynccontextmanager

from fastapi import Depends, FastAPI, Request
from pydantic import BaseModel, HttpUrl
from sudomock import AsyncSudoMock

MOCKUP_UUID = "your-mockup-uuid"
SMART_OBJECT_UUID = "your-smart-object-uuid"


@asynccontextmanager
async def lifespan(app: FastAPI):
    app.state.sudomock = AsyncSudoMock(
        api_key=os.environ["SUDOMOCK_API_KEY"],
    )
    yield
    await app.state.sudomock.close()


app = FastAPI(lifespan=lifespan)


def get_client(request: Request) -> AsyncSudoMock:
    return request.app.state.sudomock


class RenderIn(BaseModel):
    artwork_url: HttpUrl


@app.post("/renders")
async def create_render(
    body: RenderIn,
    client: AsyncSudoMock = Depends(get_client),
) -> dict:
    render = await client.renders.create(
        mockup_uuid=MOCKUP_UUID,
        smart_objects=[
            {
                "uuid": SMART_OBJECT_UUID,
                "asset": {
                    "url": str(body.artwork_url),
                    "fit": "fit",
                },
            }
        ],
        export_options={
            "image_format": "webp",
            "image_size": 2048,
        },
    )
    return {"image_url": render.url}