Update code2.py

code2.py

@@ -5,8 +5,10 @@ import re
 from datetime import datetime, timedelta
 
 from telethon import TelegramClient, events
+from telethon.tl.custom import Button
 from telethon.tl.types import ChatAdminRights, ChatBannedRights, ChannelParticipantsAdmins
-from telethon.tl.custom.buttons import InlineKeyboardButton
+
+
 
 API_ID = os.getenv("API_ID")
 API_HASH = os.getenv("API_HASH")
@@ -32,7 +34,6 @@ async def start(event):
 async def help(event):
     await event.respond('I can blacklist certain words. Just type them in a chat I am in!')
 
-
 @client.on(events.NewMessage(incoming=True))
 async def check_blacklist(event):
     if any(
@@ -68,8 +69,8 @@ async def check_blacklist(event):
                 logger.error("Failed to mute the user. Skipping action.")
 
             buttons = [
-                [InlineKeyboardButton("Unmute", callback_data="unmute")],
-                [InlineKeyboardButton("Ban", callback_data="ban")],
+                [Button.inline("Unmute", b"unmute")],
+                [Button.inline("Ban", b"ban")],
             ]
             await client.send_message(
                 entity=event.chat_id,
@@ -109,4 +110,5 @@ async def handle_button(event):
     except Exception as e:
         logger.error(f"An error occurred: {e}")
 
-client.run_until_disconnected()
+client.run_until_disconnected()
+