bot to logging

import telebot
import random

TOKEN = 'your_bot_token_here'

bot = telebot.TeleBot(TOKEN)

@bot.message_handler(commands=['start'])
def start(message):
    markup = telebot.types.ReplyKeyboardMarkup()
    markup.add("Искать", "Взломать сундук")
    bot.send_message(message.chat.id, "Привет! Давай поищем сокровища!", reply_markup=markup)
    bot.register_next_step_handler(message, start_handler)

def start_handler(message):
    if message.text == "Искать":
        search_treasure(message)
    elif message.text == "Взломать сундук":
        play_game(message)

def search_treasure(message):
    if random.random() < 0.3:
        treasure = random.choice(["золото", "драгоценные камни", "сокровищница"])
        bot.send_message(message.chat.id, f"Поздравляю! Ты нашел {treasure}!")
    else:
        bot.send_message(message.chat.id, "К сожалению, здесь только камни и булыжники.")

def generate_random_code():
    return str(random.randint(1000, 9999))

def play_game(message):
    correct_code = generate_random_code()
    button_numbers = [correct_code] + [generate_random_code() for _ in range(4)]
    random.shuffle(button_numbers)
    markup = telebot.types.ReplyKeyboardMarkup(row_width=2)
    for number in button_numbers:
        markup.add(telebot.types.KeyboardButton(number))
    bot.send_message(message.chat.id, "Отгадай пароль от сундука, выбрав "
                                      "правильное 4-значное число:", reply_markup=markup)
    bot.register_next_step_handler(message, check_answer, correct_code)

def check_answer(msg, correct_code):
    if int(msg.text) == correct_code:
        treasure = random.choice(["золото", "драгоценные камни", "сокровищница"])
        bot.send_message(msg.chat.id, f"Поздравляю! Ты нашел {treasure}!")
    else:
        bot.send_message(msg.chat.id, "К сожалению, здесь только камни и булыжники.")


bot.infinity_polling()