Upload bot.py

bot.py

@@ -528,7 +528,11 @@ GEMINI_MODELS: dict[str, dict[str, Any]] = {
     "gemma-4-31b-it": {
         "name": "Gemma 4 31B", "badge": "Открытая модель",
         "desc": "Флагманская открытая модель Google на 31 млрд параметров — топ-3 среди открытых моделей мира по качеству рассуждений, кода и работы с изображениями.", "no_system": True, "no_search": True, "stream": True,
-    }
+    },
+    "gemma-4-26b-a4b-it": {
+        "name": "Gemma 4 26B", "badge": "Открытая модель",
+        "desc": "Компактная открытая модель Google на 26 млрд параметров с поддержкой расширенного мышления (thinking). Хороший баланс качества и скорости среди Gemma-семейства.", "no_system": True, "stream": True,
+    },
 }
 DEFAULT_GEMINI_MODEL = "gemini-3.5-flash"
 
@@ -849,35 +853,38 @@ HF_IMAGE_MODEL_PAGE_SIZE = 8
 HF_IMAGE_MODEL_CALLBACKS: dict[str, str] = {}
 
 HF_IMAGE_MODELS: dict[str, dict[str, Any]] = {
-    "black-forest-labs/FLUX.1-dev": {
-        "name": "FLUX.1 Dev", "badge": "⭐ Лучшая",
-        "desc": "Флагманская модель Black Forest Labs. Фотореализм, точное следование промпту, детализация — эталон качества для FLUX-линейки."
+    "pollinations:flux": {
+        "name": "FLUX Pro",
+        "desc": "Высококачественный FLUX через Pollinations.ai. Фотореализм, точное следование промпту, богатая детализация.",
     },
-    "black-forest-labs/FLUX.1-Kontext-dev": {
-        "name": "FLUX.1 Kontext", "badge": "Редактирование",
-        "desc": "Специализирована на редактировании изображений и следовании контексту. Позволяет точечно менять стиль, объекты и фон по текстовой инструкции."
+    "pollinations:flux-realism": {
+        "name": "FLUX Realism",
+        "desc": "FLUX с акцентом на гиперреализм — детализированные текстуры, естественное освещение, кинематографичность.",
     },
-    "stabilityai/stable-diffusion-3.5-large": {
-        "name": "SD 3.5 Large", "badge": "Качество",
-        "desc": "Крупная модель Stability AI третьего поколения. Высокая детализация, понимание сложных промптов, улучшенная анатомия и текст внутри изображений."
+    "pollinations:flux-anime": {
+        "name": "FLUX Anime",
+        "desc": "FLUX для аниме и иллюстраций — характерные пропорции, яркие цвета, стилизация под японскую графику.",
     },
-    "black-forest-labs/FLUX.1-schnell": {
-        "name": "FLUX.1 Schnell", "badge": "Быстрая",
-        "desc": "Дистиллированная версия FLUX.1 Dev для скоростной генерации. 1–4 шага вместо 20–50 — результат за секунды при хорошем качестве."
+    "pollinations:turbo": {
+        "name": "Turbo",
+        "desc": "Быстрая дистиллированная модель. Результат за несколько секунд — для черновиков и быстрых итераций.",
     },
-    "stabilityai/stable-diffusion-xl-base-1.0": {
-        "name": "SDXL Base 1.0", "badge": "Классика",
-        "desc": "Проверенная базовая модель Stable Diffusion XL. Широкое сообщество, тысячи LoRA и стилей, стабильный предсказуемый результат."
+    "pollinations:dreamshaper": {
+        "name": "DreamShaper",
+        "desc": "Художественная модель для фэнтези, концепт-арта и стилизованных иллюстраций.",
     },
-    "stabilityai/sdxl-turbo": {
-        "name": "SDXL Turbo", "badge": "Сверхбыстрая",
-        "desc": "Adversarial дистилляция SDXL — генерация за 1–4 шага. Идеально для быстрых набросков, итераций и проверки идей."
+    "black-forest-labs/FLUX.1-schnell": {
+        "name": "FLUX.1 Schnell (HF)",
+        "desc": "Оригинальная FLUX.1 Schnell от Black Forest Labs через Hugging Face Inference API. Быстрая, Apache 2.0.",
     },
 }
 
 
 def _hf_image_headers() -> dict[str, str]:
-    headers = {"Accept": "application/json, image/*, application/octet-stream", "User-Agent": "IlvorBot/8.2"}
+    headers = {
+        "Accept": "image/png, image/jpeg, image/webp, application/json",
+        "User-Agent": "IlvorBot/8.2"
+    }
     if HF_TOKEN:
         headers["Authorization"] = f"Bearer {HF_TOKEN}"
     return headers
@@ -894,9 +901,7 @@ async def _hf_fetch_model_catalog(force: bool = False) -> list[dict[str, Any]]:
 
 def _imgmodel_button_text(model: dict[str, Any], current: str) -> str:
     prefix = "• " if model["id"] == current else ""
-    badge = model.get("badge")
-    suffix = f" [{badge}]" if badge else ""
-    return f"{prefix}{model['name']}{suffix}"
+    return f"{prefix}{model['name']}"
 
 
 def _imgmodel_keyboard(current: str, page: int = 0) -> InlineKeyboardMarkup:
@@ -925,7 +930,31 @@ def _imgmodel_keyboard(current: str, page: int = 0) -> InlineKeyboardMarkup:
     return InlineKeyboardMarkup(inline_keyboard=rows)
 
 
+async def _pollinations_generate(model_name: str, prompt: str) -> bytes:
+    """Бесплатная генерация через Pollinations.ai — не требует авторизации."""
+    session = await _get_http_session()
+    encoded = urllib.parse.quote(prompt[:600], safe="")
+    url = (
+        f"https://image.pollinations.ai/prompt/{encoded}"
+        f"?width=1024&height=1024&model={model_name}&nologo=true&enhance=false"
+    )
+    async with session.get(url, timeout=aiohttp.ClientTimeout(total=90)) as resp:
+        if resp.status == 200:
+            ctype = (resp.headers.get("Content-Type") or "").lower()
+            body = await resp.read()
+            if body and (ctype.startswith("image/") or body[:4] in (b"\x89PNG", b"\xff\xd8\xff", b"RIFF", b"GIF8")):
+                return body
+            raise RuntimeError(f"Pollinations вернул не-изображение: {ctype}")
+        raise RuntimeError(f"Pollinations.ai HTTP {resp.status}")
+
+
 async def _hf_text_to_image(model_id: str, prompt: str) -> bytes:
+    # Pollinations.ai — бесплатный провайдер без авторизации
+    if model_id.startswith("pollinations:"):
+        poll_model = model_id.split(":", 1)[1]
+        return await _pollinations_generate(poll_model, prompt)
+
+    # HF Inference API (router.huggingface.co/hf-inference)
     session = await _get_http_session()
     url = f"{HF_IMAGE_API_BASE}/models/{urllib.parse.quote(model_id, safe='')}"
     payload = {
@@ -1889,6 +1918,7 @@ async def ask_gemini(chat_id: int, user_text: str, media: list[tuple[bytes, str]
         "gemini-2.5-flash",
         "gemini-2.5-flash-lite",
         "gemma-4-31b-it",
+        "gemma-4-26b-a4b-it",
     ]
 
     loop_guard = 0
@@ -2215,13 +2245,18 @@ def gemini_kb(current: str) -> InlineKeyboardMarkup:
     return InlineKeyboardMarkup(inline_keyboard=grid)
 
 def _get_gemini_model_limit_info(model_id: str) -> str:
+    # Реальные лимиты по официальному дашборду AI Studio (June 2026).
+    # Строка вида "32 / 20" означает: пик использования / лимит.
+    # 3.x Flash/Preview = 20-25 RPD; 3.1 Flash-Lite = 500 RPD;
+    # 2.5 Flash = 20 RPD; 2.5 Flash-Lite = 20 RPD; Gemma 4 = 1500 RPD.
     limits_map = {
-        "gemini-3.5-flash": 1500,
-        "gemini-3-flash-preview": 1500,
+        "gemini-3.5-flash": 20,
+        "gemini-3-flash-preview": 25,
         "gemini-3.1-flash-lite": 500,
-        "gemini-2.5-flash": 250,
-        "gemini-2.5-flash-lite": 1000,
+        "gemini-2.5-flash": 20,
+        "gemini-2.5-flash-lite": 20,
         "gemma-4-31b-it": 1500,
+        "gemma-4-26b-a4b-it": 1500,
     }
     limit = limits_map.get(model_id, 20)
     e = _quota_entry(None, "gemini", model_id)