Интеграция с Telegram ботом

from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
import requests

# Инициализация бота с вашими токенами
def init_bot(telegram_token, filebrain_api_key, project_id):
    updater = Updater(token=telegram_token, use_context=True)
    dispatcher = updater.dispatcher

    # Обработка команды /start
    def start(update, context):
        update.message.reply_text('Привет! Отправь мне сообщение для поиска в ваших документах.')

    # Обработка сообщений
    def handle_message(update, context):
        query = update.message.text
        
        # Вызов API FileBrain
        response = requests.post(
            'https://filebrain.pro/api/chat/v1/',
            headers={'Authorization': f'Bearer {filebrain_api_key}'},
            json={
                'message': query,
                'project_id': project_id
            }
        )
        
        if response.status_code == 200:
            answer = response.json()['answer']
            update.message.reply_text(answer)
        else:
            update.message.reply_text('Извините, произошла ошибка.')

    # Регистрация обработчиков
    dispatcher.add_handler(CommandHandler("start", start))
    dispatcher.add_handler(MessageHandler(Filters.text, handle_message))

    return updater

# Запуск бота
if __name__ == '__main__':
    updater = init_bot(
        'ВАШ_TELEGRAM_TOKEN',
        'ВАШ_FILEBRAIN_API_KEY',
        'ВАШ_PROJECT_ID'
    )
    updater.start_polling()
    updater.idle()