trying to fix admins can only use it

DragMusic/plugins/plugins/nsfw.py

@@ -5,6 +5,7 @@ from PIL import Image, UnidentifiedImageError
 import torch
 import torch.nn.functional as F
 from pyrogram import Client, filters
+from pyrogram.enums import ChatMemberStatus
 from pyrogram.types import Message
 from transformers import AutoModelForImageClassification, ViTImageProcessor
 from DragMusic import app
@@ -35,48 +36,65 @@ async def enable_nsfw(chat_id: int):
 async def disable_nsfw(chat_id: int):
     await nsfw_chats_collection.delete_one({"chat_id": chat_id})
 
-# Admin check function
+# Admin check using official ChatMemberStatus
 async def is_admin(client: Client, chat_id: int, user_id: int) -> bool:
     try:
         member = await client.get_chat_member(chat_id, user_id)
-        return member.status in ["administrator", "creator"]
+        print(f"[DEBUG] is_admin → user_id={user_id}, status={member.status}")
+        return member.status in [ChatMemberStatus.ADMINISTRATOR, ChatMemberStatus.OWNER, "administrator", "creator"]
     except Exception as e:
-        print(f"[is_admin] Error: {e}")
+        print(f"[Admin Check Error] {e}")
         return False
 
-# Command to enable/disable NSFW scanning
+# Command: /nsfw enable / disable
 @app.on_message(filters.command("nsfw") & filters.group)
 async def toggle_nsfw_scan(client: Client, message: Message):
     if len(message.command) != 2:
-        return await message.reply("Usage: `/nsfw enable` or `/nsfw disable`", parse_mode="markdown")
+        return await message.reply("Usage:\n`/nsfw enable`\n`/nsfw disable`", parse_mode="markdown")
 
-    if not await is_admin(client, message.chat.id, message.from_user.id):
-        return await message.reply("Only group admins and the group owner can use this command.")
+    user_is_admin = await is_admin(client, message.chat.id, message.from_user.id)
+    if not user_is_admin:
+        return await message.reply("❌ Only group admins or the group owner can use this command.")
 
     arg = message.command[1].lower()
     if arg == "enable":
         await enable_nsfw(message.chat.id)
-        await message.reply("✅ Enabled NSFW scanning for this group.")
+        await message.reply("✅ NSFW scanning has been *enabled* for this group.", parse_mode="markdown")
     elif arg == "disable":
         await disable_nsfw(message.chat.id)
-        await message.reply("❌ Disabled NSFW scanning for this group.")
+        await message.reply("🚫 NSFW scanning has been *disabled* for this group.", parse_mode="markdown")
     else:
-        await message.reply("Invalid argument. Use `/nsfw enable` or `/nsfw disable`.", parse_mode="markdown")
+        await message.reply("⚠️ Invalid option. Use `/nsfw enable` or `/nsfw disable`", parse_mode="markdown")
 
-# Command to manually scan a media reply
+# Debug command: /whoami
+@app.on_message(filters.command("whoami") & filters.group)
+async def whoami(client: Client, message: Message):
+    try:
+        member = await client.get_chat_member(message.chat.id, message.from_user.id)
+        await message.reply_text(
+            f"👤 Your user ID: `{message.from_user.id}`\n"
+            f"📢 Chat ID: `{message.chat.id}`\n"
+            f"🛡️ Your status: `{member.status}`",
+            quote=True,
+            parse_mode="markdown"
+        )
+    except Exception as e:
+        await message.reply_text(f"❌ Failed to fetch status: `{e}`", parse_mode="markdown")
+
+# Command: /scan (manual)
 @app.on_message(filters.command("scan") & filters.reply)
 async def scan_command(_: Client, message: Message):
     if not message.reply_to_message:
         return await message.reply("Reply to an image, video, or sticker to scan.")
     await scan_media(message.reply_to_message, message)
 
-# Auto-scan media when enabled
+# Auto scan media
 @app.on_message(filters.group & (filters.photo | filters.video | filters.sticker))
 async def auto_nsfw_scan(client: Client, message: Message):
     if await is_nsfw_enabled(message.chat.id):
         await scan_media(message, message)
 
-# Main media scanner
+# NSFW scan handler
 async def scan_media(media_msg: Message, reply_msg: Message):
     media = media_msg.photo or media_msg.video or media_msg.sticker
     path = None
@@ -99,7 +117,6 @@ async def scan_media(media_msg: Message, reply_msg: Message):
         else:
             return await reply_msg.reply("Unable to download media.")
 
-        # Convert to image if needed
         if path.endswith((".webm", ".mp4", ".webp")):
             image_path = path + ".png"
             subprocess.run(
@@ -121,7 +138,6 @@ async def scan_media(media_msg: Message, reply_msg: Message):
             confidence = round(probs[0][1].item() * 100, 2)
             label = "NSFW" if confidence > 50 else "SFW"
 
-        # Handle results
         if label == "NSFW":
             try:
                 await media_msg.delete()