import os
from prompts import SYSTEM_PROMPTS, IMAGE_PROMPT
from dotenv import load_dotenv

load_dotenv()

ADMIN_ID = 8243488179

MAX_CONTEXT = 7
MAX_PROMPT = 1500
MAX_TOKENS = 1500

BEN_IDS = [-1002147940521, -1001834656942, -1003142512395]
REPORN_IDS = [-1002147940521, -1002775610996, -1001834656942, -1003142512395]

BOT_USERNAME = "Uzbek_GPTrobot"
BOT_CALLS = ["узбек", "uzbek", f"@{BOT_USERNAME}"]

DEFAULT_MODEL = "ollama*gemma4:31b-cloud"
IMAGE_MODEL = "ollama*gemma3:27b-cloud"
DEFAULT_STREAM = "none"
DEFAULT_CALL = "da"
DEFAULT_PROMPT = "стандартный"
DEFAULT_NOTIFY = "net"

# SHOW_ACC_MSG показывать ли сообщение, что владелец выключил команду
SHOW_ACC_MSG = True
# команды
SETTINGS_COMMAND = True
CLEAR_COMMAND = True
PROMPT_COMMAND = True
CONFIG_COMMAND = True

ABOUT_BOT = (
    "✅ *об узбекгпт* ✅\n"
    "- создал @zsh69\n"
    "- репозиторий бота: https://git.hkr.at.by/cgit.cgi/uzbekgpt\n"
    "- `git clone https://g.hkr.at.by/uzbekgpt`\n"
    "- подпешис: t.me/hukerass"
)

DONATE_LINK = "https://t.me/send?start=IVRLB58MrhCp"

PROVIDERS = {
    "ollama": {
        "name": "оллама",
        "module": "openai",
        "url": "http://127.0.0.1:11434/v1",
        "key": "da",
    },
    # "hzai": {
    #     "name": "хуйзнает ai",
    #     "module": "openai",
    #     "url": os.getenv("HZAI_URL"),
    #     "key": os.getenv("HZAI_KEY"),
    # },
    # "openrouter": {
    #     "name": "опенроутер",
    #     "module": "openai",
    #     "url": "https://openrouter.ai/api/v1/",
    #     "key": os.getenv("ORAI_KEY")
    # }
}

MODELS = {
    "ollama": {
        "gemma3:27b-cloud": {"n_tok": 7, "e_tok": 15},
        "gemma4:31b-cloud": {"n_tok": 30, "e_tok": 50},
    },
    # "hzai": {
    #     "google/gemma-3-27b-it": {"n_tok": 60, "e_tok": 120},
    #     "allah-2-7b": {"n_tok": 50, "e_tok": 100},
    # },
}

API_TOKEN = os.getenv("API_TOKEN")
