import cv2
import numpy as np
from pytesseract import pytesseract
from operator import itemgetter

pytesseract.tesseract_cmd = r"C:\Program Files (x86)\Tesseract-OCR\tesseract.exe"
img = cv2.imread('static/im11.png')
img = cv2.resize(img, (850,350))
img2 = img
# threshold image
ret, threshed_img = cv2.threshold(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY),
								  130, 255, cv2.THRESH_BINARY)
# find contours and get the external one
image, contours, hierarchy = cv2.findContours(threshed_img, cv2.RETR_TREE,
										 cv2.CHAIN_APPROX_SIMPLE)

boxes = list()
cropped_imageList = list()
plate = list()
#получаем координаты прямоугольников вокруг контуров
for c in contours:
	x, y, w, h = cv2.boundingRect(c)
	boxes.append((x,y,w,h)) #хранит координаты каждого прямоугольника

boxesBasic = [i for i in boxes if (49 < i[3] < 250) and i[0] > 60 and (i[2]/i[3] < 0.8)]#убираем слишком высокие и слишком низкие контуры
boxesBasic.sort(key=itemgetter(0))

boxesBasic = sorted(boxesBasic)
#убираем контуры внутри контуров
for x in boxesBasic:
	for y in boxesBasic:
		if x[0] < y[0] and x[1] < y[1] and (x[1] + x[3]) > (y[1] + y[3]) and (x[0] + x[2]) > (y[0] + y[2]):
			boxesBasic.remove(y)

#что-то в блоке выше работает не так, это для надежности
boxesInner = [i for i in boxesBasic if  not( 180 < i[0] and i[3] < 150) or not (i[0] < 370 and i[3] < 150)]

#так как контуры выводятся в порядке возрастания координаты Х получается что-то типа а008мрr1u9s9
#нужно поменять местами цифры номера и буквы названия региона
if len(boxesInner) == 12:
	boxesInner[7],boxesInner[9]=boxesInner[9],boxesInner[7]
	boxesInner[8],boxesInner[11],=boxesInner[11],boxesInner[8]
	boxesInner[10],boxesInner[11]=boxesInner[11],boxesInner[10]
else:
	boxesInner[6],boxesInner[7]=boxesInner[7],boxesInner[6]
	boxesInner[9], boxesInner[10] = boxesInner[10], boxesInner[9]
	boxesInner[8], boxesInner[9] = boxesInner[9], boxesInner[8]
	boxesInner[7], boxesInner[8] = boxesInner[8], boxesInner[7]

#вырезание прямоугольников (и их отрисовка заодно)
for each in boxesInner:
	x = each[0]
	y = each[1]
	w = each[2]
	h = each[3]
	#рисует прямоугольники (изобр, координаты, цвет, толщина контура)
	#cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255), 3)
	crop_img = img[y:y+h, x:x+w]#вырезает прямоугольник с контуром
	crop_img = cv2.resize(crop_img,(150,300))#увеличивает в размере (ТЕССЕРАКТ ТРЕБОВАТЕЛЕН К РАЗМЕРУ, НЕ МЕНЬШЕ 300 ПИКСЕЛЕЙ ПО ВЕРТИКАЛИ)
	cropped_imageList.append(crop_img)

#БУКОВКА В НАЧАЛЕ
config1 = ("-l platechar --psm 10 --oem 3")
x = cropped_imageList[0]
ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
								  127, 255, cv2.THRESH_BINARY)
text = pytesseract.image_to_string(threshed_x, config=config1)

plate.append(text)

#ЦИФЕРКИ В СЕРЕДИНЕ
config1 = ("-l platenum --psm 10")
x = cropped_imageList[1]
ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
								  127, 255, cv2.THRESH_BINARY)
text = pytesseract.image_to_string(threshed_x, config=config1)
text = pytesseract.image_to_string(x, config=config1)
plate.append(text)

config1 = ("-l platenum --psm 10")
x = cropped_imageList[2]
ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
								  127, 255, cv2.THRESH_BINARY)
text = pytesseract.image_to_string(threshed_x, config=config1)
text = pytesseract.image_to_string(x, config=config1)
plate.append(text)

config1 = ("-l platenum --psm 10")
x = cropped_imageList[3]
ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
								  127, 255, cv2.THRESH_BINARY)
text = pytesseract.image_to_string(threshed_x, config=config1)
text = pytesseract.image_to_string(x, config=config1)
plate.append(text)
#КОНЕЦ ЦИФЕРОК В СЕРЕДИНЕ

#БУКОВКИ ПЕРЕД РЕГИОНОМ
config1 = ("-l platechar --psm 10 --oem 3")
x = cropped_imageList[4]
ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
								  127, 255, cv2.THRESH_BINARY)
text = pytesseract.image_to_string(threshed_x, config=config1)
plate.append(text)

config1 = ("-l platechar --psm 10 --oem 3")
x = cropped_imageList[5]
ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
								  127, 255, cv2.THRESH_BINARY)
text = pytesseract.image_to_string(threshed_x, config=config1)
plate.append(text)
#КОНЕЦ БУКОВОК ПЕРЕД РЕГИОНОМ

#если регион двузначный
if len(boxesInner) == 11:
	#ЦИФРЫ
	config1 = ("-l platenum --psm 10")
	x = cropped_imageList[6]
	ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
									127, 255, cv2.THRESH_BINARY)
	text = pytesseract.image_to_string(threshed_x, config=config1)
	text = pytesseract.image_to_string(x, config=config1)
	plate.append(text)

	config1 = ("-l platenum --psm 10")
	x = cropped_imageList[7]
	ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
									127, 255, cv2.THRESH_BINARY)
	text = pytesseract.image_to_string(threshed_x, config=config1)
	text = pytesseract.image_to_string(x, config=config1)
	plate.append(text)

	text = 'R'
	plate.append(text)
	text = 'U'
	plate.append(text)
	text = 'S'
	plate.append(text)

#если регион трехзначный
if len(boxesInner) == 12:
	config1 = ("-l platenum --psm 10")
	x = cropped_imageList[6]
	ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
									127, 255, cv2.THRESH_BINARY)
	text = pytesseract.image_to_string(threshed_x, config=config1)
	text = pytesseract.image_to_string(x, config=config1)
	plate.append(text)

	config1 = ("-l platenum --psm 10")
	x = cropped_imageList[7]
	ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
									127, 255, cv2.THRESH_BINARY)
	text = pytesseract.image_to_string(threshed_x, config=config1)
	text = pytesseract.image_to_string(x, config=config1)
	plate.append(text)

	config1 = ("-l platenum --psm 10")
	x = cropped_imageList[8]
	ret, threshed_x = cv2.threshold(cv2.cvtColor(x, cv2.COLOR_BGR2GRAY),
									127, 255, cv2.THRESH_BINARY)
	text = pytesseract.image_to_string(threshed_x, config=config1)
	text = pytesseract.image_to_string(x, config=config1)
	plate.append(text)

	text = 'R'
	plate.append(text)
	text = 'U'
	plate.append(text)
	text = 'S'
	plate.append(text)

result = ''.join(plate)
result = result.upper()
print(result)

imstack = cropped_imageList[0]
for imgs in cropped_imageList[1:]:
	imstack = np.hstack((imstack,imgs))#собственно, все вырезанные контуры

#повторная бинаризация, т.к. после масштабирования она слетает
ret, threshed_stack = cv2.threshold(cv2.cvtColor(imstack, cv2.COLOR_BGR2GRAY),
								  127, 255, cv2.THRESH_BINARY)

cv2.drawContours(img, contours, -1, (255, 255, 0), 1) #отрисовать котуры

#для проверки
boxesBasic = sorted(boxesBasic)
print(boxesBasic)
cv2.imshow('123', imstack)

#вывести на экран, если нужно
threshed_stack = cv2.resize(threshed_stack, (900,300))
cv2.imshow("threshed_numbers", threshed_stack)#вырезанные контуры
cv2.imshow("contours", img)#пикча для проверки правильности контуров
cv2.waitKey(0)
cv2.destroyAllWindows()