import os
from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters, ContextTypes, CallbackQueryHandler
from src.bot import ThreadsBot

BOT_TOKEN = os.environ.get('THREAD_TELEGRAM_TOKEN')

if __name__ == '__main__':
    bot = ThreadsBot()

    app = ApplicationBuilder().token(BOT_TOKEN).build()

    # app.add_handler(CommandHandler("start", start))
    app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, bot.echo))
    app.add_handler(CallbackQueryHandler(bot.button_callback))

    print("🤖 Bot is running...")
    app.run_polling()

    '''
    client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])



    print(response.choices[0].message.content)

    print("使用的 token：")
    print("  prompt_tokens =", response.usage.prompt_tokens)
    print("  completion_tokens =", response.usage.completion_tokens)
    print("  total_tokens =", response.usage.total_tokens)
    cost = 1.25/1000000 * response.usage.prompt_tokens + 10.00/1000000 * response.usage.completion_tokens
    print("  cost = {} USD, {} NTD".format(cost, cost*30))
    '''