import logging
from telegram import Update
from telegram.ext import Updater, CommandHandler, CallbackContext
from transformers import pipeline

# Включаем логирование
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)

# Инициализация модели генерации текста
generator = pipeline('text-generation', model='distilgpt2')

# Функция для обработки команды /start
def start(update: Update, context: CallbackContext) -> None:
    update.message.reply_text('Привет! Я бот для ответов на сообщения с Авито.')

# Функция для обработки команды /respond
def respond(update: Update, context: CallbackContext) -> None:
    user_message = update.message.text  # Получаем текст сообщения от пользователя
    response = generate_response(user_message)
    update.message.reply_text(response)

# Функция генерации ответа с помощью модели Hugging Face
def generate_response(message: str) -> str:
    # Генерация ответа с помощью модели
    generated = generator(message, max_length=50, num_return_sequences=1)
    return generated[0]['generated_text']

# Основная функция
def main():
    # Замените 'ВАШ_ТОКЕН' на токен вашего бота
    updater = Updater("7651372274:AAF-BCJL1PTKm_BbG9TpUllCnp8e0ifXPkA", use_context=True)
    dispatcher = updater.dispatcher

    # Обработчики команд
    dispatcher.add_handler(CommandHandler("start", start))
    dispatcher.add_handler(CommandHandler("respond", respond))

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

if __name__ == '__main__':
    main()