Update bot/client.py

bot/client.py

@@ -5,21 +5,32 @@ from bot.config import Telegram
 
 logger = getLogger("bot")
 
-def create_client() -> Client:
-    base = dict(
-        name="app",
+def create_clients() -> tuple[Client, Client]:
+    base_args = dict(
         api_id=Telegram.API_ID,
         api_hash=Telegram.API_HASH,
         in_memory=True,
-        sleep_threshold=-1,
+        sleep_threshold=30,
         max_concurrent_transmissions=10,
     )
 
-    if Telegram.SESSION_STRING:
-        logger.info("Using SESSION_STRING auth")
-        return Client(session_string=Telegram.SESSION_STRING, **base)
-
+    # 1. Main Bot
     if not Telegram.BOT_TOKEN:
-        raise RuntimeError("Missing BOT_TOKEN and SESSION_STRING is empty.")
-    logger.info("Using BOT_TOKEN auth")
-    return Client(bot_token=Telegram.BOT_TOKEN, **base)
+        raise RuntimeError("BOT_TOKEN is required.")
+    
+    bot = Client("bot_session", bot_token=Telegram.BOT_TOKEN, **base_args)
+
+    # 2. User Client (Downloader)
+    user = None
+    if Telegram.USER_SESSION_STRING:
+        logger.info("✅ Downloader Client Configured")
+        user = Client(
+            "downloader_session",
+            session_string=Telegram.USER_SESSION_STRING,
+            no_updates=True,
+            **base_args
+        )
+    else:
+        logger.warning("⚠️ No USER_SESSION_STRING. Restricted links will fail.")
+
+    return bot, user