Update Akeno/plugins/emojiprem.py

Akeno/plugins/emojiprem.py

@@ -4,6 +4,7 @@ from pyrogram import Client as ren
 from pyrogram import Client, filters
 from pyrogram import *
 from pyrogram.types import *
+from pyrogram.enums import MessageEntityType
 from Akeno.utils.handler import *
 from config import *
 
@@ -39,22 +40,28 @@ async def set_emoji_status(client: Client, message: Message):
     if not text_get:
         await message.reply_text(f"Example ?setpemoji {custom_verifed}")
         return
-    get_custom_emoji_id = message.text_get.entities[0].custom_emoji_id.strip()
-    only_premium = await client.get_users("me")
-    if only_premium.is_premium:
+    if not message.entities:
+        await message.reply_text("No custom emoji found in the message.")
+        return
+    get_custom_emoji_id = None
+    for entity in message.entities:
+        if entity.type == MessageEntityType.CUSTOM_EMOJI:
+            get_custom_emoji_id = entity.custom_emoji_id
+            break
+    if not get_custom_emoji_id:
+        await message.reply_text("No custom emoji found in the message.")
+        return
+    if client.me.is_premium:
         try:
             await client.set_emoji_status(
-                EmojiStatus(
-                    custom_emoji_id=get_custom_emoji_id
-                )
+                EmojiStatus(custom_emoji_id=get_custom_emoji_id)
             )
-            await message.reply_text(f"Successfully profile change emoji status <emoji id={get_custom_emoji_id}>🗿</emoji>")
+            await message.reply_text(f"Successfully changed profile emoji status to <emoji id={get_custom_emoji_id}>🗿</emoji>")
         except Exception as e:
             await message.reply_text(str(e))
-            return
     else:
-        await message.reply_text("can't non-premium user")
-
+        await message.reply_text("Non-premium users cannot set custom emojis.")
+        
 @Akeno(
     ~filters.scheduled
     & filters.command(["stealpemoji"], CMD_HANDLER)