Update bot/client.py

bot/client.py

@@ -5,7 +5,7 @@ from bot.config import Telegram
 
 logger = getLogger("bot")
 
-def create_clients() -> tuple[Client, Client]:
+def create_clients() -> tuple[Client, Client | None]:
     base_args = dict(
         api_id=Telegram.API_ID,
         api_hash=Telegram.API_HASH,
@@ -14,23 +14,25 @@ def create_clients() -> tuple[Client, Client]:
         max_concurrent_transmissions=10,
     )
 
-    # 1. Main Bot
-    if not Telegram.BOT_TOKEN:
-        raise RuntimeError("BOT_TOKEN is required.")
-    
-    bot = Client("bot_session", bot_token=Telegram.BOT_TOKEN, **base_args)
+    if Telegram.BOT_TOKEN:
+        logger.info("✅ Using BOT_TOKEN auth")
+        bot = Client("bot_session", bot_token=Telegram.BOT_TOKEN, **base_args)
+    elif Telegram.SESSION_STRING:
+        logger.warning("⚠️ BOT_TOKEN missing; running legacy SESSION_STRING mode (not a bot).")
+        bot = Client("user_session", session_string=Telegram.SESSION_STRING, **base_args)
+    else:
+        raise RuntimeError("Missing BOT_TOKEN (preferred) and SESSION_STRING is empty.")
 
-    # 2. User Client (Downloader)
     user = None
-    if Telegram.USER_SESSION_STRING:
-        logger.info("✅ Downloader Client Configured")
+    if Telegram.BOT_TOKEN and Telegram.USER_SESSION_STRING:
+        logger.info("✅ Worker (Downloader) Client Configured")
         user = Client(
             "downloader_session",
             session_string=Telegram.USER_SESSION_STRING,
             no_updates=True,
-            **base_args
+            **base_args,
         )
-    else:
-        logger.warning("⚠️ No USER_SESSION_STRING. Restricted links will fail.")
+    elif Telegram.BOT_TOKEN:
+        logger.warning("⚠️ No USER_SESSION_STRING. Link-archive disabled.")
 
     return bot, user