Main.py

# Main.py
import os
from _curses import flash
from builtins import int, round, tuple, len, float

import cv2
import pymysql
from flask import Flask, render_template, request
from werkzeug.utils import secure_filename, redirect

from flask_sqlalchemy import SQLAlchemy

import DetectCharactersHelper
import DetectPlatesHelper

app = Flask(__name__)

global numberplateintotext
numberplateintotext = ""

ALLOWED_EXTENSION = ['.jpg, .png, .jpeg, .gif']
APP_ROOT = os.path.dirname(os.path.dirname(__file__))
UPLOAD_FOLDER = '/home/sarin/PycharmProjects/AI_PROJECTS/UPLOAD_FOLDER/'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER


black = (0.0, 0.0, 0.0)
white = (255.0, 255.0, 255.0)
yellow = (0.0, 255.0, 255.0)
green = (0.0, 255.0, 0.0)
red = (0.0, 0.0, 255.0)


@app.route("/", methods=['GET'])
def index():
    return render_template("upload.html")


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


@app.route("/uploads", methods=['GET', 'POST'])
def upload():
    if request.method == 'POST':
        # check if the post request has the file part
        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')
            print("No fileee")
            return redirect(request.url)
        else:
            filename = secure_filename(file.filename)
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
            print("file uploaded ")
            main(UPLOAD_FOLDER + str(filename))
            return render_template("completed.html", numberplate=numberplateintotext)
        # print("Skippeddddd")


@app.route("/uploads/data", methods=['GET', 'POST'])
def data():
    db = pymysql.connect("localhost", "testuser", "test123", "number_plate_recognition")
    cursor = db.cursor()
    sql = "SELECT * FROM information WHERE ID='"+numberplateintotext+"'"
    print(sql)
    #ya hera
    try:
        # Execute the SQL command
        cursor.execute(sql)
        results = cursor.fetchall()
        for row in results:
            ID = row[0]
            CompanyName = row[1]
            ModelNumber = row[2]
            ManufactureDate = row[3]
            HorsePower = row[4]
            CC = row[5]
            EngineNumber = row[6]
            ChassisNo = row[7]
            EngineType = row[8]
            SeaterCapacity = row[9]
            BorderEntrance = row [10]
            VechileType = row[11]
            VechileColor = row[12]


            print("=======================================DATABASE QUERY=============================================")
            print("====================================================================================")
            print("")
            print("ID=%s,CompanyName=%s,ModelNumber=%s,ManufactureDate=%s,HorsePower=%d, CC=%d, EngineNumber=%s, ChassisNo=%s, EngineType=%s, SeaterCapacity=%d, BorderEntrance=%s, VechileType=%s, VechileColor=%s "                       %
                  (ID, CompanyName, ModelNumber, ManufactureDate, HorsePower,CC,EngineNumber,ChassisNo,EngineType,SeaterCapacity, BorderEntrance ,VechileType, VechileColor ))
            print("")
            print("====================================================================================")
            print("====================================================================================")
    except:
        print("Error: unable to fetch data")
    finally:
        # disconnect from server
        db.close()
        return render_template("data.html", query=results, id=ID, companyname = CompanyName, modelnumber= ModelNumber, manufacturedate=ManufactureDate, horsepower= HorsePower, cc= CC, enginenumber= EngineNumber, chassisno = ChassisNo, enginetype=EngineType, seatercapacity= SeaterCapacity, border=BorderEntrance, vechiletype= VechileType, color= VechileColor)

def main(path):
    dataTraining = DetectCharactersHelper.loadDataAndTrain()  # attempt KNN training

    if dataTraining == False:  # if KNN training was not successful
        print("\n error: KNN-traning was not successful\n")  # show error message
        return  # and exit program
    # end if

    originalImage = cv2.imread(path)  # open image
    print(path)

    if originalImage is None:  # if image was not read successfully
        print("\n ERROR:: image not read from the picture \n\n")
        os.system("pause")  # pauses the code so error can be checked
        return  # and exit program
    # end if

    listOfPossiblePlates = DetectPlatesHelper.detectPlatesInScene(originalImage)  # detect plates

    listOfPossiblePlates = DetectCharactersHelper.detectCharsInPlates(listOfPossiblePlates)  # detect chars in plates

    # cv2.imshow("Original_Image", originalImage)  # show scene image

    if len(listOfPossiblePlates) == 0:  # if no plates were found
        print("\n ERROR:: no license plates were detected\n")  # print user no plates were found
    else:

        # sort the list of possible plates in DESCENDING order (most number of chars to least number of chars)
        listOfPossiblePlates.sort(key=lambda possiblePlate: len(possiblePlate.strChars), reverse=True)

        # suppose the plate with the most recognized chars (the first plate in sorted by string length descending order) is the actual plate
        licPlate = listOfPossiblePlates[0]

        # cv2.imshow("Plate_image", licPlate.imgPlate)  # show crop of plate and threshold of plate
        # cv2.imshow("Threshold_image", licPlate.imgThresh)

        if len(licPlate.strChars) == 0:  # if no chars were found in the plate
            print("\n ERROR :: no characters were detected\n\n")  # show message
            return  # and exit program
        # end if

        # draw red rectangle around plate
        drawRedRectangleAroundPlate(originalImage, licPlate)

        print("------------------DETECTED-NUMBER PLATE FROM IMAGE----------------------")
        print("\n " + licPlate.strChars + "\n")
        print("----------------------------------------")
        writeLicensePlateCharsOnImage(originalImage, licPlate)  # write license plate text on the image
        # cv2.imshow("final_image", originalImage)  # showing image again from image comparision
        cv2.imwrite("original_Image.png", originalImage)  # write image out to file
        cv2.waitKey(0)  # hold windows open until user presses a key
        # end if else

        global numberplateintotext
        numberplateintotext = licPlate.strChars

    return


# end main


def drawRedRectangleAroundPlate(imgOriginalScene, licPlate):
    p2fRectPoints = cv2.boxPoints(licPlate.rrLocationOfPlateInScene)  # four sides of rotated rectangle plate

    cv2.line(imgOriginalScene, tuple(p2fRectPoints[0]), tuple(p2fRectPoints[1]), red, 2)  # draw 4 red lines
    cv2.line(imgOriginalScene, tuple(p2fRectPoints[1]), tuple(p2fRectPoints[2]), red, 2)
    cv2.line(imgOriginalScene, tuple(p2fRectPoints[2]), tuple(p2fRectPoints[3]), red, 2)
    cv2.line(imgOriginalScene, tuple(p2fRectPoints[3]), tuple(p2fRectPoints[0]), red, 2)


# end function

###################################################################################################
def writeLicensePlateCharsOnImage(imgOriginalScene, licPlate):
    ptCenterOfTextAreaX = 0  # this will be the center of the area the text will be written to
    ptCenterOfTextAreaY = 0

    BottomLowerLeftTextFromNumberPlateX = 0  # this will be the bottom left of the area that the text will be written to
    BottomLowerLeftTextFromNumberPlateY = 0

    sceneHeight, sceneWidth, sceneNumChannels = imgOriginalScene.shape
    plateHeight, plateWidth, plateNumChannels = licPlate.imgPlate.shape

    intFontFace = cv2.FONT_HERSHEY_SIMPLEX  # choose a plain jane font
    fltFontScale = float(plateHeight) / 30.0  # base font scale on height of plate area
    intFontThickness = int(round(fltFontScale * 1))  # base font thickness on font scale

    textSize, baseline = cv2.getTextSize(licPlate.strChars, intFontFace, fltFontScale,
                                         intFontThickness)  # call getTextSize

    # unpack roatated rect into center point, width and height, and angle
    ((intPlateCenterX, intPlateCenterY), (intPlateWidth, intPlateHeight),
     fltCorrectionAngleInDeg) = licPlate.rrLocationOfPlateInScene

    intPlateCenterX = int(intPlateCenterX)  # make sure center is an integer
    intPlateCenterY = int(intPlateCenterY)

    ptCenterOfTextAreaX = int(intPlateCenterX)  # the horizontal location of the text area is the same as the plate

    if intPlateCenterY < (sceneHeight * 0.75):  # if the license plate is in the upper 3/4 of the image
        ptCenterOfTextAreaY = int(round(intPlateCenterY)) + int(
            round(plateHeight * 1.6))  # write the chars in below the plate
    else:  # else if the license plate is in the lower 1/4 of the image
        ptCenterOfTextAreaY = int(round(intPlateCenterY)) - int(
            round(plateHeight * 1.6))  # write the chars in above the plate
    # end if

    textSizeWidth, textSizeHeight = textSize  # unpack text size width and height

    BottomLowerLeftTextFromNumberPlateX = int(
        ptCenterOfTextAreaX - (textSizeWidth / 2))  # calculate the lower left origin of the text area
    BottomLowerLeftTextFromNumberPlateY = int(
        ptCenterOfTextAreaY + (textSizeHeight / 2))  # based on the text area center, width, and height

    # write the text on the image
    cv2.putText(imgOriginalScene, licPlate.strChars,
                (BottomLowerLeftTextFromNumberPlateX, BottomLowerLeftTextFromNumberPlateY), intFontFace, fltFontScale,
                yellow, intFontThickness)


# end function
showSteps = False

# main()
# database()

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