Skip to content

Commit

Permalink
fixes for speach recogniztion
Browse files Browse the repository at this point in the history
  • Loading branch information
twowannabe committed Jun 25, 2024
1 parent 76e5b48 commit b50dfd0
Showing 1 changed file with 41 additions and 1 deletion.
42 changes: 41 additions & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,4 +88,44 @@ def handle_voice(update: Update, context: CallbackContext) -> None:
os.remove(wav_file_path)

# Обработчик текстовых сообщений
def handle_message(update: Update, context
def handle_message(update: Update, context: CallbackContext) -> None:
user_id = update.message.from_user.id
user_message = extract_text_from_message(update.message)

# Проверка на повторяющиеся вопросы
question_counters[user_id][user_message] += 1
if question_counters[user_id][user_message] > 3:
update.message.reply_text("Вы уже спрашивали об этом несколько раз. Пожалуйста, задайте другой вопрос.")
return

# Добавляем сообщение пользователя в контекст
conversation_context[user_id].append({"role": "user", "content": user_message})

# Подготавливаем сообщения для отправки в ChatGPT
messages = initial_instructions + conversation_context[user_id]

# Получаем ответ от ChatGPT
reply = ask_chatgpt(messages)

# Добавляем ответ ChatGPT в контекст
conversation_context[user_id].append({"role": "assistant", "content": reply})

# Отправляем ответ пользователю
update.message.reply_text(reply, parse_mode=ParseMode.MARKDOWN)

def main():
# Создаем апдейтера и диспетчера
updater = Updater(TELEGRAM_TOKEN)
dispatcher = updater.dispatcher

# Регистрируем обработчики
dispatcher.add_handler(CommandHandler("start", start))
dispatcher.add_handler(MessageHandler(Filters.text & ~Filters.command, handle_message))
dispatcher.add_handler(MessageHandler(Filters.voice, handle_voice))

# Запуск бота
updater.start_polling()
updater.idle()

if __name__ == '__main__':
main()

0 comments on commit b50dfd0

Please sign in to comment.