Create main.py

main.py

@@ -0,0 +1,93 @@
+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
+import hashlib
+
+logging.getLogger("pyrogram").setLevel(logging.WARNING)
+logging.basicConfig(level=logging.INFO)
+
+WELCOME_TEXT = """
+Halo {}
+Saya adalah bot untuk mengunduh video tiktok di telegram.
+
+Saya dapat mengunduh video dengan tanda air atau tanpa tanda air dan mengunduh audio dari url. Kirimkan saja saya url tiktok.
+"""
+
+client = Client(
+    "TTK-BOT",
+    api_id=API_ID,
+    api_hash=API_HASH,
+    bot_token=BOT_TOKEN
+)
+
+link_storage = {}
+
+def generate_callback_data(user_id, query):
+    identifier = hashlib.md5(query.encode()).hexdigest()
+    callback_data = f"audiodownload_{user_id}_{identifier}"
+    link_storage[callback_data] = query
+    return callback_data
+
+@client.on_message(filters.command("start") & filters.private)
+async def welcome_start(client: Client, message: Message):
+    keyboard = InlineKeyboardMarkup(
+        [
+            [
+                InlineKeyboardButton(
+                    text="📢 Saluran Bot",
+                    url="https://t.me/RendyProjects"
+                )
+            ]
+        ]
+    )
+    await message.reply_text(
+        WELCOME_TEXT.format(message.from_user.first_name),
+        reply_markup=keyboard
+    )
+
+@client.on_callback_query(filters.regex("^audiodownload_"))
+async def callback_button(client: Client, cb: CallbackQuery):
+    try:
+        data = cb.data
+        user_id = cb.from_user.id
+        query = link_storage.get(data)
+        if query:
+            response = await Tiktok.download(tt, query)
+            await client.send_audio(user_id, response[1])
+            await cb.answer("Audio sent successfully!")
+        else:
+            await cb.answer("Invalid or expired link.", show_alert=True)
+    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
+        if not query.startswith("https://vt.tiktok.com/") and not query.startswith("https://www.tiktok.com/"):
+            return await message.reply_text("Invalid link")
+        callback_data = generate_callback_data(message.from_user.id, query)
+        keyboard = InlineKeyboardMarkup(
+            [
+                [
+                    InlineKeyboardButton(
+                        text="Audio Download",
+                        callback_data=callback_data
+                    )
+                ]
+            ]
+        )
+        try:
+            dll = await message.reply_text("Processing....")
+            await message.delete()
+            response = await Tiktok.download(tt, query)
+            await message.reply_video(response[0], reply_markup=keyboard)
+            await dll.delete()
+        except Exception as e:
+            await dll.delete()
+            await message.reply_text(f"Error: {str(e)}")
+
+client.run()