113 lines
3.8 KiB
Python
113 lines
3.8 KiB
Python
import models
|
|
import os
|
|
import tornado.ioloop
|
|
import tornado.web
|
|
import colour
|
|
import charts
|
|
from PIL import Image
|
|
import io
|
|
|
|
models.db.connect()
|
|
|
|
curpath = os.path.dirname(os.path.abspath(__file__))
|
|
tplpath = os.path.join(curpath, "templates");
|
|
|
|
loader = tornado.template.Loader(tplpath, autoescape=None)
|
|
|
|
config = {
|
|
"port": 8881,
|
|
"conn": models.db
|
|
}
|
|
|
|
emotions = [ e for e in models.Emotion.select().order_by(models.Emotion.name)]
|
|
|
|
class MainHandler(tornado.web.RequestHandler):
|
|
def get(self):
|
|
loader.reset()
|
|
html = loader.load('index.html').generate(host=self.request.host, emotions=emotions, ages=models.getAges(), groups=models.Group.select())
|
|
self.write(html)
|
|
|
|
class ThumbHandler(tornado.web.RequestHandler):
|
|
def get(self, imgId):
|
|
artwork = models.Artwork.select().where(models.Artwork.id == imgId).get()
|
|
if artwork is None:
|
|
self.send_error(404)
|
|
else:
|
|
img = Image.open(artwork.filename)
|
|
img.thumbnail((200,200))
|
|
fp = io.BytesIO()
|
|
img.save(fp, "jpeg")
|
|
fp.seek(0)
|
|
self.set_header("Content-Type","image/jpeg")
|
|
self.set_header("Cache-Control","max-age=2592000, public")
|
|
self.write(fp.read())
|
|
|
|
class ColourHandler(tornado.web.RequestHandler):
|
|
def get(self):
|
|
req_group = self.get_query_argument("group", default=None)
|
|
req_age = self.get_query_argument("age", default=None)
|
|
req_emotion = self.get_query_argument("emotion", default=None)
|
|
req_work = self.get_query_argument("work", default=None)
|
|
|
|
q = models.Artwork.select().join(models.Group).order_by(models.Group.name)
|
|
if req_group is not None:
|
|
group = models.Group.get(models.Group.id == req_group)
|
|
q = q.where(models.Artwork.group == group)
|
|
elif req_age is not None:
|
|
q = q.where(models.Artwork.age == req_age)
|
|
elif req_emotion is not None:
|
|
emotion = models.Emotion.get(models.Emotion.id == req_emotion)
|
|
q = q.where(models.Artwork.emotion == emotion)
|
|
elif req_work is not None:
|
|
q = q.where(models.Artwork.id == req_work)
|
|
|
|
images = q
|
|
svg = colour.getSvgFromDbImages(images)
|
|
|
|
imgHtml = "<aside id='works'>";
|
|
if req_group is not None or req_age is not None or req_emotion is not None or req_work is not None:
|
|
imgHtml += "<div class='group' onclick='loadResults(\"no\", \"thing\")'>←</div>"
|
|
lastGroup = None
|
|
for artwork in images:
|
|
if lastGroup != artwork.group.name:
|
|
imgHtml += "<div class='group' onclick='loadResults(\"group\", %s)'>%s</div>" % (artwork.group.id, artwork.group.name)
|
|
lastGroup = artwork.group.name
|
|
imgHtml += "<img src='/thumbs/%s.jpg' onclick='loadResults(\"work\", %s)'>" % (artwork.id,artwork.id)
|
|
imgHtml += "</aside>";
|
|
|
|
statHtml = "<aside id='stats'>";
|
|
if not req_emotion:
|
|
statHtml += "<h4>Emoties</h4>"
|
|
statHtml += charts.createPie(models.getEmotionFractionsFromArtworks(images))
|
|
else:
|
|
statHtml += "<h4>Groep</h4>"
|
|
statHtml += charts.createPie(models.getGroupFractionsFromArtworks(images))
|
|
|
|
statHtml += "<h4>Gezichtsherkenning</h4>"
|
|
faceScores = models.getFaceFractionsFromArtworks(images)
|
|
statHtml += charts.createPie(faceScores) if len(faceScores) else "Geen gezichten gevonden"
|
|
|
|
# statHtml += "<h4>Gezichten</h4>"
|
|
# statHtml += "Geen gezichten gevonden"
|
|
# statHtml += "<h4>Unieke woorden</h4>" # sorted(tf * idf)[:5]
|
|
# statHtml += "Geen woorden gebruikt"
|
|
|
|
# statHtml += "<h4>Meest voorkomende woorden</h4>"
|
|
# statHtml += "Geen woorden gevonden"
|
|
|
|
# TODO: licht/donker
|
|
# if not req_group:
|
|
statHtml += "</aside>";
|
|
|
|
self.write("\n".join([svg, imgHtml, statHtml]) )
|
|
|
|
if __name__ == "__main__":
|
|
print("Start server", config)
|
|
app = tornado.web.Application([
|
|
(r"/", MainHandler),
|
|
(r"/colours", ColourHandler),
|
|
(r"/thumbs/(\d+.jpg)", tornado.web.StaticFileHandler, {'path': './thumbs/'}),
|
|
(r'/images/(.*)', tornado.web.StaticFileHandler, {'path': './images/'}),
|
|
], debug=True)
|
|
app.listen(config['port'])
|
|
tornado.ioloop.IOLoop.current().start()
|