Update main.py

main.py

@@ -1,4 +1,5 @@
 import logging
+import re
 from pyrogram import Client, filters
 from pyrogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
 from pyrogram.types import *
@@ -63,11 +64,16 @@ async def callback_button(client: Client, cb: CallbackQuery):
     except Exception as e:
         await cb.answer(f"Error: {str(e)}", show_alert=True)
 
+def is_tiktok_url(url):
+    pattern = r"(https?)://(vt|www)\.tiktok\.com/(\w+)"
+    match = re.search(pattern, name)
+    return bool(match)
+
 @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/"):
+        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)
         keyboard = InlineKeyboardMarkup(
@@ -83,7 +89,7 @@ async def tiktok_downloader(client: Client, message: Message):
         try:
             dll = await message.reply_text("Processing....")
             await message.delete()
-            response = await Tiktok.download(tt, query)
+            response = await Tiktok.download(tt, query_url)
             await message.reply_video(response[0], reply_markup=keyboard)
             await dll.delete()
         except Exception as e: