Main version

Author: MinIN-git

README.md

@@ -1 +1,15 @@
-# Tesseract-OCR
+# Hello!
+*Based on Google's artificial intelligence (Tesseract OCR), 
+a telegram bot was written. The bot's job is to `read the 
+sent photo and output the entire text from it`. A database 
+is connected to the board for further paid subscriptions. 
+The bot is still in development, so it is not deploided.*    
+
+# Привет!
+*На основе искусственного интеллекта от Google (Teseract OCR) 
+был написан бот в телеграмме. Работа бота заключается в `чтении 
+отправленной фотографии и вывода с нее всего текста`. Подключена 
+база данных для дальнейшей платный подписки. Бот 
+еще находится в разработке, поэтому не задеплоен.*
+
+![](../../../Downloads/2f26bf1ca7.jpg)

config.py

@@ -0,0 +1,8 @@
+from aiogram.dispatcher.filters.state import StatesGroup, State
+
+TOKEN = '1734420280:AAEUIk9Hzee_WmAllt_Ok-d1PyB8hUzAljs'
+
+class FSM(StatesGroup):
+    language_selection = State()
+
+

db.db

@@ -0,0 +1,104 @@
+import logging
+from asyncio import sleep
+
+from PIL import Image
+import pytesseract
+
+from aiogram import Bot, Dispatcher, executor, types
+from aiogram.contrib.fsm_storage.memory import MemoryStorage
+from aiogram.dispatcher import FSMContext
+
+from config import TOKEN, FSM
+from sqlighter import SQLighter
+
+
+# Чтобы tesseract работал на винде
+pytesseract.pytesseract.tesseract_cmd = 'C:\\Program Files (x86)\\tesseract.exe'
+
+# parse_mode=types.ParseMode.HTML - чтобы можно было ставить теги как в html
+bot = Bot(token=TOKEN, parse_mode=types.ParseMode.HTML)
+
+# инициализируем соединение с БД
+db = SQLighter('db.db')
+
+dp = Dispatcher(bot, storage=MemoryStorage())
+logging.basicConfig(level=logging.INFO)
+
+
+@dp.message_handler(commands='start')
+async def command_start(message: types.Message):
+    await message.reply("<b>Привет! Отправь фото с текстом (не рукописным), чтобы увидеть меня в действие."
+                        "\n\nBeta version 1.0\nПоддержка: [email protected]</b>")
+
+    # Добавление пользователя в БД
+    if (not db.subscriber_exists(message.from_user.id)):
+        # если юзера нет в базе, добавляем его
+        db.add_subscriber(message.from_user.id)
+    else:
+        # если он уже есть, то просто обновляем ему статус подписки
+        db.update_subscription(message.from_user.id, True)
+
+@dp.message_handler(commands='help')
+async def command_help(message: types.Message):
+    await message.reply("<b>На основе искусственного интеллекта от Google (Teseract OCR) был написан этот бот. "
+                        "Работа бота заключается в чтении отправленной фотографии и вывода с нее всего текста. "
+                        "К боту подключен база данных для дальнейшей платный подписки. Пока все бесплатно :D"
+                        "\n\nBeta version 1.0\nПоддержка: [email protected]</b>")
+
+@dp.message_handler(commands='game')
+async def game(message: types.Message):
+    await message.answer('Привет, сейчас я запущу кубики...')
+    await sleep(1)
+
+    await message.answer('Я кидаю первый!')
+    bot_data = await bot.send_dice(message.from_user.id)
+    bot_data = bot_data["dice"]["value"]
+    await sleep(5)
+
+    user_data = await bot.send_dice(message.from_user.id)
+    user_data = user_data["dice"]["value"]
+    await sleep(5)
+
+    if bot_data > user_data:
+        await message.answer('Вы проиграли, попробуйте еще раз попозже ;(')
+    elif bot_data < user_data:
+        await message.answer('Сейчас победа за вами, но это не надолго!')
+    else:
+        await message.answer('Боевая ничья!')
+
+@dp.message_handler(content_types=['photo'])
+async def download_photo(message: types.Message):
+    await message.reply('Какой язык на этой фотографии?'
+                        '\n\nВозможные языки: \nrus - Русский, eng - English')
+    await message.photo[-1].download("work_photo.jpg")
+
+    await FSM.language_selection.set()
+    # Сделать выбор языка инлайн кнопкой
+
+@dp.message_handler(state=FSM.language_selection)
+async def language_selection(message: types.Message, state=FSMContext):
+    image = Image.open("work_photo.jpg")
+    image = image.convert('1')
+
+    language = message.text.lower()
+
+    async with state.proxy() as data:
+        data['language'] = language
+
+    if language == 'rus':
+        text = pytesseract.image_to_string(image, lang="rus")
+    elif language == 'eng':
+        text = pytesseract.image_to_string(image, lang="eng")
+
+    await message.answer(f'Текст с фото: \n{text}')
+    await state.finish()
+ 
+@dp.message_handler()
+async def response_to_user_message(message: types.Message):
+    if message.text == 'admin':
+        await message.answer('Ввдите пароль, чтобы зайти под именем администратора')
+    else:
+        await message.reply('Я был создан для работы с фото, давай каждый будет заниматься тем, что он умеет...')
+
+if __name__ == '__main__':
+    executor.start_polling(dp, skip_updates=True)

main.py

@@ -0,0 +1,33 @@
+import sqlite3
+
+class SQLighter:
+
+    def __init__(self, database):
+        """Подключаемся к БД и сохраняем курсор соединения"""
+        self.connection = sqlite3.connect(database)
+        self.cursor = self.connection.cursor()
+
+    def get_subscriptions(self, status = True):
+        """Получаем всех активных подписчиков бота"""
+        with self.connection:
+            return self.cursor.execute("SELECT * FROM `subscriptions` WHERE `status` = ?", (status,)).fetchall()
+
+    def subscriber_exists(self, user_id):
+        """Проверяем, есть ли уже юзер в базе"""
+        with self.connection:
+            result = self.cursor.execute('SELECT * FROM `subscriptions` WHERE `user_id` = ?', (user_id,)).fetchall()
+            return bool(len(result))
+
+    def add_subscriber(self, user_id, status = True):
+        """Добавляем нового подписчика"""
+        with self.connection:
+            return self.cursor.execute("INSERT INTO `subscriptions` (`user_id`, `status`) VALUES(?,?)", (user_id,status))
+
+    def update_subscription(self, user_id, status):
+        """Обновляем статус подписки пользователя"""
+        with self.connection:
+            return self.cursor.execute("UPDATE `subscriptions` SET `status` = ? WHERE `user_id` = ?", (status, user_id))
+
+    def close(self):
+        """Закрываем соединение с БД"""
+        self.connection.close()