app.py

import os
import base64
import time
import numpy as np
from flask import Flask, flash, request, redirect, url_for,render_template
from werkzeug.utils import secure_filename
from test_ml import ml_softmax_test,cnn_test,cnn_new_test

UPLOAD_FOLDER = './pics'

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

@app.route("/")
def index():
    return render_template("index.html",count=1)

# WRITE ON A 32X32 CANVAS AND GIVE IMAGE TO THE PICKLE EXPORT TO GUESS WHAT NUMBER IT IS

@app.route("/test", methods=['GET', 'POST'])
def test():
     if request.method == 'POST':
         
        if 'file' not in request.values:
    
            flash('No file part')
            return redirect(request.url)

        file = request.values['file']
        
        #remove the begining of the file
        file = file.replace("data:image/png;base64,", "")
        
        convert_and_save(file)

        cnn_result,cnn_all = cnn_new_test("imageToSave.png")

        sga_result,sga_all = ml_softmax_test("imageToSave.png")

        total_results = {"cnn":[str(cnn_result),cnn_all.tolist()],"sga":[str(sga_result),sga_all.tolist()]}

        return total_results

@app.route("/save", methods=['GET','POST'])
def save():
    if request.method == 'POST':
        if 'file' not in request.values:
    
            flash('No file part')
            return redirect(request.url)
        file = request.values['file']
        number = request.values['number']
        #remove the begining of the file
        file = file.replace("data:image/png;base64,", "")
        
        convert_and_save(file,"pics/"+str(number)+"/"+str(time.time())+".png")

        return 'Image saved at '+ str(number)+"/"+str(time.time())


def convert_and_save(b64_string,filename="imageToSave.png"):
    with open(filename, "wb") as fh:
        fh.write(base64.decodebytes(b64_string.encode()))

if __name__ == "__main__":
    app.run(debug=True)