import os
import telebot
from groq import Groq

# Hugging Face Secrets থেকে টোকেনগুলো নেওয়া হচ্ছে
BOT_TOKEN = os.environ.get('bot_token')
GROQ_API_KEY = os.environ.get('groq_api')

bot = telebot.TeleBot(BOT_TOKEN)
client = Groq(api_key=GROQ_API_KEY)

# আপনার সেই কঠোর সিস্টেম প্রম্পট
SYSTEM_PROMPT = {
    "role": "system",
    "content": "তুমি একজন কঠোর কিন্তু সাপোর্টিভ স্টাডি মেন্টর। সংক্ষেপে বাংলায় কথা বলবে। সময়ের গুরুত্ব বোঝাবে এবং কোনো আলতু-ফালতু কথা বলবে না।"
}

@bot.message_handler(func=lambda message: True)
def chat_with_ai(message):
    try:
        # Groq API কল করা হচ্ছে
        chat_completion = client.chat.completions.create(
            messages=[
                SYSTEM_PROMPT,
                {"role": "user", "content": message.text}
            ],
            model="llama3-8b-8192",
        )
        
        # এআই এর উত্তর টেলিগ্রামে পাঠানো
        reply = chat_completion.choices[0].message.content
        bot.reply_to(message, reply)
        
    except Exception as e:
        bot.reply_to(message, "এখন কথা বলা যাবে না, পড়ার টেবিলে যাও!")

if __name__ == "__main__":
    print("বোট চালু হয়েছে...")
    bot.infinity_polling()