import os
from aiogram import Bot, Dispatcher, types
from aiogram.filters import Command
import asyncio
from huggingface_hub import InferenceClient

# --- НАСТРОЙКИ ---
TOKEN = os.getenv('BOT_TOKEN')
HF_TOKEN = os.getenv('HF_TOKEN')

bot = Bot(token=TOKEN)
dp = Dispatcher()
client = InferenceClient("IlyaGusev/saiga_llama_3_8b", token=HF_TOKEN)

# --- КОМАНДА /CIRCUS (УЦЦ) ---
@dp.message(Command("Circus"))
async def send_circus(message: types.Message):
    await message.answer(
        "🎪 **ДОБРО ПОЖАЛОВАТЬ В ЦИФРОВОЙ ЦИРК, БРО!** 🎪\n\n"
        "Илюха одобряет этот вайб. Садись поудобнее, хватай компот и смотри пилотную серию УЦЦ!\n"
        "Тут реальный 52-й уровень безумия! 👇\n\n"
        "🔗 https://rutube.ru/video/c31ea3e4893c60c18204cdd3e1931d30/"
    )

# --- ОСНОВНОЙ ЧАТ С ИЛЬЕЙ ---
@dp.message()
async def talk_to_ilya(message: types.Message):
    # Промпт Самурая
    prompt = f"Ты — Илья, бро из банды 52. Ты шаришь за сленг, любишь компот и уважаешь Никиту. Отвечай коротко. Вопрос: {message.text}"
    
    # Генерируем ответ через Хаггинг
    response = client.text_generation(prompt, max_new_tokens=200)
    await message.answer(response)

# --- ЗАПУСК ---
async def main():
    print("🚀 Космический Илья с Цифровым Цирком запущен!")
    await dp.start_polling(bot)

if __name__ == "__main__":
    asyncio.run(main())