app.py

import os
from flask import Flask, request, send_from_directory, json, jsonify

from doggyML.dog_classify import classify_image

from waitress import serve

def create_app(test_config=None):
    app = Flask(__name__)

    @app.route("/", methods=['POST'])
    def index():
        if not 'image' in request.files:
            content = {'error': 'please send an image'}
            return json.jsonify(content), 400
       
        file = request.files['image']
        file.save(file.filename)

        breed = classify_image(file.filename)

        return breed

    @app.route('/static/<path:path>', methods=['GET'])
    def send_static(path):
        return send_from_directory('static', path)
    
    return app

serve(create_app(), listen='*:8080')