Update main.py

main.py

@@ -70,31 +70,31 @@ def is_tiktok_url(url):
     match = re.search(pattern, url)
     return bool(match)
 
-@client.on_message(filters.text & (filters.private | filters.group))
+@client.on_message(filters.command(["tt"]) & (filters.private | filters.group))
 async def tiktok_downloader(client: Client, message: Message):
-    if message.text:
-        query_url = message.text
-        if not is_tiktok_url(query_url):
-            return await message.reply_text("Invalid link")
-        callback_data = generate_callback_data(message.from_user.id, query_url)
-        keyboard = InlineKeyboardMarkup(
+    query_url = message.text.split(" ", 1)[1] if len(message.command) > 1 else None:
+    if not query_url:
+        return await message.reply_text("?")
+    if not is_tiktok_url(query_url):
+        return await message.reply_text("Invalid link")
+    callback_data = generate_callback_data(message.from_user.id, query_url)
+    keyboard = InlineKeyboardMarkup(
+        [
             [
-                [
-                    InlineKeyboardButton(
-                        text="Audio Download",
-                        callback_data=callback_data
-                    )
-                ]
+                InlineKeyboardButton(
+                    text="Audio Download",
+                    callback_data=callback_data
+                )
             ]
-        )
-        try:
-            dll = await message.reply_text("Processing....")
-            await message.delete()
-            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()
-            await message.reply_text(f"Error: {str(e)}")
+        ]
+    )
+    try:
+        dll = await message.reply_text("Processing....")
+        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()
+        await message.reply_text(f"Error: {str(e)}")
 
 client.run()