Update bot_telegram.py

bot_telegram.py

@@ -76,12 +76,12 @@ class TelegramBot:
         user_id = update.message.from_user.id
         user_message = update.message.text.strip()
 
-        # Check if the user is awaiting password input
+        # If user is waiting to enter a password, validate it
         if user_id in AWAITING_PASSWORD:
             await self.check_password(update, context)
             return
 
-        # If the user is authenticated, process the AI request
+        # If user is authenticated, process AI request
         if user_id in AUTHENTICATED_USERS:
             await self.chat_with_ai(update, context)
         else:
@@ -97,8 +97,9 @@ class TelegramBot:
             return
 
         if user_message == SECRET_PASSWORD:
-            AUTHENTICATED_USERS.add(user_id)
-            AWAITING_PASSWORD.remove(user_id)
+            AUTHENTICATED_USERS.add(user_id)  # ✅ Save authentication status
+            AWAITING_PASSWORD.discard(user_id)  # ✅ Remove user from the waiting list
+            logging.info(f"User {user_id} authenticated successfully.")
             await update.message.reply_text("✅ Authentication successful! You can now use the bot.")
         else:
             await update.message.reply_text("❌ Wrong password. Try again.")