bot.py

"""
TechSmart Support Telegram Bot
Main bot module with handlers and commands
"""
import logging
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import (
    Application,
    CommandHandler,
    MessageHandler,
    CallbackQueryHandler,
    ContextTypes,
    filters
)

from config import TELEGRAM_BOT_TOKEN, MESSAGES
from knowledge_base import get_knowledge_base

# Configure logging
logging.basicConfig(
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    level=logging.INFO
)
logger = logging.getLogger(__name__)


# Initialize knowledge base
kb = None


async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle /start command"""
    keyboard = [
        [
            InlineKeyboardButton("📦 Доставка", callback_data="delivery"),
            InlineKeyboardButton("🛡️ Гарантия", callback_data="warranty")
        ],
        [
            InlineKeyboardButton("❓ FAQ", callback_data="faq"),
            InlineKeyboardButton("📞 Оператор", callback_data="operator")
        ]
    ]
    reply_markup = InlineKeyboardMarkup(keyboard)
    
    await update.message.reply_text(
        MESSAGES["welcome"],
        reply_markup=reply_markup
    )


async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle /help command"""
    await update.message.reply_text(MESSAGES["help"])


async def delivery_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle /delivery command"""
    global kb
    if kb:
        info = kb.get_delivery_info()
        await update.message.reply_text(f"📦 *Доставка и возврат*\n\n{info}", parse_mode="Markdown")
    else:
        await update.message.reply_text(MESSAGES["error"])


async def warranty_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle /warranty command"""
    global kb
    if kb:
        info = kb.get_warranty_info()
        await update.message.reply_text(f"🛡️ *Гарантийная политика*\n\n{info}", parse_mode="Markdown")
    else:
        await update.message.reply_text(MESSAGES["error"])


async def faq_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle /faq command"""
    global kb
    if kb:
        info = kb.get_faq()
        await update.message.reply_text(f"❓ *Часто задаваемые вопросы*\n\n{info}", parse_mode="Markdown")
    else:
        await update.message.reply_text(MESSAGES["error"])


async def operator_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle /operator command"""
    await update.message.reply_text(MESSAGES["operator_contact"])


async def button_callback(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle inline button callbacks"""
    query = update.callback_query
    await query.answer()
    
    global kb
    
    if query.data == "delivery":
        if kb:
            info = kb.get_delivery_info()
            await query.message.reply_text(f"📦 *Доставка и возврат*\n\n{info}", parse_mode="Markdown")
        else:
            await query.message.reply_text(MESSAGES["error"])
    
    elif query.data == "warranty":
        if kb:
            info = kb.get_warranty_info()
            await query.message.reply_text(f"🛡️ *Гарантийная политика*\n\n{info}", parse_mode="Markdown")
        else:
            await query.message.reply_text(MESSAGES["error"])
    
    elif query.data == "faq":
        if kb:
            info = kb.get_faq()
            await query.message.reply_text(f"❓ *Часто задаваемые вопросы*\n\n{info}", parse_mode="Markdown")
        else:
            await query.message.reply_text(MESSAGES["error"])
    
    elif query.data == "operator":
        await query.message.reply_text(MESSAGES["operator_contact"])


async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle user text messages - main Q&A functionality"""
    global kb
    
    user_message = update.message.text
    user = update.effective_user
    
    logger.info(f"Question from {user.username or user.id}: {user_message}")
    
    # Send "thinking" message
    thinking_msg = await update.message.reply_text(MESSAGES["thinking"])
    
    try:
        if kb:
            # Get answer from knowledge base
            answer = kb.get_answer(user_message)
            
            if answer:
                await thinking_msg.edit_text(answer)
            else:
                await thinking_msg.edit_text(MESSAGES["no_answer"])
        else:
            await thinking_msg.edit_text(MESSAGES["error"])
            
    except Exception as e:
        logger.error(f"Error processing message: {e}")
        await thinking_msg.edit_text(MESSAGES["error"])


async def error_handler(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    """Handle errors"""
    logger.error(f"Update {update} caused error {context.error}")


def main() -> None:
    """Main function to run the bot"""
    global kb
    
    # Check for token
    if not TELEGRAM_BOT_TOKEN:
        print("❌ Error: TELEGRAM_BOT_TOKEN not set in .env file")
        print("Please copy .env.example to .env and add your bot token")
        return
    
    # Initialize knowledge base
    print("[BOT] Starting TechSmart Support Bot...")
    kb = get_knowledge_base()
    
    # Create application
    application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
    
    # Add handlers
    application.add_handler(CommandHandler("start", start))
    application.add_handler(CommandHandler("help", help_command))
    application.add_handler(CommandHandler("delivery", delivery_command))
    application.add_handler(CommandHandler("warranty", warranty_command))
    application.add_handler(CommandHandler("faq", faq_command))
    application.add_handler(CommandHandler("operator", operator_command))
    application.add_handler(CallbackQueryHandler(button_callback))
    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
    
    # Add error handler
    application.add_error_handler(error_handler)
    
    # Start the bot
    print("[OK] Bot is running! Press Ctrl+C to stop.")
    application.run_polling(allowed_updates=Update.ALL_TYPES)


if __name__ == "__main__":
    main()