config.py

import os
from dotenv import load_dotenv

load_dotenv()

# Telegram
TG_BOT_TOKEN = os.getenv("TG_BOT_TOKEN")
ADMIN_ID = int(os.getenv("ADMIN_ID", 0))

# Cloudflare Worker (прокси для Telegram API)
WORKER_URL = os.getenv("WORKER_URL", "https://telegram-forwarder.rubl1313.workers.dev").strip()

# Cloudflare AI (для генерации изображений)
CF_WORKER_URL = os.getenv("CF_WORKER_URL")
CF_API_KEY = os.getenv("CF_API_KEY")
CF_MODEL = os.getenv("CF_MODEL", "@cf/black-forest-labs/flux-2-klein-4b")

# FaceFusion API (для замены лиц)
FACEFUSION_URL = os.getenv("FACEFUSION_URL", "https://Dmitry1313-facefusion-api.hf.space").strip()

# Настройки
BOT_NAME = "🎨 AI PhotoStudio"
DAILY_LIMIT = int(os.getenv("DAILY_LIMIT", 50))
GENERATION_TIMEOUT = 90
MAX_REFERENCE_IMAGES = 2  # Ограничиваем для бесплатного тарифа

def check_config():
    """Проверяет наличие критически важных переменных."""
    missing = []
    if not TG_BOT_TOKEN:
        missing.append("TG_BOT_TOKEN")
    if not WORKER_URL:
        missing.append("WORKER_URL")
    # CF_WORKER_URL и CF_API_KEY могут отсутствовать, если генерация не используется
    if missing:
        raise ValueError(f"❌ Отсутствуют обязательные переменные окружения: {', '.join(missing)}")
    return True