import logging from telegram import Update from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes from utils import clean_text from os import environ from app import generate_response, detect_language # Configure logging logging.basicConfig( format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO ) logger = logging.getLogger(__name__) # Get token from environment variable TOKEN = environ.get("TELEGRAM_BOT_TOKEN") if not TOKEN: raise ValueError("No TELEGRAM_BOT_TOKEN found in environment variables!") async def start(update: Update, context: ContextTypes.DEFAULT_TYPE): """Handler for the /start command.""" if not update.message or not update.message.text: return await update.message.reply_text("Hello! Tell me your decision-making issue, and I'll try to help.") logger.info("Start command received.") # ------------------------- # Message handler for incoming text messages # ------------------------- async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE): """ Handler for incoming messages. Sends the user's message to the API and replies with the response. """ try: if not update.message or not update.message.text: return user_text = clean_text(update.message.text) logger.info(f"User message: {user_text}") response = generate_response(user_text) await update.message.reply_text(response) except Exception as e: logger.error(f"Error processing message: {e}") # async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE): # """Handle regular messages.""" # if not update.message or not update.message.text: # return # text = clean_text(update.message.text) # if '?' in text: # response = generate_response(text) # else: # user_lang = detect_language(text) # response = get_response('no_question', user_lang) # await update.message.reply_text(response) # async def error_handler(update: Update, context: ContextTypes.DEFAULT_TYPE): # """Handle errors.""" # logger.error(f"Error occurred: {context.error}") # try: # if update and update.message: # user_lang = detect_language(update.message.from_user.language_code) # await update.message.reply_text( # get_response('error', user_lang) # ) # except Exception as e: # logger.error(f"Error in error handler: {e}") def main(): """Start the bot.""" # Create the application application = Application.builder().token(TOKEN).build() # Add handlers application.add_handler(CommandHandler("start", start)) # application.add_handler(CommandHandler("help", help_command)) # application.add_handler(CommandHandler("decide", decide)) application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message)) # Add error handler # application.add_error_handler(error_handler) # Start the bot application.run_polling(allowed_updates=Update.ALL_TYPES) if __name__ == '__main__': main()