Update bot_telegram.py

bot_telegram.py

@@ -13,6 +13,7 @@ logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(
 # Load environment variables from Hugging Face Secrets
 BOT_TOKEN = os.getenv("BOT_TOKEN")
 BASE_URL = os.getenv("BASE_URL")
+HF_TOKEN = os.getenv("HF_TOKEN")
 API_USERNAME = os.getenv("API_USERNAME")
 API_PASSWORD = os.getenv("API_PASSWORD")
 
@@ -23,6 +24,7 @@ SECRET_PASSWORD = "secure123"  # Change this to your desired password
 AUTHENTICATED_USERS = set()
 AWAITING_PASSWORD = set()
 
+
 class TelegramBot:
     """A Telegram bot with password-based authentication."""
 
@@ -132,11 +134,13 @@ class TelegramBot:
 
         user_message = update.message.text
 
+        hf_authorization = "Bearer " + HF_TOKEN
         headers = {
-            "Authorization": f"Bearer {self.auth_token}",
+            "Authorization": hf_authorization,
             "accept": "application/json"
         }
 
+
         json_payload = {"question": user_message}
         form_payload = {"question": user_message}
 
@@ -146,7 +150,8 @@ class TelegramBot:
 
             if response.status_code == 422:
                 logging.warning("JSON format rejected. Retrying with form-data...")
-                response = requests.post(self.ai_url, headers={**headers, "Content-Type": "application/x-www-form-urlencoded"}, data=form_payload)
+                response = requests.post(self.ai_url, headers={**headers, "Content-Type": "application/x-www-form-urlencoded"},
+                                         data=form_payload)
 
             if response.status_code == 200:
                 bot_reply = response.json().get("answer", "I didn't understand that.")
@@ -156,9 +161,11 @@ class TelegramBot:
                 await self.chat_with_ai(update, context)
                 return
             else:
+                logging.error(f"Error: {response.status_code}")
                 bot_reply = f"Error: {response.status_code}"
 
         except Exception as e:
+            logging.error(f"Connection error: {e}")
             bot_reply = f"Connection error: {e}"
 
         await update.message.reply_text(bot_reply)
@@ -192,7 +199,7 @@ class TelegramBot:
             file_bytes = await file.download_as_bytearray()
 
             headers = {
-                "Authorization": f"Bearer {self.auth_token}",
+                "Authorization": f"Bearer {HF_TOKEN}",
                 "accept": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
             }
 
@@ -200,16 +207,16 @@ class TelegramBot:
                 # Create form data with the file
                 form_data = aiohttp.FormData()
                 form_data.add_field('file',
-                                  file_bytes,
-                                  filename=document.file_name,
-                                  content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
+                                    file_bytes,
+                                    filename=document.file_name,
+                                    content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
 
                 # Send file to API
                 async with session.post(self.excel_url, headers=headers, data=form_data) as response:
                     if response.status == 200:
                         # Get the Excel file content directly
                         file_content = await response.read()
-                        
+
                         # Send the Excel file back to Telegram
                         await context.bot.send_document(
                             chat_id=update.effective_chat.id,
@@ -242,6 +249,7 @@ class TelegramBot:
         logging.info("Starting Telegram bot...")
         self.app.run_polling()
 
+
 if __name__ == "__main__":
     bot = TelegramBot(
         bot_token=BOT_TOKEN,