Update telegram_bot.py

telegram_bot.py

@@ -5,13 +5,17 @@ from dotenv import load_dotenv
 from telegram import Update, ReplyKeyboardMarkup, ReplyKeyboardRemove
 import datetime
 from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes, ConversationHandler
+# NEW IMPORT: Necessary to explicitly configure the request object
+from telegram.request import HTTPXRequest 
 import logging
 
-# FIX 1: Using the langchain_community package imports for stability
-from langchain_community.embeddings import HuggingFaceEmbeddings
-from langchain_community.vectorstores import Chroma
+# Using the dedicated packages as recommended by LangChain warnings
+from langchain_huggingface import HuggingFaceEmbeddings
+from langchain_community.vectorstores import Chroma # Chroma is still in community
 
 # Load environment variables
+# NOTE: load_dotenv() is mainly for local development. In deployment, 
+# environment variables are set by the hosting platform (Hugging Face Secrets).
 load_dotenv()
 
 # Enable logging
@@ -30,9 +34,14 @@ class TelegramHomeopathyBot:
         self.logs_dir = "UserChatLogs"
         os.makedirs(self.logs_dir, exist_ok=True)
         
+        # Switched to a standard, reliable embedding model. 
+        TINY_EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2" 
+
         # Initialize embeddings and vector store
         self.embeddings = HuggingFaceEmbeddings(
-            model_name="sentence-transformers/all-MiniLM-L6-v2"
+            model_name=TINY_EMBEDDING_MODEL,
+            # Explicitly set device to 'cpu' for memory constrained environments
+            model_kwargs={'device': 'cpu'} 
         )
         self.vector_store = Chroma(
             persist_directory="./vector_db",
@@ -43,7 +52,6 @@ class TelegramHomeopathyBot:
         # Store user sessions: includes chat_history
         self.user_sessions = {}
     
-    # FIX 4: Removed duplicate log_chat definition
     def log_chat(self, user_id: int, username: str, message: str, is_bot: bool = False):
         """Log user/bot messages to individual files"""
         try:
@@ -62,7 +70,7 @@ class TelegramHomeopathyBot:
             self.user_sessions[user_id] = {
                 'chat_history': [],
                 'consultation_count': 0,
-                'last_query': None  # Track query context
+                'last_query': None
             }
         return self.user_sessions[user_id]
     
@@ -71,11 +79,9 @@ class TelegramHomeopathyBot:
         Retrieve relevant context from the vector database using the latest query 
         and the accumulated user history for better RAG results.
         """
-        # Create a composite query string from previous user inputs and the latest query
         user_history = [msg['content'] for msg in history if msg['role'] == 'user']
         history_summary = " ".join(user_history)
         
-        # Combine the user's history and current message for a comprehensive retrieval query
         composite_query = f"Patient's case summary: {history_summary} {query}" if history_summary else query
         
         docs = self.retriever.invoke(composite_query) 
@@ -114,14 +120,12 @@ class TelegramHomeopathyBot:
         ]
         
         data = {
-            # FIX 3: Changed to the working model that resolved the 404 error in the console bot
             "model": "meituan-longcat/LongCat-Flash-Chat-FP8", 
             "messages": messages,
             "temperature": 0.2,
             "max_tokens": 500
         }
         
-        # FIX 2: Corrected API URL to the one that successfully connected
         api_url = "https://llm.chutes.ai/v1/chat/completions"
 
         try:
@@ -139,7 +143,7 @@ class TelegramHomeopathyBot:
                 error_message = error_data.get("error", {}).get("message", "No detailed error message.")
                 
                 if response.status_code == 401:
-                    logger.error("🚨 CRITICAL API ERROR (401 Unauthorized) 🚨: The CHUTEAI_API_KEY is likely invalid or missing. Please check your .env file.")
+                    logger.error("🚨 CRITICAL API ERROR (401 Unauthorized) 🚨: The CHUTEAI_API_KEY is likely invalid or missing. Please check your setup.")
                 
                 logger.error(f"Chute.ai Error {response.status_code}: {error_message}")
                 return f"❌ I'm having technical difficulties. Please try again later. (Error: {response.status_code})"
@@ -149,11 +153,20 @@ class TelegramHomeopathyBot:
             return f"❌ Connection error. Please try again. Error: {str(e)}"
 
 # Create bot instance
-homeopathy_bot = TelegramHomeopathyBot()
+# We wrap the initialization in a try/except block to catch memory-related failures early
+try:
+    homeopathy_bot = TelegramHomeopathyBot()
+except Exception as e:
+    logger.error(f"Failed to initialize TelegramHomeopathyBot (likely due to memory): {e}")
+    homeopathy_bot = None
 
-# Telegram Bot Handlers
+# Telegram Bot Handlers (start, handle_symptoms, etc. functions remain the same)
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
     """Send welcome message when the command /start is issued."""
+    if not homeopathy_bot:
+        await update.message.reply_text("❌ Initialization Error: The AI service failed to start due to resource constraints. Please check the logs.")
+        return ConversationHandler.END
+        
     user = update.effective_user
     welcome_text = f"""
 👋 Hello *{user.first_name}*! I'm your AI Homeopathy Doctor 🤖
@@ -187,6 +200,10 @@ Type your symptoms below to begin...
 
 async def handle_symptoms(update: Update, context: ContextTypes.DEFAULT_TYPE):
     """Handle user's symptom description and continue the diagnosis."""
+    if not homeopathy_bot:
+        await update.message.reply_text("❌ Service is unavailable due to an earlier initialization failure. Please try later.")
+        return ConversationHandler.END
+
     user_id = update.effective_user.id
     user_input = update.message.text
     
@@ -261,8 +278,6 @@ async def handle_quick_actions(update: Update, context: ContextTypes.DEFAULT_TYP
     user_input = update.message.text
     
     if user_input in ["🔍 Describe more symptoms or answer questions", "Reset please"]:
-        # If the bot has prescribed a remedy, the user might be ready to start fresh, 
-        # so this is a good opportunity to offer the reset.
         if len(homeopathy_bot.get_user_session(update.effective_user.id)['chat_history']) >= 4: # If 2 turns have passed
              await update.message.reply_text("Please provide any additional details or clarify the Doctor's previous questions...")
         else:
@@ -334,34 +349,48 @@ async def error_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
 
 def check_dependencies():
     """Checks for required environment variables and the vector database."""
-    if not os.getenv("TELEGRAM_BOT_TOKEN"):
-        print("❌ TELEGRAM_BOT_TOKEN not found in .env file. Please set it to run the bot.")
-        return False
     
     if not os.path.exists("./vector_db"):
-        print("❌ Vector database not found. Please run 'python ingest_book.py' first.")
+        print("❌ Vector database not found. Please ensure 'vector_db' folder is uploaded.")
         return False
     
-    # FIX 5: Changed dependency check to look for CHUTEAI_API_KEY
     if not os.getenv("CHUTEAI_API_KEY"):
-        print("❌ CHUTEAI_API_KEY not found in .env file. Please set it to run the bot.")
-        return False
+        print("❌ CHUTEAI_API_KEY not found in environment. AI queries will fail.")
+        return True # Allow start, but AI will fail
     
     return True
 
 def main():
     """Start the bot."""
     
+    # 🚨 CRITICAL CHECK: Ensure the token is available
+    TELEGRAM_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+    if not TELEGRAM_TOKEN:
+        print("FATAL ERROR: TELEGRAM_BOT_TOKEN is missing from environment secrets. Cannot initialize the Telegram client.")
+        return
+    
+    # NEW: Sanity check to confirm the token is read and has a valid length
+    if len(TELEGRAM_TOKEN) < 40:
+        print(f"WARNING: TELEGRAM_BOT_TOKEN has suspicious length ({len(TELEGRAM_TOKEN)}). Double-check the secret value.")
+    
     if not check_dependencies():
         return
     
     print("🚀 Starting Telegram Homeopathy Bot...")
     
-    # Get Telegram bot token from environment variable
-    TELEGRAM_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+    # NEW: Explicitly configure the request object to use the correct base URL
+    # This often bypasses environmental DNS resolution failures.
+    custom_request = HTTPXRequest(
+        base_url='https://api.telegram.org/bot'
+    )
     
     # Create Application
-    application = Application.builder().token(TELEGRAM_TOKEN).build()
+    application = (
+        Application.builder()
+        .token(TELEGRAM_TOKEN)
+        .request(custom_request) # Inject the custom request object
+        .build()
+    )
     
     # Add conversation handler
     conv_handler = ConversationHandler(