@Garipov-Amir

Telegram bot с Tesseract на борту не хочет работать, пожалуйста просмотрите мой код?

import telebot
import paho.mqtt as mqtt
from time import *
import pytesseract
from paho.mqtt import client as mqtt
import socket
import cv2
import numpy
from PIL import Image
message_bot = ''
is_ready = False
on_point_4 = False
start_charge = False
last_message = ''
bot_message = ''
last_x_coordinate = 0
last_y_coordinate = 0

charge_stoped = False
def on_message(client, userdata, message):
    message_from_bot = message.payload.decode()
    global charge_stoped
    print(message_from_bot)
    if message_from_bot == 'ok':
        is_ready = True
        last_message = message_from_bot
        print(is_ready)
    if len(message_from_bot) >= 6:
        message_from_bot = message_from_bot.split('"')
        bot_message = message_from_bot[3]
        bot_message = bot_message.lower()
        print(bot_message)

pytesseract.pytesseract.tesseract_cmd = 'C:/Program Files/Tesseract-OCR/tesseract.exe'
# bot connect
token = '&&&&'
bot = telebot.TeleBot(token)

# mqtt connect
hostname = "mqtt.pi40.ru"
port = 1883
username = "techbkirill"
password = "madshark"
clientID = "nvjrfhruhnv"
client = mqtt.Client('techbkirill')
client.subscribe('techbkirill/charge_stoped')
client.username_pw_set(username, password)
client.connect(hostname, port)

cascade = cv2.CascadeClassifier('haarcascade_russian_plate_number.xml')

user_message = ''
image = 0
gray = 0
plates = ''
result = ''



# start of the bot
@bot.message_handler(commands=['start'])
def send_welcome(message):
    bot.send_message(message.chat.id,
                     f'Добро пожаловать на Enegry Parking! \nВведите пожалуйста номер своей машины латинскими буквами, {message.from_user.first_name}')


@bot.message_handler(content_types=['photo'])
def recieve_photo(message_photo):
    global plates
    global gray
    photo_id = message_photo.photo[-1].file_id
    file_info = bot.get_file(photo_id)
    downloaded_file = bot.download_file(file_info.file_path)
    with open("image.jpg", 'wb') as new_file:
        new_file.write(downloaded_file)
    image = cv2.imread('image.jpg')
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    plates = cascade.detectMultiScale(gray)

@bot.message_handler(content_types=['text'])
def send_reply(message):
    global start_charge
    global last_x_coordinate
    global last_y_coordinate
    if message.text.lower() != '' and len(message.text.lower()) >= 6:
        bot.send_message(message.from_user.id,
        'Спасибо, в скором времени начнется зарядка вашего эллектромобиля \nМы оповестим вас о завершении загрузки.')
        # user_message = str(message.text.lower())
        for plate in plates:
            x, y, w, h = plate
        centreX = int((x + (x + w)) / 2)
        centreY = int((y + (y + h)) / 2)
        cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 255), 10)
        crop = gray[y:y + h, x:x + w]
        # value, mask = cv2.threshold(crop, 100, 255, cv2.THRESH_BINARY_INV)
        # cv2.imshow('window_mask', mask)
        text2 = pytesseract.image_to_string(crop, lang='eng')
        print(text2)
        if len(text2) >= 6 and text2 != '':
            getVals = list([val for val in text2
                            if val.isalpha() or val.isnumeric()])

            text2 = "".join(getVals)
            text2 = text2.lower()
            text2 = text2[0:6]
            print(text2)
            print('readed')
        if charge_stoped == False:
            if text2 == message.text.lower()[0:6] and len(text2) >= 6:
                print("readed")
                if centreX != last_x_coordinate:
                    client.publish("techbkirill/centreX_plate", centreX)
                    last_x_coordinate = centreX
                    print("Angle carnum Sent")
                if centreY != last_y_coordinate:
                    client.publish("techbkirill/centreY_plate", centreY)
                    last_y_coordinate = centreY
                    print("dist carnume sent")
                start_charge = True

            client.publish("techbkirill/start_charge", start_charge)
            start_charge = False
        else:
            bot.send_message(message.from_user.id, 'Зарядка была успешно произведена!\nХорошего вам дня!')

    else:
        bot.send_message(message.from_user.id, 'Простите, но номер машины должен иметь минимум из 6 символов.')
    print(message.text.lower()[0:6])


print('До свидания!')
bot.polling(none_stop=True)
client.loop_forever()
  • Вопрос задан
  • 154 просмотра
Пригласить эксперта
Ответы на вопрос 3
@o5a
Логично, что такие ошибки, если не придерживаться одного цикла. Этот фрагмент
...
        for plate in plates:
            x, y, w, h = plate
        centreX = int((x + (x + w)) / 2)
...

Если предполагалось обрабатывать все найденные знаки, то и все последующие расчеты должны быть внутри цикла, а не вне его.
А если предполагалось обрабатывать только единственный/последний из найденных знаков, то тоже получится ошибка в случае отсутствия определенных знаков на фото.
Ответ написан
Zettabyte
@Zettabyte
Проф. восстановление данных ▪ Вопрос? См. профиль
f'Добро пожаловать на Enegry Parking! \nВведите пожалуйста номер своей машины латинскими буквами, {message.from_user.first_name}')

f'Добро пожаловать на Energy Parking! \nВведите, пожалуйста, номер своей машины латинскими буквами, {message.from_user.first_name}')

'Спасибо, в скором времени начнется зарядка вашего эллектромобиля \nМы оповестим вас о завершении загрузки.')

'Спасибо, в скором времени начнется зарядка вашего электромобиля \nМы оповестим вас о завершении зарядки.')

bot.send_message(message.from_user.id, 'Простите, но номер машины должен иметь минимум из 6 символов.')

bot.send_message(message.from_user.id, 'Простите, но номер машины должен состоять минимум из 6 символов.')

Увы, но мои скиллы пока позволяют поправить только это :-\
Ответ написан
2ord
@2ord
продвинутый чайник
Нужно проверить код статическим анализатором кода. Например, Pylint.
Он укажет на проблемы в коде, которые надо решить. В том числе, и с необъвленными переменными.
Ответ написан
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы
05 авг. 2021, в 17:58
20000 руб./за проект
05 авг. 2021, в 17:55
100000 руб./за проект
05 авг. 2021, в 17:43
3000 руб./за проект