API return image from output directory?

[Graylington]

When using the API, is there a way to access the actual image generated in the output directory of Auto1111?

[missionfloyd]

Files can be accessed with the /file endpoint, e.g., http://127.0.0.1:7860/file=outputs/txt2img-images/2024-02-11/00000-2709453521.png

By default, the API doesn't save images. Instead, it returns the images as base64. They can be saved locally like so.

import requests
import base64

payload = {
    "prompt": "cat",
    "steps": 20,
}

response = requests.post(url='http://127.0.0.1:7860/sdapi/v1/txt2img', json=payload)
r = response.json()

with open('output.png', 'wb') as f:
    f.write(base64.b64decode(r['images'][0]))

or read with pillow

import requests
import base64
import io
from PIL import Image

payload = {
    "prompt": "cat",
    "steps": 20,
}

response = requests.post(url='http://127.0.0.1:7860/sdapi/v1/txt2img', json=payload)
r = response.json()

with io.BytesIO(base64.b64decode(r['images'][0])) as f:
    image = Image.open(f)

To save them on the server, add save_images: True to your payload. If you don't want API requests to return base64 images, add send_images: False

[Graylington]

Thanks for the info.

I have been saving new images from base64 data, however I was hoping to include the same metadata that gets embedded into the png file that Auto1111 outputs so that the resulting image could be dropped into the WebUI PNG Info and easily recreated . Hence the desire to grab the actual outputted file.

To use the file endpoint, you need the filename. Can info such as this in regards to the last image created be found somewhere?

[missionfloyd]

The first example above creates files identical to those saved by webui.

[Graylington]

Thank you! This is exactly what I needed.

[Graylington]

Files can be accessed with the /file endpoint, e.g., http://127.0.0.1:7860/file=outputs/txt2img-images/2024-02-11/00000-2709453521.png

By default, the API doesn't save images. Instead, it returns the images as base64. They can be saved locally like so.

import requests
import base64

payload = {
    "prompt": "cat",
    "steps": 20,
}

response = requests.post(url='http://127.0.0.1:7860/sdapi/v1/txt2img', json=payload)
r = response.json()

with open('output.png', 'wb') as f:
    f.write(base64.b64decode(r['images'][0]))

or read with pillow

import requests
import base64
import io
from PIL import Image

payload = {
    "prompt": "cat",
    "steps": 20,
}

response = requests.post(url='http://127.0.0.1:7860/sdapi/v1/txt2img', json=payload)
r = response.json()

with io.BytesIO(base64.b64decode(r['images'][0])) as f:
    image = Image.open(f)

To save them on the server, add save_images: True to your payload. If you don't want API requests to return base64 images, add send_images: False

import requests
import base64

payload = {
    "prompt": "cat",
    "steps": 20,
}

response = requests.post(url='http://127.0.0.1:7860/sdapi/v1/txt2img', json=payload)
r = response.json()

with open('output.png', 'wb') as f:
    f.write(base64.b64decode(r['images'][0]))