# utils.py
import logging
import numpy as np
from PIL import Image
from hydrogram import filters
from hydrogram.types import Message

from config import Config # Import ADMIN_IDS from config

logger = logging.getLogger(__name__)

# --- Admin Check Filter ---
async def admin_filter_func(_, __, message: Message):
    if message.from_user and message.from_user.id in Config.ADMIN_IDS:
        logger.debug(f"Message from admin user {message.from_user.id} allowed.")
        return True
    elif message.from_user:
        logger.info(f"Message from non-admin user {message.from_user.id} (name: {message.from_user.first_name}) blocked by admin filter.")
        return False
    else:
        logger.debug("Message without from_user blocked by admin filter.")
        return False

is_admin = filters.create(admin_filter_func, name="AdminFilter")

# --- Image Helper Functions ---
def add_noise_to_image(img: Image.Image, intensity: float = 0.02) -> Image.Image:
    try:
        if img.mode != 'RGB': img = img.convert('RGB')
        img_array = np.array(img, dtype=np.float32) / 255.0
        noise = np.random.randn(*img_array.shape) * intensity
        noisy_img_array = np.clip(img_array + noise, 0.0, 1.0)
        noisy_img = Image.fromarray((noisy_img_array * 255).astype(np.uint8), 'RGB')
        return noisy_img
    except Exception as e:
        logger.error(f"Error adding noise: {e}", exc_info=True)
        return img