Aragorn

import telebot
from telebot import types
import time
import works  # Предполагается, что works теперь словарь

token = 'YOUR_TOKEN'
bot = telebot.TeleBot(token)

# Храним состояние и данные пользователей
user_data = {}

MAIN_MENU = 0
FIRST_WORK_MENU = 1
ADD_WORK_MENU = 2

@bot.message_handler(commands=['start'])
def start(message):
    user_id = message.from_user.id
    # Инициализация данных пользователя при старте
    if user_id not in user_data:
        user_data[user_id] = {
            'state': MAIN_MENU,
            'count_price': 0,
            'count_all': 0
        }
    
    markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
    markup.add(" Главное меню", "ℹ️ Информация", "ℹ️️ начать подсчет зарплаты",
               "ℹ️️ итог за месяц", "ℹ️ итог за все время")
    
    bot.send_message(message.chat.id, "Добро пожаловать! Выберите пункт меню:", reply_markup=markup)

@bot.message_handler(content_types=['text'])
def handle_text(message):
    user_id = message.from_user.id
    # Инициализируем данные, если пользователь новый
    if user_id not in user_data:
        user_data[user_id] = {
            'state': MAIN_MENU,
            'count_price': 0,
            'count_all': 0
        }
    
    current_user = user_data[user_id]

    if message.text == " Главное меню":
        # Код создания главного меню...
    
    elif message.text == "ℹ️️ начать подсчет зарплаты":
        # Код перехода в меню подсчета...
    
    elif message.text == "добавление выполненых работ":
        # Код меню добавления работ...
    
    elif message.text == "ℹ️️ итог за месяц":
        bot.send_message(message.chat.id, f'Сумма за месяц: {current_user["count_price"]} руб.')
    
    elif message.text == "ℹ️ итог за все время":
        bot.send_message(message.chat.id, f'Сумма за все время: {current_user["count_all"]} руб.')
    
    elif message.text == "обнуление месяца/сохранение итогов":
        current_user['count_all'] += current_user['count_price']
        current_user['count_price'] = 0
        bot.send_message(message.chat.id, 
                       f'Месяц обнулен. Текущий месяц: 0 руб.\n'
                       f'Итог за все время: {current_user["count_all"]} руб.')
    
    elif message.text in works:  # Предполагается, что works - это словарь
        price = works[message.text]
        current_user['count_price'] += price
        bot.send_message(message.chat.id,
                       f'{message.text}: {price} руб.\n'
                       f'Итог за месяц: {current_user["count_price"]} руб.')

bot.polling(none_stop=True)