Upload 4 files

app.py

@@ -959,6 +959,45 @@ def sync_ig_cookies_to_db() -> None:
         logger.warning(f"[IG] could not sync cookies to DB: {e}")
 
 
+def _parse_id_csv(raw: str) -> list:
+    """Parse a CSV of Telegram user ids into a list of ints (ignores junk)."""
+    ids = []
+    for tok in (raw or "").replace(" ", "").split(","):
+        if not tok:
+            continue
+        try:
+            ids.append(int(tok))
+        except ValueError:
+            logger.warning(f"[bot] ignoring non-numeric authorized id '{tok}'")
+    return ids
+
+
+def sync_telegram_config_to_db() -> None:
+    """Store the HF env Telegram settings into MongoDB (config._id='telegram'):
+      • BOT_TOKEN  → bot_token          (the bot's login token)
+      • USER_ID    → authorized_user_ids (only this user may send videos to the bot)
+    The Koyeb bot logs in with the token and writes back its own bot_id/bot_username,
+    so HF learns where to send. Only provided fields are written (no clobbering)."""
+    token = os.getenv("BOT_TOKEN", "").strip()
+    # USER_ID is the new HF variable; AUTHORIZED_USER_IDS kept as an alias.
+    ids = _parse_id_csv(os.getenv("USER_ID", "") or os.getenv("AUTHORIZED_USER_IDS", ""))
+
+    fields = {}
+    if token:
+        fields["bot_token"] = token
+    if ids:
+        fields["authorized_user_ids"] = ids
+
+    if not fields:
+        logger.info("[bot] no BOT_TOKEN/USER_ID on HF — leaving MongoDB telegram config as-is")
+        return
+    try:
+        config.update_one({"_id": "telegram"}, {"$set": fields}, upsert=True)
+        logger.info(f"[bot] ✅ stored Telegram config → MongoDB ({', '.join(fields.keys())})")
+    except Exception as e:
+        logger.warning(f"[bot] could not store Telegram config in DB: {e}")
+
+
 # ───────────────────────────── SEND EDITED VIDEO → INSTAGRAM TELEGRAM BOT
 # Instagram blocks this (HF) server's IP, so the actual Reel upload is done by the
 # companion Telegram bot in koyeb/. This app just DELIVERS the finished video to
@@ -971,10 +1010,12 @@ def sync_ig_cookies_to_db() -> None:
 # IG_BOT_TARGET env overrides it.
 
 def get_bot_target():
-    """Resolve the Telegram bot the edited video is sent to (MongoDB, env fallback)."""
+    """Resolve the Telegram bot the edited video is sent to. The Koyeb bot writes
+    its own bot_username/bot_id into MongoDB after it logs in, so HF reads it from
+    there. IG_BOT_TARGET env is a manual override (e.g. before Koyeb's first run)."""
     override = os.getenv("IG_BOT_TARGET", "").strip()
     if override:
-        return override
+        return override.lstrip("@")
     try:
         doc = config.find_one({"_id": "telegram"}) or {}
     except Exception as e:
@@ -1208,9 +1249,10 @@ def auto_loop():
         "makes YouTube use the higher-quality VP9 codec to minimise it."
     )
 
-    # Instagram is uploaded by the companion Telegram bot (koyeb/). Mirror the
-    # HF env cookies into MongoDB so the bot always has the latest, and log the
-    # bot we deliver videos to.
+    # Instagram is uploaded by the companion Telegram bot (koyeb/). Mirror ALL
+    # HF env config into MongoDB (cookies + bot token/username/authorized ids) so
+    # the bot is set up entirely from Hugging Face — Koyeb only needs MONGO_URI.
+    sync_telegram_config_to_db()
     sync_ig_cookies_to_db()
     logger.info(f"[ℹ️] Instagram via Telegram bot: {get_bot_target() or '(not configured)'}")