Update Akeno/plugins/antinsfw.py

Akeno/plugins/antinsfw.py

@@ -78,18 +78,13 @@ async def antinsfw_filter(client: Client, message: Message):
         return
     if not user or user.id == 777000:
         return
-    file_id = message.photo.file_id
-    media = await client.download_media(file_id)
-    try:
+    if message.photo:
+        file_id = message.photo.file_id
+        media = await client.download_media(file_id)
         if check_anti_nsfw(media):
-            me = await client.get_me()
-            if await can_delete(client, chat.id, me.id):
+            if await can_delete(client, chat.id, client.me.id):
                 await message.delete()
-    except Exception as e:
-        LOGS.info(f"Error processing NSFW check: {e}")
-    finally:
-        if os.path.exists(media):
-            os.remove(media)
+                os.remove(media)
 
 module = modules_help.add_module("antinsfw", __file__)
 module.add_command("antinsfw", "to anti nsfw auto delete messages")