#! /usr/bin/python3 from io import BytesIO from numpy import asarray from PIL import Image from redis import Redis import base64 import json import potrace import re import time color = 65280 r = Redis() def convertImg(src, image_path="/tmp/"): base64_data = re.sub('^data:image/.+;base64,', '', src) byte_data = base64.b64decode(base64_data) image_data = BytesIO(byte_data) img = Image.open(image_data).convert("1") return img # Read results from redis data = r.lpop('image-convert') while data : try: item = json.loads(data) image_data = item["image_data"] text = item["text"] hash_name = item["hash_name"] # Vectorize the image image = convertImg( image_data ) bmp = potrace.Bitmap( asarray( image ) ) # Trace the bitmap to a path path = bmp.trace(turdsize=16,alphamax=0.0, opticurve=0, opttolerance=1.0) pl = [] for curve in path: start = curve.start_point pl.append([int(start[0]),int(start[1]),0]) pl.append([int(start[0]),int(start[1]),color]) for segment in curve: end_point_x, end_point_y = segment.end_point if segment.is_corner: c_x, c_y = segment.c pl.append([int(c_x),int(c_y),color]) pass # else: c1_x, c1_y = segment.c1 x, y = segment.c2 pl.append([int(x),int(y),color]) pl.append([int(c1_x),int(c1_y),color]) # pl.append([start[0],start[1],0]) item["points_list"] = pl item["created_at"] = time.time() r.hset("images",hash_name, json.dumps(item)) data = r.lpop('image-convert') except Exception as e: print("woops",e) break