Update bot_telegram.py

bot_telegram.py

@@ -4,18 +4,21 @@ import asyncio
 import requests
 from telegram import Update
 from telegram.ext import Application, CommandHandler, MessageHandler, filters, CallbackContext
+import aiohttp
 
+# Configure logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
+# Load environment variables from Hugging Face Secrets
 BOT_TOKEN = os.getenv("BOT_TOKEN")
 BASE_URL = os.getenv("BASE_URL")
 API_USERNAME = os.getenv("API_USERNAME")
 API_PASSWORD = os.getenv("API_PASSWORD")
 
+# Set the secret password for authentication
+SECRET_PASSWORD = "secure123"  # Change this to your desired password
 
-SECRET_PASSWORD = "secure123"
-
-
+# Dictionary to store authenticated users
 AUTHENTICATED_USERS = set()
 AWAITING_PASSWORD = set()
 
@@ -33,6 +36,7 @@ class TelegramBot:
         # API Endpoints
         self.login_url = f"{self.base_url}/api/v1/auth/login"
         self.ai_url = f"{self.base_url}/api/v1/questions/text"
+        self.excel_url = f"{self.base_url}/api/v1/questions/excel"
 
         # Start Telegram Bot
         self.app = Application.builder().token(self.bot_token).build()
@@ -155,10 +159,62 @@ class TelegramBot:
 
         await update.message.reply_text(bot_reply)
 
+    async def handle_excel(self, update: Update, context: CallbackContext):
+        """Handles Excel file uploads."""
+        user_id = update.message.from_user.id
+
+        if user_id not in AUTHENTICATED_USERS:
+            await update.message.reply_text("❌ You are not authenticated. Please enter the password first.")
+            return
+
+        if not self.auth_token:
+            self.authenticate()
+
+        if not self.auth_token:
+            await update.message.reply_text("Authentication failed. Please try again later.")
+            return
+
+        try:
+            # Get file from Telegram
+            file = await context.bot.get_file(update.message.document.file_id)
+            file_bytes = await file.download_as_bytearray()
+
+            # Prepare the file upload
+            headers = {
+                "Authorization": f"Bearer {self.auth_token}",
+                "accept": "application/json"
+            }
+
+            # Create form data with the file
+            form_data = aiohttp.FormData()
+            form_data.add_field('file',
+                              file_bytes,
+                              filename=update.message.document.file_name,
+                              content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
+
+            async with aiohttp.ClientSession() as session:
+                async with session.post(self.excel_url, headers=headers, data=form_data) as response:
+                    if response.status == 200:
+                        response_json = await response.json()
+                        await update.message.reply_text(response_json.get("message", "Excel file processed successfully!"))
+                    elif response.status == 401:
+                        logging.warning("Authorization expired. Re-authenticating...")
+                        self.authenticate()
+                        await self.handle_excel(update, context)
+                    else:
+                        error_text = await response.text()
+                        await update.message.reply_text(f"Error processing Excel file: {error_text}")
+
+        except Exception as e:
+            logging.error(f"Error handling Excel file: {e}")
+            await update.message.reply_text(f"Error processing Excel file: {str(e)}")
+
     def setup_handlers(self):
         """Set up Telegram command and message handlers."""
         self.app.add_handler(CommandHandler("start", self.start_command))
         self.app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_message))
+        # Add handler for document messages
+        self.app.add_handler(MessageHandler(filters.Document.FileExtension("xlsx") | filters.Document.FileExtension("xls"), self.handle_excel))
 
     def run(self):
         """Start the bot and listen for messages."""