Add application file

Dockerfile

@@ -30,4 +30,4 @@ COPY . .
 EXPOSE 7860
 
 # Command to run the application
-CMD ["python", "telegram_bot.py"] 
+CMD ["python", "telegram_bot1.py"] 

telegram_bot.py

@@ -36,6 +36,8 @@ class WeBookBot:
             "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
         }
         self.monitor = EventMonitor(self)
+        # Add storage for user credentials
+        self.user_credentials = {}
 
     def _get_events_payload(self, category=None, date_filter=None, price_min=None, price_max=None, zones=None, country_code="SA", limit=20, skip=0):
         """Generate the payload for the events API request with flexible filtering using the getShows query."""
@@ -217,68 +219,156 @@ class WeBookBot:
 
     async def start_command(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """Handle the /start command"""
-        # Store default filters in context user_data if they don't exist
-        if 'category' not in context.user_data:
-            context.user_data['category'] = None # Or a default category slug if needed
-        if 'date_filter' not in context.user_data:
-            context.user_data['date_filter'] = 'today' # Default date filter
-        # Add placeholders for other filters
-        if 'price_min' not in context.user_data:
-            context.user_data['price_min'] = None
-        if 'price_max' not in context.user_data:
-            context.user_data['price_max'] = None
-        if 'zones' not in context.user_data:
-            context.user_data['zones'] = None # Or a list of default zone slugs if needed
-        if 'country_code' not in context.user_data:
-             context.user_data['country_code'] = 'SA' # Default country
-
         keyboard = [
             [
-                InlineKeyboardButton("☀️ اليوم", callback_data='date_today'),
-                InlineKeyboardButton("🗓️ غداً", callback_data='date_tomorrow'),
-                InlineKeyboardButton("📅 هذا الأسبوع", callback_data='date_this_week')
-            ],
-            [
-                InlineKeyboardButton("🎁 العروض", callback_data='category_offers'),
-                InlineKeyboardButton("⚽ الرياضة", callback_data='category_sports-events')
+                InlineKeyboardButton("🔑 Login to WeBook", callback_data='login_webook'),
+                InlineKeyboardButton("ℹ️ Help", callback_data='help')
             ],
             [
-                InlineKeyboardButton("🎭 المسرح والفنون الأدائية", callback_data='category_theater-and-performing-arts'),
-                InlineKeyboardButton("⛰️ الأنشطة والمغامرات", callback_data='category_activities-adventures')
+                InlineKeyboardButton("👤 Set Account Info", callback_data='set_account_info'),
+                InlineKeyboardButton("📊 Account Status", callback_data='account_status')
             ],
             [
-                InlineKeyboardButton("🎶 موسیقی", callback_data='category_music-events'),
-                InlineKeyboardButton("✨ التجارب", callback_data='category_experience')
-            ],
-            [
-                InlineKeyboardButton("🍽️ المطاعم", callback_data='category_restaurant-and-cafe')
-            ],
-            # Add buttons for other filters would be complex here.
-            # You might consider a "Filter Options" button that leads to a new message
-            # with more specific filter buttons or instructions.
-            # For now, we just keep the category and date buttons.
-            [
-                InlineKeyboardButton("✅ Show Events with Current Filters", callback_data='show_events')
-            ],
-             [
-                InlineKeyboardButton("🔄 Reset Filters", callback_data='reset_filters')
-             ],
-             [
-                InlineKeyboardButton("🔍 Start Event Monitoring", callback_data='start_monitoring'),
-                InlineKeyboardButton("⏹️ Stop Monitoring", callback_data='stop_monitoring')
-             ]
+                InlineKeyboardButton("🎫 Browse Events", callback_data='browse_events')
+            ]
         ]
         reply_markup = InlineKeyboardMarkup(keyboard)
 
-        # Display current filters in the message
-        current_filters_text = self._get_current_filters_text(context.user_data)
-
         await update.message.reply_text(
-            f'Welcome to WeBook Events Bot! 🎉\n'
-            f'Please select filters or "Show Events":\n{current_filters_text}',
+            'Welcome to WeBook Events Bot! 🎉\n'
+            'Please select an option:',
             reply_markup=reply_markup
         )
 
+    async def set_account_info_command(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """Handle the /setinfoaccountwebook command"""
+        await update.message.reply_text(
+            "Please enter your WeBook credentials in the following format:\n"
+            "email:your.email@example.com\n"
+            "password:yourpassword"
+        )
+        context.user_data['awaiting_credentials'] = True
+
+    async def account_status_command(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """Handle the /stateloginwebook command"""
+        user_id = update.effective_user.id
+        if user_id in self.user_credentials:
+            await update.message.reply_text(
+                f"✅ WeBook Account Status:\n"
+                f"Email: {self.user_credentials[user_id]['email']}\n"
+                f"Status: Logged In"
+            )
+        else:
+            await update.message.reply_text(
+                "❌ No WeBook account configured.\n"
+                "Use /setinfoaccountwebook to set up your account."
+            )
+
+    async def help_command(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """Handle the /help command"""
+        help_text = (
+            "🤖 *WeBook Bot Commands*\n\n"
+            "/start - Start the bot and show main menu\n"
+            "/setinfoaccountwebook - Set your WeBook account credentials\n"
+            "/stateloginwebook - Check your WeBook account status\n"
+            "/help - Show this help message\n"
+            "/info - Show bot information\n\n"
+            "To use the bot:\n"
+            "1. Set your WeBook credentials using /setinfoaccountwebook\n"
+            "2. Login using the 'Login to WeBook' button\n"
+            "3. Browse and book events!"
+        )
+        await update.message.reply_text(help_text, parse_mode='Markdown')
+
+    async def info_command(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """Handle the /info command"""
+        info_text = (
+            "🤖 *WeBook Bot Information*\n\n"
+            "Version: 1.0.0\n"
+            "Description: A Telegram bot for browsing and booking events on WeBook\n"
+            "Features:\n"
+            "- Browse events by category\n"
+            "- Filter events by date\n"
+            "- Book events automatically\n"
+            "- Monitor for new events\n"
+            "- Account management"
+        )
+        await update.message.reply_text(info_text, parse_mode='Markdown')
+
+    async def handle_credentials_input(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """Handle user input for WeBook credentials"""
+        if not context.user_data.get('awaiting_credentials'):
+            return
+
+        try:
+            text = update.message.text.strip()
+            email = None
+            password = None
+
+            for line in text.split('\n'):
+                if line.startswith('email:'):
+                    email = line.split(':', 1)[1].strip()
+                elif line.startswith('password:'):
+                    password = line.split(':', 1)[1].strip()
+
+            if email and password:
+                user_id = update.effective_user.id
+                self.user_credentials[user_id] = {
+                    'email': email,
+                    'password': password
+                }
+                context.user_data['awaiting_credentials'] = False
+                await update.message.reply_text(
+                    "✅ WeBook credentials saved successfully!\n"
+                    "You can now use the 'Login to WeBook' button to log in."
+                )
+            else:
+                await update.message.reply_text(
+                    "❌ Invalid format. Please use:\n"
+                    "email:your.email@example.com\n"
+                    "password:yourpassword"
+                )
+        except Exception as e:
+            logger.error(f"Error handling credentials: {e}")
+            await update.message.reply_text("❌ Error saving credentials. Please try again.")
+
+    async def handle_login(self, query: CallbackQuery, context: ContextTypes.DEFAULT_TYPE):
+        """Handle WeBook login button click"""
+        user_id = query.from_user.id
+        if user_id not in self.user_credentials:
+            await query.edit_message_text(
+                "❌ No WeBook credentials found.\n"
+                "Please use /setinfoaccountwebook to set up your account first."
+            )
+            return
+
+        try:
+            # Create login instance
+            login = WeBookLogin(
+                email=self.user_credentials[user_id]['email'],
+                password=self.user_credentials[user_id]['password']
+            )
+            
+            # Attempt login
+            page = login.login()
+            
+            if page:
+                await query.edit_message_text(
+                    "✅ Successfully logged in to WeBook!\n"
+                    "You can now browse and book events."
+                )
+                # Store the page object for future use
+                context.user_data['webook_page'] = page
+            else:
+                await query.edit_message_text(
+                    "❌ Login failed. Please check your credentials and try again."
+                )
+        except Exception as e:
+            logger.error(f"Login error: {e}")
+            await query.edit_message_text(
+                "❌ An error occurred during login. Please try again later."
+            )
+
     async def button_callback(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """Handle button callbacks"""
         query = update.callback_query
@@ -287,8 +377,18 @@ class WeBookBot:
         try:
             callback_data = query.data
 
-            # --- Handle Filter Button Presses ---
-            if callback_data.startswith('category_'):
+            if callback_data == 'login_webook':
+                await self.handle_login(query, context)
+            elif callback_data == 'help':
+                await self.help_command(update, context)
+            elif callback_data == 'set_account_info':
+                await self.set_account_info_command(update, context)
+            elif callback_data == 'account_status':
+                await self.account_status_command(update, context)
+            elif callback_data == 'browse_events':
+                # Show the original event browsing interface
+                await self.start_command(update, context)
+            elif callback_data.startswith('category_'):
                 selected_category = callback_data.replace('category_', '')
                 # Toggle category selection (assuming single selection for simplicity for now)
                 if context.user_data.get('category') == selected_category:
@@ -475,7 +575,7 @@ class WeBookBot:
 
                 # Reset skip to 0 for a new search
                 context.user_data['skip'] = 0
-                current_limit = 20
+                current_limit = 10
 
                 payload = self._get_events_payload(
                     category=current_category,
@@ -612,7 +712,7 @@ class WeBookBot:
             logger.error(f"Error processing callback: {str(e)}")
             logger.error(f"Response content: {response.text if 'response' in locals() else 'No response'}")
             await query.edit_message_text(
-                "Sorry, there was an error fetching the events. Please try again later."
+                "Sorry, there was an error processing your request. Please try again later."
             )
 
     async def _start_monitoring_setup(self, query: CallbackQuery, context: ContextTypes.DEFAULT_TYPE):
@@ -756,9 +856,12 @@ class WeBookBot:
             
             # Add handlers
             application.add_handler(CommandHandler("start", self.start_command))
+            application.add_handler(CommandHandler("setinfoaccountwebook", self.set_account_info_command))
+            application.add_handler(CommandHandler("stateloginwebook", self.account_status_command))
+            application.add_handler(CommandHandler("help", self.help_command))
+            application.add_handler(CommandHandler("info", self.info_command))
             application.add_handler(CallbackQueryHandler(self.button_callback))
-            # Add a MessageHandler for text input to handle event number selection
-            application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_event_number_input)) # Listen for non-command text
+            application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_credentials_input))
             
             # Start the bot
             logger.info("Starting bot...")