-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtelegramspeechbot.py
90 lines (67 loc) · 4.3 KB
/
telegramspeechbot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import main
import settings
from aiogram import Bot, Dispatcher, types, executor
from aiogram.dispatcher.filters import Command
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton, InlineKeyboardButton, InlineKeyboardMarkup, CallbackQuery
TG_TOKEN = settings.TOKEN
bot = Bot(token=TG_TOKEN)
dp = Dispatcher(bot)
StartButton = KeyboardButton("/help")
SettingsButton = KeyboardButton("/settings")
main_kb = ReplyKeyboardMarkup(resize_keyboard=True).add(StartButton, SettingsButton)
Female_btn = InlineKeyboardButton('Женский', callback_data='FEMALE')
Male_btn = InlineKeyboardButton('Мужской', callback_data='MALE')
inline_kb1 = InlineKeyboardMarkup().add(Female_btn, Male_btn)
Anna_btn = InlineKeyboardButton('Анна Кравчук', callback_data="ru-RU_Anna Kravchuk")
Natalya_btn = InlineKeyboardButton('Наталия Сучова', callback_data="ru-RU_Natalia Sychyov")
FemaleSpeaker_kb = InlineKeyboardMarkup().add(Anna_btn, Natalya_btn)
Alex_btn = InlineKeyboardButton('Алексей Сёмин', callback_data="ru-RU_Alexei Syomi")
Ivan_btn = InlineKeyboardButton('Иван Чкалов', callback_data="ru-RU_Ivan Chkalov")
MaleSpeaker_kb = InlineKeyboardMarkup().add(Alex_btn, Ivan_btn)
@dp.message_handler(commands=['settings'])
async def process_command_1(message: types.Message):
await message.reply(text="Выберите пол спикера:", reply_markup=inline_kb1)
@dp.callback_query_handler(text=["FEMALE", "MALE"])
async def process_command_2(callback: CallbackQuery):
if callback.data == Female_btn.callback_data:
settings.Sex = f"{Female_btn.callback_data}"
await callback.message.edit_text(text="Выберите чей голос использовать:", reply_markup=FemaleSpeaker_kb)
elif callback.data == Male_btn.callback_data:
settings.Sex = f"{Male_btn.callback_data}"
await callback.message.edit_text(text="Выберите чей голос использовать:", reply_markup=MaleSpeaker_kb)
@dp.callback_query_handler(text=["ru-RU_Anna Kravchuk", "ru-RU_Natalia Sychyov", "ru-RU_Alexei Syomi", "ru-RU_Ivan Chkalov"])
async def process_command_2(callback: CallbackQuery):
await callback.message.edit_text(text="Готово!")
if callback.data == Anna_btn.callback_data:
settings.Voice = f"{Anna_btn.callback_data}"
elif callback.data == Natalya_btn.callback_data:
settings.Voice = f"{Natalya_btn}"
if callback.data == Alex_btn.callback_data:
settings.Voice = f"{Alex_btn.callback_data}"
elif callback.data == Ivan_btn.callback_data:
settings.Voice = f"{Ivan_btn.callback_data}"
@dp.message_handler(Command("start"))
async def cmd_start(message: types.Message):
await message.reply(
"Привет! Это бот для конвертации текста в голосовое сообщение.\nНажмите на кнопку для того чтобы узнать как пользоваться ботом\nАвтор: @hkk89",
reply_markup=main_kb
)
@dp.message_handler(Command("help"))
async def cmd_help(message: types.Message):
await message.reply(
"Для того чтобы бот распознал текст:\n🔸 Сообщение должно содержать менее 500 букв.\n🔸 Сообщение не должно содержать математических символов.\n🔸 Бот может не принять сообщение, это ошибка не бота, а используемой нейронной сети.\n🔸 Напишите сообщение боту и ждите.\n🔸 Для настройки голоса бота нажмите вторую кнопку"
)
@dp.message_handler(content_types='text')
async def MainButton(message: types.Message):
await bot.send_message(message.from_user.id, "Текст получен")
try:
main.text_to_speech(message.text)
voice = types.InputFile('speech.wav')
await bot.send_voice(message.from_user.id, voice, caption="Ответ от бота", reply_markup=main_kb)
except AttributeError:
await message.reply("Бот не принял данный текст")
if __name__ == "__main__":
try:
executor.start_polling(dp, skip_updates=True)
except (KeyboardInterrupt, SystemExit):
pass