Update bot/config.py

bot/config.py

@@ -3,29 +3,31 @@ import os
 
 def _get_int(name: str, default: int) -> int:
     try:
-        return int(os.environ.get(name, str(default)))
+        return int(str(os.environ.get(name, default)).strip())
     except Exception:
         return default
 
 def _get_str(name: str, default: str = "") -> str:
-    return str(os.environ.get(name, default))
+    return str(os.environ.get(name, default)).strip()
 
 class Telegram:
-    # REQUIRED
-    API_ID: int = _get_int("TELEGRAM_API_ID", 12345)
-    API_HASH: str = _get_str("TELEGRAM_API_HASH", "YOUR_API_HASH")
+    API_ID: int = _get_int("API_ID", 0)
+    API_HASH: str = _get_str("API_HASH", "")
+    OWNER_ID: int = _get_int("OWNER_ID", 0)
 
-    # Use either SESSION_STRING (preferred on HF if bot token blocked) OR BOT_TOKEN
+    # HF: SESSION_STRING preferred (works even if BOT_TOKEN blocked)
     SESSION_STRING: str = _get_str("SESSION_STRING", "")
     BOT_TOKEN: str = _get_str("BOT_TOKEN", "")
 
-    OWNER_ID: int = _get_int("OWNER_ID", 0)  # put your tg id
-    BOT_USERNAME: str = _get_str("TELEGRAM_BOT_USERNAME", "BotFather")
+class Workers:
+    WORKER1_URL: str = _get_str("WORKER1_URL", "").rstrip("/")
+    WORKER2_URL: str = _get_str("WORKER2_URL", "").rstrip("/")
+    BOT_BACKEND_KEY: str = _get_str("BOT_BACKEND_KEY", "")
+    HF_API_KEY: str = _get_str("HF_API_KEY", "")
 
 class Server:
     BIND_ADDRESS: str = _get_str("BIND_ADDRESS", "0.0.0.0")
     PORT: int = _get_int("PORT", 7860)
-    BASE_URL: str = _get_str("BASE_URL", "http://127.0.0.1:7860")
 
 LOGGER_CONFIG_JSON = {
     "version": 1,
@@ -36,13 +38,12 @@ LOGGER_CONFIG_JSON = {
             "datefmt": "%d/%m/%Y %H:%M:%S",
         }
     },
-    "handlers": {
-        "stream": {"class": "logging.StreamHandler", "formatter": "default"}
-    },
+    "handlers": {"stream": {"class": "logging.StreamHandler", "formatter": "default"}},
     "loggers": {
         "uvicorn": {"level": "INFO", "handlers": ["stream"]},
         "uvicorn.error": {"level": "INFO", "handlers": ["stream"]},
         "bot": {"level": "INFO", "handlers": ["stream"]},
         "hydrogram": {"level": "INFO", "handlers": ["stream"]},
+        "httpx": {"level": "WARNING", "handlers": ["stream"]},
     },
 }