Mural Registry

1"""Mural Registry API built with Pixeltable.
2
3    pxt schema update app.py murals
4    pxt service run app.py murals
5"""
6import pixeltable as pxt
7import pixeltable.functions as pxtf
8from pixeltable.serving import FastAPIRouter
9
10from udfs import aspect_band, dominant_channel, tone_tag
11
12# ---- tables ----
13TableModel = pxt.model_base()
14
15
16class Murals(TableModel, name='murals', has_default_idxs=False):
17    id = pxt.Column(value=pxtf.uuid.uuid7(), primary_key=True)
18    title: pxt.String
19    artist: pxt.String
20    neighborhood: pxt.String
21    wall_photo: pxt.Image
22
23    thumb = wall_photo.resize((96, 64))      # generated media
24    gray = wall_photo.convert('L')           # generated media
25    aspect = aspect_band(wall_photo)
26    channel = dominant_channel(wall_photo)
27    tone = tone_tag(aspect, channel)         # chained on two computed columns
28
29    __indexes__ = [pxt.BtreeIndex(neighborhood)]
30
31
32# ---- queries ----
33@pxt.query
34def in_neighborhood(neighborhood: str):
35    """Murals in one neighborhood (neighborhood index)."""
36    return Murals.where(Murals.neighborhood == neighborhood).select(
37        Murals.id, Murals.title, Murals.artist, Murals.aspect, Murals.tone
38    ).order_by(Murals.title)
39
40
41@pxt.query
42def mural_thumb(title: str):
43    """The thumbnail for a mural title (served as an image file)."""
44    return Murals.where(Murals.title == title).order_by(Murals.id, asc=False).limit(1).select(Murals.thumb)
45
46
47# ---- routes ----
48mural_api = FastAPIRouter(name='mural_api')
49mural_api.add_insert_route(
50    Murals, path='/murals',
51    inputs=[Murals.title, Murals.artist, Murals.neighborhood],
52    uploadfile_inputs=[Murals.wall_photo],
53    outputs=[Murals.id, Murals.aspect, Murals.channel, Murals.tone],
54)
55mural_api.add_compute_route(Murals, path='/analyze', inputs=[], uploadfile_inputs=[Murals.wall_photo],
56                            outputs=[Murals.aspect, Murals.channel, Murals.tone])
57mural_api.add_delete_route(Murals, path='/murals/delete')
58mural_api.add_query_route(path='/murals/by-neighborhood', query=in_neighborhood, method='get')
59mural_api.add_query_route(path='/murals/thumb', query=mural_thumb, method='get', one_row=True,
60                          return_fileresponse=True)