Spaces:
Running
Running
File size: 1,448 Bytes
b2ccd7d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
'''Demo for how easily we can provide a UI for popular open-source tools.'''
from . import ops
from .executors import one_by_one
from PIL import Image, ImageFilter
import base64
import io
ENV = 'Pillow'
op = ops.op_registration(ENV)
one_by_one.register(ENV, cache=False)
@op("Open image")
def open_image(*, filename: str):
return Image.open(filename)
@op("Save image")
def save_image(image: Image, *, filename: str):
image.save(filename)
@op("Crop")
def crop(image: Image, *, top: int, left: int, bottom: int, right: int):
return image.crop((left, top, right, bottom))
@op("Flip horizontally")
def flip_horizontally(image: Image):
return image.transpose(Image.Transpose.FLIP_LEFT_RIGHT)
@op("Flip verically")
def flip_vertically(image: Image):
return image.transpose(Image.Transpose.FLIP_TOP_BOTTOM)
@op("Blur")
def blur(image: Image, *, radius: float = 5):
return image.filter(ImageFilter.GaussianBlur(radius))
@op("Detail")
def detail(image: Image):
return image.filter(ImageFilter.DETAIL)
@op("Edge enhance")
def edge_enhance(image: Image):
return image.filter(ImageFilter.EDGE_ENHANCE)
@op("To grayscale")
def to_grayscale(image: Image):
return image.convert('L')
@op("View image", view="image")
def view_image(image: Image):
buffered = io.BytesIO()
image.save(buffered, format="JPEG")
b64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
data_url = 'data:image/jpeg;base64,' + b64
return data_url
|