Add MAIN_BOT_SESSION config to support mainbot.session file

README.md

@@ -59,6 +59,7 @@ Welcome to TGDrive! This web application replicates Google Drive's functionaliti
 | `DATABASE_BACKUP_TIME` | integer (in seconds) | 60                                         | Interval in seconds for database backups to the storage channel                                             |
 | `MAX_FILE_SIZE`        | float (in GBs)       | 1.98 (3.98 if `STRING_SESSIONS` are added) | Maximum file size (in GBs) allowed for uploading to Telegram                                                |
 | `WEBSITE_URL`          | string               | None                                       | Website URL (with https/http) to auto-ping to keep the website active                                       |
+| `MAIN_BOT_SESSION`     | string               | mainbot.session                            | Main Bot Session file name for [TG Drive's Bot Mode](#tg-drives-bot-mode). Place session file in cache/ directory. |
 | `MAIN_BOT_TOKEN`       | string               | None                                       | Your Main Bot Token to use [TG Drive's Bot Mode](#tg-drives-bot-mode)                                       |
 | `TELEGRAM_ADMIN_IDS`   | string               | None                                       | List of Telegram User IDs of admins who can access the [bot mode](#tg-drives-bot-mode), separated by commas |
 

SETUP_SUMMARY.txt

@@ -15,11 +15,14 @@ NOTE: You need to upload cache/bot.session file to your Hugging Face Space
 
 OPTIONAL - For Bot Mode (users can upload files via Telegram bot):
 -------------------------------------------------------------------
+MAIN_BOT_SESSION=mainbot.session  # Default: mainbot.session
 MAIN_BOT_TOKEN=your_main_bot_token
 TELEGRAM_ADMIN_IDS=123456789,987654321
 
-NOTE: MAIN_BOT_TOKEN will automatically create main_bot.session file
-      You DON'T need to provide a session file for this - just the token!
+NOTE: MAIN_BOT_SESSION defaults to "mainbot.session"
+      - If session file exists in cache/, it will be used (faster connection)
+      - If session file doesn't exist, MAIN_BOT_TOKEN will create it
+      - You can upload cache/mainbot.session file to your Hugging Face Space
 
 OPTIONAL - For Premium (files > 2GB):
 -------------------------------------

config.py

@@ -79,7 +79,12 @@ WEBSITE_URL = os.getenv("WEBSITE_URL", None)
 
 # For Using TG Drive's Bot Mode
 
+# Main Bot Session file name for TG Drive's Bot Mode (e.g., "mainbot.session")
+# Session file should be in cache directory. Pyrogram will use it if it exists.
+MAIN_BOT_SESSION = os.getenv("MAIN_BOT_SESSION", "mainbot.session").strip()
+
 # Main Bot Token for TG Drive's Bot Mode
+# Used with session file (if session file doesn't exist, token creates it)
 MAIN_BOT_TOKEN = os.getenv("MAIN_BOT_TOKEN", "")
 if MAIN_BOT_TOKEN.strip() == "":
     MAIN_BOT_TOKEN = None

sample.env

@@ -13,8 +13,14 @@ DATABASE_BACKUP_MSG_ID=123
 # Optional: String sessions for Premium accounts (files > 2GB)
 # STRING_SESSIONS=1BVtsOMwBu5...your_session_string_here...
 
+# Optional: For TG Drive's Bot Mode
+# MAIN_BOT_SESSION=mainbot.session  # Session file name (default: mainbot.session)
+# MAIN_BOT_TOKEN=your_main_bot_token  # Bot token (required for bot mode)
+# TELEGRAM_ADMIN_IDS=123456789,987654321  # Admin user IDs
+
 # Note: 
 # - BOT_TOKENS is required (Pyrogram needs token for bot clients)
 # - BOT_SESSIONS is optional - if provided, Pyrogram will use those session file names
 # - If session file exists, it will be used (faster connection)
-# - If session file doesn't exist, Pyrogram will create it
+# - If session file doesn't exist, Pyrogram will create it
+# - MAIN_BOT_SESSION defaults to "mainbot.session" - place it in cache/ directory

utils/bot_mode.py

@@ -28,7 +28,7 @@ session_cache_path = Path(f"./cache")
 session_cache_path.parent.mkdir(parents=True, exist_ok=True)
 
 main_bot = Client(
-    name="main_bot",
+    name=config.MAIN_BOT_SESSION.replace(".session", ""),  # Remove .session extension for name
     api_id=config.API_ID,
     api_hash=config.API_HASH,
     bot_token=config.MAIN_BOT_TOKEN,