First draft of API in Python

This commit is contained in:
Paul Warren 2019-07-30 20:34:33 +10:00
parent a11745e952
commit 541b9184bd
2 changed files with 64 additions and 0 deletions

61
app.py Normal file
View File

@ -0,0 +1,61 @@
from flask import Flask, request, send_file
#import numpy
from PIL import Image, ImageChops
from io import BytesIO
import numpy
app = Flask(__name__)
usage = """
Post image Data to /desteg and receive a copy of that image with the least significant bits XORed with system entropy
Example: curl -H "Content-Type: application/png" --data-binary @image.png http://localhost:5002/desteg -o image_destegged.png
"""
@app.route("/desteg", methods=["GET"])
def get_desteg():
return usage
@app.route("/desteg", methods=["POST"])
def post_desteg():
image_string = BytesIO(request.data)
output_buffer = BytesIO()
try:
input_image = Image.open(image_string)
except IOError:
# insert relevant error return here
pass
# Create an array the same size
# Stuff it full of entropy
if (input_image.mode == "RGBA"):
entropy_array = numpy.random.rand(input_image.size[1],input_image.size[0],4)*3
entropy_image = Image.fromarray(entropy_array.astype('uint8'), mode="RGBA")
elif (input_image.mode == "RGB"):
entropy_array = numpy.random.rand(input_image.size[1],input_image.size[0],3)*3
entropy_image = Image.fromarray(entropy_array.astype('uint8'), mode="RGB")
else:
# Insert relevant error return here
pass
# Add the entropy to input
output_image = ImageChops.add(input_image, entropy_image)
# save with same format as output
output_image.save(output_buffer,format=input_image.format)
#return buffer to start!
output_buffer.seek(0)
return send_file(output_buffer,
mimetype="image/" + input_image.format.lower())
@app.route("/")
def root():
return usage
if __name__ == '__main__':
app.run(port='5002')

3
requirements.txt Normal file
View File

@ -0,0 +1,3 @@
numpy==1.16.2
Flask==1.0.2
Pillow==6.1.0