120 lines
5.0 KiB
Python
120 lines
5.0 KiB
Python
|
from aiogram.filters.callback_data import CallbackData
|
|||
|
|
from aiogram.types import InlineKeyboardButton
|
|||
|
|
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
|||
|
|
|
|||
|
|
|
|||
|
|
models_description = {
|
|||
|
|
"1o": "Это базовая модель ИИ, которая \"умеет думать\". Она хороша для выполнения повседневных задач, \n\
|
|||
|
|
таких как анализ данных, обработка текста и создание отчетов. Она подходит для малого и среднего бизнеса, \n\
|
|||
|
|
где важно автоматизировать рутинные задачи и получать быстрые результаты.\n\n\
|
|||
|
|
Выберите тариф к модели 1о.",
|
|||
|
|
"1o mini": "Уменьшенная версия 1о, которая также \"умеет думать\". Эта модель менее мощная, \n\
|
|||
|
|
но зато работает быстрее на устройствах с ограниченными ресурсами, таких как смартфоны или небольшие компьютеры. \n\
|
|||
|
|
Прекрасно подойдет для мобильных приложений и стартапов, где ресурсы ограничены, но нужны умные решения.\n\n\
|
|||
|
|
Выберите тариф к модели 1о mini.",
|
|||
|
|
"4o": "Это мощная модель ИИ, предназначенная для сложных и объемных задач. Она хорошо справляется \n\
|
|||
|
|
с глубоким анализом больших данных, сложными прогнозами и инновационными бизнес-проектами. \n\
|
|||
|
|
Идеальна для крупных компаний, требующих высокую производительность для динамичного принятия решений и стратегического анализа.\n\n\
|
|||
|
|
Выберите тариф к модели 4о.",
|
|||
|
|
"4o mini": "Уменьшенная версия модели 4о. Она сохраняет высокую мощность при меньшем использовании ресурсов. \n\
|
|||
|
|
Подходит для сложных бизнес-задач в условиях ограниченных ресурсов, таких как аналитика в режиме реального времени \n\
|
|||
|
|
на портативных и мобильных устройствах. Отличный выбор для бизнеса, который хочет получить максимум от своих технологий \n\
|
|||
|
|
без больших затрат на оборудование.\n\n\
|
|||
|
|
Выберите тариф к модели 4о."
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
model_prices = {
|
|||
|
|
"1o" : {
|
|||
|
|
"base": {
|
|||
|
|
"tokens": "60",
|
|||
|
|
"amount": "2000"
|
|||
|
|
},
|
|||
|
|
"advanced": {
|
|||
|
|
"tokens": "128",
|
|||
|
|
"amount": "3500"
|
|||
|
|
},
|
|||
|
|
"pro": {
|
|||
|
|
"tokens": "300",
|
|||
|
|
"amount": "7500"
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
"1o mini": {
|
|||
|
|
"base": {
|
|||
|
|
"tokens": "50",
|
|||
|
|
"amount": "900"
|
|||
|
|
},
|
|||
|
|
"advanced": {
|
|||
|
|
"tokens": "100",
|
|||
|
|
"amount": "1500"
|
|||
|
|
},
|
|||
|
|
"pro": {
|
|||
|
|
"tokens": "270",
|
|||
|
|
"amount": "2700"
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
"4o": {
|
|||
|
|
"base": {
|
|||
|
|
"tokens": "100",
|
|||
|
|
"amount": "1500"
|
|||
|
|
},
|
|||
|
|
"advanced": {
|
|||
|
|
"tokens": "150",
|
|||
|
|
"amount": "2000"
|
|||
|
|
},
|
|||
|
|
"pro": {
|
|||
|
|
"tokens": "303",
|
|||
|
|
"amount": "3750"
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
"4o mini": {
|
|||
|
|
"base": {
|
|||
|
|
"tokens": "800",
|
|||
|
|
"amount": "400"
|
|||
|
|
},
|
|||
|
|
"advanced": {
|
|||
|
|
"tokens": "1650",
|
|||
|
|
"amount": "650"
|
|||
|
|
},
|
|||
|
|
"pro": {
|
|||
|
|
"tokens": "3300",
|
|||
|
|
"amount": "1250"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
name_of_model = {
|
|||
|
|
"base" : "Базовый",
|
|||
|
|
"advanced": "Продвинутый",
|
|||
|
|
"pro" : "Профессиональный"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
class PayCallBack(CallbackData, prefix="pay"):
|
|||
|
|
name_of_model: str
|
|||
|
|
option_of_model: str
|
|||
|
|
tokens: str
|
|||
|
|
amount: str
|
|||
|
|
|
|||
|
|
def get_model_page(*, name_of_model: str, sizes: tuple[int] = (1,)):
|
|||
|
|
text: str = models_description[name_of_model]
|
|||
|
|
|
|||
|
|
temp_kbds: InlineKeyboardBuilder = InlineKeyboardBuilder()
|
|||
|
|
|
|||
|
|
for option_name, item in models_description[name_of_model].items():
|
|||
|
|
option_name_rus, tokens, amount = name_of_model[option_name], item["tokens"], item["amount"]
|
|||
|
|
temp_text = f"{name_of_model[option_name]} \n {tokens} запросов за {amount} RUB"
|
|||
|
|
temp_kbds.add(InlineKeyboardButton(
|
|||
|
|
text=temp_text,
|
|||
|
|
callback_data=PayCallBack(
|
|||
|
|
name_of_model=name_of_model,
|
|||
|
|
option_of_model=option_name_rus,
|
|||
|
|
tokens=tokens,
|
|||
|
|
amount=amount
|
|||
|
|
).pack()
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
return text, temp_kbds.adjust(*sizes).as_markup()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|