main.py

from flask import Flask, flash, request, redirect, url_for, Response
from werkzeug.utils import secure_filename
import redis
import os
from model import query_model

r = redis.Redis(host='localhost', port=6379, db=0)

r.set("test", "2")

UPLOAD_FOLDER = "D:\Projects\WED\emotes"
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])

app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER


@app.route("/")
def hello_world():
    return "<h1>Hello, World!</h1>"


def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS


@app.route("/upload", methods=['POST'])
def upload():
    print("upload")
    if request.method == 'POST':
        if 'file' not in request.files:
            flash('No file part')
            return redirect(request.url)
        file = request.files['file']
        # if user does not select file, browser also
        # submit a empty part without filename
        if file.filename == '':
            flash('No selected file')
            return redirect(request.url)
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
            file.save(path)
            score = query_model.get_weeb_score(path)
            print(score)
            os.remove(path)
            return {"response_code": 200, "score": int(score)}