from openai import OpenAI

client = OpenAI(base_url="https://api.naga.ac/v1", api_key="YOUR_API_KEY")

resp = client.images.edit(
    model="grok-imagine-image-2.0",
    image=open("image.png", "rb"),
    prompt="A cute baby sea otter wearing a beret",
    n=1,
    size="1024x1024",
    response_format="url"
)
print(resp.data)