Update main.py

main.py

@@ -1,11 +1,16 @@
 import logging
-from pyrogram import Client, filters
+from pyrogram import Client, filters, idle
 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
+from config import TIKTOK_WEB as tt, API_ID, API_HASH, BOT_TOKEN, HUGGING_TOKEN
 import hashlib
 
+try:
+    from aiohttp import ClientSession as aiohttp_client
+except ImportError:
+    aiohttp_client = None
+
 logging.getLogger("pyrogram").setLevel(logging.WARNING)
 logging.basicConfig(level=logging.INFO)
 
@@ -31,6 +36,34 @@ def generate_callback_data(user_id, query):
     link_storage[callback_data] = query
     return callback_data
 
+async def async_searcher(
+    url: str,
+    post: bool = False,
+    head: bool = False,
+    headers: dict = None,
+    evaluate=None,
+    object: bool = False,
+    re_json: bool = False,
+    re_content: bool = False,
+    *args,
+    **kwargs,
+):
+    if aiohttp_client:
+        async with aiohttp_client(headers=headers) as client:
+            method = client.head if head else (client.post if post else client.get)
+            data = await method(url, *args, **kwargs)
+            if evaluate:
+                return await evaluate(data)
+            if re_json:
+                return await data.json()
+            if re_content:
+                return await data.read()
+            if head or object:
+                return data
+            return await data.text()
+    else:
+        raise DependencyMissingError("install 'aiohttp' to use this.")
+
 @client.on_message(filters.command("start") & filters.private)
 async def welcome_start(client: Client, message: Message):
     keyboard = InlineKeyboardMarkup(
@@ -88,4 +121,14 @@ async def tiktok_downloader(client: Client, message: Message):
             await dll.delete()
             await message.reply_text(f"Error: {str(e)}")
 
+async def setup():
+    try:
+        await client.start()
+        logger.info("Bot started successfully")
+        await start_periodic_task()
+        # Run the client
+        await idle()
+    except Exception as e:
+        logger.error(f"Error in setup: {e}")
+
 client.run()