Update telegram_bot.py

telegram_bot.py

@@ -1,16 +1,24 @@
 import logging
+import os
 import requests
 from telegram import Update
 from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackContext
 
+
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 
-TOKEN = "TELEGRAM_BOT_TOKEN"
+
+# Get the Telegram Bot Token from environment variables
+TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+if not TOKEN:
+    raise ValueError("Missing Telegram Bot Token. Please set TELEGRAM_BOT_TOKEN in environment variables.")
+    
 
 # Flask API URL
 API_URL = "https://demaking-decision-helper-bot.hf.space/ask"
 
+
 # Function to send user input to Flask API and return the response
 def get_response_from_api(user_text):
     try:
@@ -23,16 +31,19 @@ def get_response_from_api(user_text):
         logging.error(f"Error connecting to API: {e}")
         return "Error: Could not connect to the server"
 
+
 # Function to handle incoming messages from users
 def handle_message(update: Update, context: CallbackContext):
     user_text = update.message.text
     response = get_response_from_api(user_text)
     update.message.reply_text(response)
 
+
 # Function to handle /start command
 def start(update: Update, context: CallbackContext):
     update.message.reply_text("Hello! Tell me your Decision-Making issue in HE or EN, and I'll try to help you.")
 
+
 # Initialize Telegram bot
 def run_telegram_bot():
     updater = Updater(TOKEN, use_context=True)
@@ -44,5 +55,6 @@ def run_telegram_bot():
     updater.start_polling()
     updater.idle()
 
+
 if __name__ == "__main__":
     run_telegram_bot()