Update main.py

main.py

@@ -1,6 +1,7 @@
 import logging
 from pyrogram import Client, filters
 from pyrogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
+from pyrogram.types import *
 from RyuzakiLib import Tiktok
 from config import TIKTOK_WEB as tt, API_ID, API_HASH, BOT_TOKEN
 
@@ -38,18 +39,40 @@ async def welcome_start(client: Client, message: Message):
         reply_markup=keyboard
     )
 
+@client.on_callback_query(filters.regex("^audiodownload_"))
+async def callback_button(client: Client, cb: CallbackQuery):
+    try:
+        data = cb.data.split("_")
+        user_id = int(data[1].split("|")[0])
+        link = cb.data.split("|")[1]
+        response = Tiktok.download(tt, link)
+        await client.send_audio(user_id, response[1])
+        await cb.answer("Audio sent successfully!")
+    except Exception as e:
+        await cb.answer(f"Error: {str(e)}", show_alert=True)
+
 @client.on_message(filters.text & filters.private)
 async def tiktok_downloader(client: Client, message: Message):
     if message.text:
+        query = message.text
+        keyboard = InlineKeyboardMarkup(
+            [
+                [
+                    InlineKeyboardButton(
+                        text="Audio Download",
+                        callback_data=f"audiodownload_{message.from_user.id}|{query}"
+                    )
+                ]
+            ]
+        )
         try:
-            query = message.text
             dll = await message.reply_text("Processing....")
             await message.delete()
             response = Tiktok.download(tt, query)
-            await message.reply_video(response[0])
+            await message.reply_video(response[0], reply_markup=keyboard)
             await dll.delete()
         except Exception as e:
-            await message.reply_text(f"Error: {str(e)}")
             await dll.delete()
+            await message.reply_text(f"Error: {str(e)}")
 
-client.run()
+client.run()