Update main.py

main.py

@@ -3,14 +3,14 @@ import re
 from pyrogram import Client, filters
 from pyrogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
 from pyrogram.types import *
-from akenoai import AkenoXToJs
+from RyuzakiLib import Tiktok
 from config import API_ID, API_HASH, BOT_TOKEN
 import hashlib
 
-logger = logging.getLogger(__name__)
-LOGS = logging.getLogger("[akeno]")
-logger.setLevel(logging.DEBUG)
+logging.getLogger("pyrogram").setLevel(logging.WARNING)
+logging.basicConfig(level=logging.INFO)
 
+TIKTOK_WEB = "https://www.tikwm.com"
 WELCOME_TEXT = """
 Halo {}
 Saya adalah bot untuk mengunduh video tiktok di telegram.
@@ -26,8 +26,6 @@ client = Client(
 )
 
 link_storage = {}
-DEMO_API_KEY = "akeno_OSrXhljIomunACd5JY18jFIeIuuB6Pdx"
-
 
 def generate_callback_data(user_id, query):
     identifier = hashlib.md5(query.encode()).hexdigest()
@@ -59,15 +57,8 @@ async def callback_button(client: Client, cb: CallbackQuery):
         user_id = cb.from_user.id
         query = link_storage.get(data)
         if query:
-            response = await AkenoXToJs.randydev(
-                "dl/tiktok",
-                api_key=DEMO_API_KEY,
-                custom_dev=True,
-                url=query
-            )
-            LOGS.info(response)
-            music_tiktok = response.results.data.music_info.get("play")
-            await client.send_audio(user_id, music_tiktok)
+            response = await Tiktok.download(TIKTOK_WEB, 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)
@@ -99,17 +90,8 @@ async def tiktok_downloader(client: Client, message: Message):
         try:
             dll = await message.reply_text("Processing....")
             await message.delete()
-            response = await AkenoXToJs.randydev(
-                "dl/tiktok",
-                api_key=DEMO_API_KEY,
-                custom_dev=True,
-                url=query_url
-            )
-            LOGS.info(response)
-            await message.reply_video(
-                response.results.data.get("play"),
-                reply_markup=keyboard
-            )
+            response = await Tiktok.download(TIKTOK_WEB, query_url)
+            await message.reply_video(response[0], reply_markup=keyboard)
             await dll.delete()
         except Exception as e:
             await dll.delete()