Showing Images with FormHandler

gramex.yaml

url:
  home:
    pattern: /$YAMLURL/
    handler: FunctionHandler
    kwargs:
      function: img.do
      headers:
        Content-Type: image/png

img.py

from scipy.misc import face
from PIL import Image
from io import BytesIO


x = face()


def do(handler):
    with BytesIO() as buff:
        Image.fromarray(x).save(buff, format="PNG")
        buff.seek(0)
        data = buff.read()
    return data