fix: adapt bot initialization for Hugging Face environment

bot_telegram.py

@@ -9,11 +9,15 @@ import asyncio
 from concurrent.futures import ThreadPoolExecutor
 import time
 import pandas as pd
+from dotenv import load_dotenv
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
-# Load environment variables from Hugging Face Secrets
+# Load environment variables
+load_dotenv()
+
+# Load environment variables from .env file
 BOT_SECRET_PASSWORD = os.getenv("BOT_SECRET_PASSWORD")
 BOT_TOKEN = os.getenv("BOT_TOKEN")
 BASE_URL = os.getenv("BASE_URL")
@@ -31,10 +35,16 @@ AWAITING_PASSWORD = set()
 
 logging.info("Bot starting... v0.1.0")  # Add version number
 
+# Add debug logging
+logging.info(f"BOT_TOKEN loaded: {'yes' if BOT_TOKEN else 'no'}")
+logging.info(f"BASE_URL loaded: {'yes' if BASE_URL else 'no'}")
+logging.info(f"HF_TOKEN loaded: {'yes' if HF_TOKEN else 'no'}")
+logging.info(f"BOT_SECRET_PASSWORD loaded: {'yes' if BOT_SECRET_PASSWORD else 'no'}")
+
 class TelegramBot:
     """A Telegram bot with password-based authentication."""
 
-    def __init__(self, bot_token, base_url):
+    def __init__(self, bot_token, base_url, application=None):
         """Initialize the bot with Telegram API token, API credentials, and authentication."""
         self.bot_token = bot_token
         self.base_url = base_url
@@ -47,7 +57,7 @@ class TelegramBot:
         self.ai_url = f"{self.base_url}/api/v1/questions/text"
         self.excel_url = f"{self.base_url}/api/v1/questions/excel"
 
-        # Executors (keep only one instance of each)
+        # Executors
         self.text_executor = ThreadPoolExecutor(max_workers=5, thread_name_prefix="text_worker")
         self.excel_executor = ThreadPoolExecutor(max_workers=5, thread_name_prefix="excel_worker")
         self.excel_semaphore = asyncio.Semaphore(10)
@@ -57,7 +67,10 @@ class TelegramBot:
         self.active_excel_files = {}
 
         # Start Telegram Bot
-        self.app = Application.builder().token(self.bot_token).build()
+        if application:
+            self.app = application
+        else:
+            self.app = Application.builder().token(self.bot_token).build()
         self.setup_handlers()
 
         # Authenticate with API
@@ -506,4 +519,19 @@ class TelegramBot:
         await self.app.shutdown()
 
 def init_bot():
-    return TelegramBot(bot_token=BOT_TOKEN,base_url=BASE_URL)
+    load_dotenv()
+    logging.info("Initializing bot...")
+    
+    try:
+        if os.getenv('SPACE_ID'):  # Check if running on Hugging Face
+            logging.info("Running on Hugging Face, using custom settings...")
+            application = Application.builder().token(BOT_TOKEN).base_url(
+                "https://api.telegram.org/bot"
+            ).get_updates_connection_pool_size(100).connection_pool_size(100).connect_timeout(30).read_timeout(30).write_timeout(30).pool_timeout(30).build()
+            return TelegramBot(bot_token=BOT_TOKEN, base_url=BASE_URL, application=application)
+        else:
+            logging.info("Running locally, using default settings...")
+            return TelegramBot(bot_token=BOT_TOKEN, base_url=BASE_URL)
+    except Exception as e:
+        logging.error(f"Error initializing bot: {str(e)}")
+        raise