Create utils.py

utils.py

@@ -0,0 +1,37 @@
+# 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