import json

import os

from app import app

from flask import render_template, request

from werkzeug.utils import secure_filename

from classifier import getPrediction

@app.route('/')

def index():

    return render_template('about.html')

@app.route('/upload-image/', methods=['POST'])

def submit_file():

    if request.method == 'POST':

        if 'file' not in request.files:

            data = "File not found in request body"

            response = app.response_class(

                response=json.dumps(data),

                mimetype='application/json'

            )

            # flash('No file part')

            # return redirect(request.url)

            return response

        file = request.files['file']

        if file.filename == '':

            data = 'No file selected for uploading'

            response = app.response_class(

                response=json.dumps(data),

                mimetype='application/json'

            )

            # flash('No file selected for uploading')

            # return redirect(request.url)

            return response

        if file:

            filename = secure_filename(file.filename)

            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))

            # upload images to DB

            # upload_image_data_mongodb(file)

            label = getPrediction.processImg(filename)

            # flash({"Given class of image is ": label})

            data = {"Given class of image is ": label}

            response = app.response_class(

                response=json.dumps(data),

                mimetype='application/json'

            )

            if remove_img(path=os.path.join(app.config['UPLOAD_FOLDER']), img_name=file.filename):

                print('Image removed')

            # return redirect('/')

            return response

def remove_img(path, img_name):

    os.remove(path + '/' + img_name)

    # check if file exists or not

    if os.path.exists(path + '/' + img_name) is False:

        # file did not exists

        return True

if __name__ == "__main__":

    app.run()