hello_app.py

import base64
import io 
from PIL import Image
from flask import request
from flask import jsonify
from flask import Flask

app = Flask(__name__)

@app.route('/hello', methods=['POST'])
def hello():
    message = request.get_json(force = True)
    name = message['name']
    response = {
        'greeting': 'Hello, ' + name + '!'
    }

    return jsonify(response)

@app.route('/predict', methods=['POST'])
def predict():
    message = request.get_json(force = True)
    encoded = message['image']
    decoded = base64.b64decode(encoded)
    image = Image.open(io.BytesIO(decoded))
    response = {
        'prediction': 'Hello, !'
    }

    return jsonify(response)