main.py

from flask import Flask,render_template,request,send_file,send_from_directory
import numpy as np
import pandas as pd
import sklearn.metrics as m
from keras.utils.np_utils import to_categorical
import os
import cv2
from sklearn.model_selection import train_test_split
from keras.models import Sequential
from keras.layers import Dense,Conv2D,Flatten,Activation,MaxPooling2D
from keras.preprocessing import image
from keras.models import load_model
from skimage import transform
import argparse
from keras.applications.vgg16 import VGG16
from keras.models import Model
import tensorflow as tf

model = load_model('model-facemask.h5')

def processesing(arr):
  for i in arr:
    if(i[0]>i[1]):
      return 0
    else:
      return 1
def images(img):
    image_read=[]
    image1=image.load_img(img)
    image2=image.img_to_array(image1)
    image3=cv2.resize(image2,(224,224))
    image_read.append(image3)
    img_array=np.asarray(image_read)
    return img_array
app = Flask(__name__,static_folder='static',template_folder='templates')

@app.route('/')
def home():
      return render_template("index.html")

def percentage(u,pre):
  sum=u[0][0]+u[0][1]
  return 100*u[0][pre]/sum

@app.route('/predict',methods=['POST','GET'])
def predict():
  if request.method=='POST':
    img=request.files['ima'].read()
    print(img)
    npimg = np.fromstring(img, np.uint8)
# convert numpy array to image
    img = cv2.imdecode(npimg,cv2.IMREAD_COLOR)
    cv2.imwrite("images/output.png",img)


    image3=cv2.resize(img,(224,224))
    image = np.expand_dims(image3, axis=0)

    imgarray=image
    print(imgarray)
    u=model.predict(imgarray)
    pre=processesing(u)
    print(u)
    perc=percentage(u,pre)
    
    if pre==0:
      print(0)
      response="Mask ON! You are Safe"
      return render_template("prediction.html",predict=response,percent=str(perc)+" %")

    if pre==1:
      print(1)
      response="Mask OFF! Please wear the Mask"
      return render_template("prediction.html",predict=response,percent=str(perc)+" %")

  if request.method=='GET':
    return render_template("index.html")

     
if __name__=='__main__':
    app.run(debug=False,threaded=False,port=8000,host='0.0.0.0')