First draft of API in Python
This commit is contained in:
parent
a11745e952
commit
541b9184bd
2 changed files with 64 additions and 0 deletions
61
app.py
Normal file
61
app.py
Normal 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
3
requirements.txt
Normal file
|
@ -0,0 +1,3 @@
|
|||
numpy==1.16.2
|
||||
Flask==1.0.2
|
||||
Pillow==6.1.0
|
Loading…
Reference in a new issue