Update app.py

app.py

@@ -2687,6 +2687,9 @@ async def handle_ai_chat_mode(from_number: str, query: str, reply_language: str
     try:
         logger.info(f"[AI Chat] Processing query: '{query}' for {from_number} in {reply_language}")
         
+        # Get user context
+        user_context = context_manager.get_context(from_number)
+        
         # Check for navigation commands first
         if query.lower().strip() in ['main', 'menu', 'start', 'home', 'back']:
             logger.info(f"[AI Chat] Navigation command detected: '{query}' -> returning to main menu")
@@ -2723,13 +2726,15 @@ async def handle_ai_chat_mode(from_number: str, query: str, reply_language: str
         
         # Get all products data for context
         all_products = []
+        total_products = 0
         if products_df is not None and not products_df.empty:
             all_products = products_df.to_dict('records')
+            total_products = len(products_df)
         
         # Create comprehensive context for AI
         products_context = ""
         if all_products:
-            products_context = "Available Veterinary Products:\n"
+            products_context = f"Available Veterinary Products (Total: {total_products} products):\n"
             for i, product in enumerate(all_products[:50], 1):  # Limit to first 50 products for context
                 product_name = product.get('Product Name', 'N/A')
                 category = product.get('Category', 'N/A')
@@ -2739,13 +2744,15 @@ async def handle_ai_chat_mode(from_number: str, query: str, reply_language: str
                 products_context += f"   Composition: {composition}\n"
                 products_context += f"   Target Species: {target_species}\n\n"
         
-        # Create AI prompt
+        # Create AI prompt with accurate product count
         if reply_language == 'ur':
             prompt = f"""
 آپ Apex Biotical کے Veterinary AI Assistant ہیں۔ آپ کو veterinary products اور treatments کے بارے میں معلومات فراہم کرنی ہیں۔
 
 یوزر کا سوال: {query}
 
+کل veterinary products کی تعداد: {total_products}
+
 دستیاب veterinary products:
 {products_context}
 
@@ -2755,6 +2762,7 @@ async def handle_ai_chat_mode(from_number: str, query: str, reply_language: str
 3. اگر یہ general veterinary advice ہے تو professional guidance دیں
 4. اردو میں جواب دیں
 5. جواب professional اور helpful ہو
+6. اگر یوزر نے products کی تعداد کے بارے میں پوچھا ہے تو صحیح تعداد ({total_products}) بتائیں
 
 جواب:
 """
@@ -2764,6 +2772,8 @@ You are Apex Biotical's Veterinary AI Assistant. You provide information about v
 
 User Query: {query}
 
+Total number of veterinary products: {total_products}
+
 Available Veterinary Products:
 {products_context}
 
@@ -2773,6 +2783,7 @@ Please:
 3. If it's general veterinary advice, provide professional guidance
 4. Answer in English
 5. Keep the response professional and helpful
+6. If the user asks about the number of products, provide the accurate count ({total_products})
 
 Response:
 """
@@ -2819,16 +2830,20 @@ Response:
             error_msg = "❌ AI Assistant encountered an error. Please try again or type 'main' to return to main menu."
         send_whatsjet_message(from_number, error_msg)
 
-def generate_veterinary_welcome_message(phone_number, user_context):
-    return (
-        "👋 Welcome to Apex Biotical Veterinary WhatsApp Bot!\n"
-        "How can I assist you today?\n"
-        "1️⃣ Search Products\n"
+def generate_veterinary_welcome_message(phone_number=None, user_context=None):
+    """Generate enhanced welcome message with veterinary domain expertise"""
+    welcome_msg = (
+        "🏥 *Welcome to Apex Biotical Veterinary Bot*\n\n"
+        "We provide comprehensive veterinary products and support.\n\n"
+        "📋 *Available Options:*\n"
+        "1️⃣ Search Veterinary Products\n"
         "2️⃣ Browse Categories\n"
         "3️⃣ Download Catalog\n"
-        "4️⃣ Chat with Veterinary AI Assistant\n"
-        "Please reply with a number (1-4) or type your question."
+        "4️⃣ Chat with Veterinary AI Assistant\n\n"
+        "💬 *Select an option or ask about specific products*\n"
+        "🎤 *You can also send voice messages*"
     )
+    return welcome_msg
 
 @app.get("/test-whatsjet")
 async def test_whatsjet():
@@ -2839,6 +2854,331 @@ async def test_whatsjet():
     except Exception as e:
         return {"error": str(e)}
 
+# Load products data on startup
+def load_products_data():
+    """Load products data from CSV file"""
+    global products_df
+    try:
+        if os.path.exists(CSV_FILE):
+            products_df = pd.read_csv(CSV_FILE)
+            logger.info(f"✅ Loaded {len(products_df)} products from {CSV_FILE}")
+        else:
+            logger.warning(f"⚠️ CSV file {CSV_FILE} not found")
+            products_df = pd.DataFrame()
+    except Exception as e:
+        logger.error(f"❌ Error loading products data: {e}")
+        products_df = pd.DataFrame()
+
+def convert_drive_link(link: str) -> str:
+    """Convert Google Drive link to direct download link"""
+    if 'drive.google.com' in link:
+        file_id = link.split('/')[-2] if '/d/' in link else link.split('/')[-1]
+        return f"https://drive.google.com/uc?export=download&id={file_id}"
+    return link
+
+def format_number_with_emoji(number: int) -> str:
+    """Format number with emoji"""
+    emoji_map = {
+        1: "1️⃣", 2: "2️⃣", 3: "3️⃣", 4: "4️⃣", 5: "5️⃣",
+        6: "6️⃣", 7: "7️⃣", 8: "8️⃣", 9: "9️⃣", 10: "🔟",
+        11: "1️⃣1️⃣", 12: "1️⃣2️⃣", 13: "1️⃣3️⃣", 14: "1️⃣4️⃣", 15: "1️⃣5️⃣",
+        16: "1️⃣6️⃣", 17: "1️⃣7️⃣", 18: "1️⃣8️⃣", 19: "1️⃣9️⃣", 20: "2️⃣0️⃣",
+        21: "2️⃣1️⃣", 22: "2️⃣2️⃣", 23: "2️⃣3️⃣"
+    }
+    return emoji_map.get(number, f"{number}.")
+
+async def display_all_products(from_number: str):
+    """Display all products in multiple messages and update menu context"""
+    try:
+        user_context = context_manager.get_context(from_number)
+        current_state = user_context.get('current_state', 'main_menu')
+        logger.info(f"[Display] display_all_products called for {from_number} in state: {current_state}")
+        if current_state == 'all_products_menu':
+            logger.warning(f"[Display] Already in all_products_menu state for {from_number}, skipping display")
+            return
+        if products_df is None or products_df.empty:
+            send_whatsjet_message(from_number, "❌ No products available at the moment.")
+            return
+        # Set state to all_products_menu and store menu context
+        products = products_df.to_dict('records')
+        context_manager.update_context(
+            from_number,
+            current_state='all_products_menu',
+            current_menu='all_products_menu',
+            current_menu_options=[p.get('Product Name', 'Unknown') for p in products],
+            available_products=products
+        )
+        logger.info(f"[Display] Set state to all_products_menu for {from_number}")
+        # Send products in chunks
+        chunk_size = 5
+        for i in range(0, len(products), chunk_size):
+            chunk = products[i:i + chunk_size]
+            message = f"📋 *Products ({i+1}-{min(i+chunk_size, len(products))} of {len(products)})*\n\n"
+            for j, product in enumerate(chunk, i+1):
+                message += f"{format_number_with_emoji(j)} {product.get('Product Name', 'Unknown')}\n"
+                if product.get('Category'):
+                    message += f"   Category: {product.get('Category')}\n"
+                message += "\n"
+            send_whatsjet_message(from_number, message)
+        send_whatsjet_message(from_number, 
+            "💬 Type a product name to get detailed information, or type 'main' to return to main menu.")
+    except Exception as e:
+        logger.error(f"[Display] Error displaying products: {e}")
+        send_whatsjet_message(from_number, "❌ Error displaying products. Please try again.")
+
+def get_all_categories():
+    """Return a list of all unique categories from the products DataFrame"""
+    if products_df is not None and not products_df.empty:
+        return list(products_df['Category'].unique())
+    return []
+
+def get_products_by_category(category: str):
+    """Get products by category"""
+    if products_df is None or products_df.empty:
+        return []
+    category_products = products_df[products_df['Category'] == category]
+    return category_products.to_dict('records')
+
+# Enhanced product follow-up handling
+async def handle_veterinary_product_followup(selection: str, from_number: str) -> None:
+    """
+    Handle product follow-up selections with enhanced veterinary domain support
+    """
+    try:
+        user_context = context_manager.get_context(from_number)
+        current_product = user_context.get('current_product')
+        
+        if selection == '1':
+            # Talk to Veterinary Consultant
+            current_product = user_context.get('current_product')
+            product_name = current_product.get('Product Name', 'the selected product') if current_product else 'the selected product'
+            consultant_msg = (
+                f"📞 Contact Veterinary Consultant\n\n"
+                f"Product: {product_name}\n\n"
+                "Please provide your details:\n"
+                "* Name and location\n"
+                "* Specific inquiry\n\n"
+                "💬 Example: Dr. Ali - Multan - Need consultation for respiratory problems\n\n"
+                "Type main at any time to go to main menu."
+            )
+            send_whatsjet_message(from_number, consultant_msg)
+        elif selection == '2':
+            # Inquire about Product Availability
+            await handle_availability_inquiry(from_number, user_context)
+        elif selection == '3':
+            # Back to Main Menu
+            welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
+            send_whatsjet_message(from_number, welcome_msg)
+            context_manager.update_context(
+                from_number,
+                current_state='main_menu',
+                current_menu='main_menu',
+                current_menu_options=list(MENU_CONFIG['main_menu']['option_descriptions'].values())
+            )
+            return
+        else:
+            send_whatsjet_message(from_number, "❌ Invalid selection. Please choose 1, 2, or 3.")
+            return
+    except Exception as e:
+        logger.error(f"Error in product follow-up: {e}")
+        welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
+        send_whatsjet_message(from_number, welcome_msg)
+        context_manager.update_context(
+            from_number,
+            current_state='main_menu',
+            current_menu='main_menu',
+            current_menu_options=list(MENU_CONFIG['main_menu']['option_descriptions'].values())
+        )
+
+# Enhanced product inquiry handling
+async def handle_veterinary_product_inquiry(product_name: str, from_number: str) -> None:
+    """
+    Handle product inquiry with enhanced veterinary domain support
+    """
+    try:
+        # Search for the product
+        products = get_veterinary_product_matches(product_name)
+        
+        if products:
+            selected_product = products[0]
+            context_manager.update_context(
+                from_number, 
+                current_product=selected_product, 
+                current_state='product_inquiry',
+                current_menu='product_inquiry',
+                current_menu_options=list(MENU_CONFIG['product_inquiry']['option_descriptions'].values())
+            )
+            # Get updated context with last message
+            context = context_manager.get_context(from_number)
+            response = generate_veterinary_product_response(selected_product, context)
+            send_whatsjet_message(from_number, response)
+            
+            # Add to conversation history
+            context_manager.add_to_history(from_number, product_name, response)
+            
+        else:
+            # Enhanced "not found" response with veterinary suggestions
+            message = (
+                "❌ *Product Not Found*\n\n"
+                f"🔍 *We couldn't find '{product_name}' in our veterinary database.*\n\n"
+                "💡 *Try these alternatives:*\n"
+                "• Check spelling (e.g., 'Hydropex' not 'Hydro pex')\n"
+                "• Search by symptoms (e.g., 'respiratory', 'liver support')\n"
+                "• Search by category (e.g., 'antibiotic', 'vitamin')\n"
+                "• Search by species (e.g., 'poultry', 'livestock')\n\n"
+                "🏥 *Popular Veterinary Products:*\n"
+                "• Hydropex (Electrolyte supplement)\n"
+                "• Heposel (Liver tonic)\n"
+                "• Bromacid (Respiratory support)\n"
+                "• Tribiotic (Antibiotic)\n"
+                "• Symodex (Multivitamin)\n\n"
+                "💬 *Type 'main' to return to main menu or try another search.*"
+            )
+            
+            send_whatsjet_message(from_number, message)
+            
+    except Exception as e:
+        logger.error(f"Error in product inquiry: {e}")
+        # Instead of sending a generic error, return to main menu
+        welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
+        send_whatsjet_message(from_number, welcome_msg)
+        context_manager.update_context(from_number, current_state='main_menu', current_menu='main_menu', current_menu_options=list(MENU_CONFIG['main_menu']['option_descriptions'].values()))
+
+async def handle_category_selection(selection: str, from_number: str):
+    """Handle category selection from category_selection_menu state"""
+    try:
+        user_context = context_manager.get_context(from_number)
+        available_categories = user_context.get('available_categories', [])
+        
+        logger.info(f"[Category] Handling selection '{selection}' for {from_number}")
+        logger.info(f"[Category] Available categories: {len(available_categories)}")
+        
+        if not available_categories:
+            logger.warning("[Category] No available categories")
+            send_whatsjet_message(from_number, "❌ No categories available. Please type 'main' to return to main menu.")
+            return
+        
+        if selection.isdigit() and 1 <= int(selection) <= len(available_categories):
+            selected_category = available_categories[int(selection) - 1]
+            logger.info(f"[Category] Selected category: '{selected_category}'")
+            
+            products = get_products_by_category(selected_category)
+            logger.info(f"[Category] Found {len(products)} products in category '{selected_category}'")
+            
+            if products:
+                product_message = f"📦 *Products in {selected_category}*\n\n"
+                for i, product in enumerate(products[:10], 1):  # Show first 10 products
+                    product_message += f"{format_number_with_emoji(i)} {product.get('Product Name', 'Unknown')}\n"
+                
+                if len(products) > 10:
+                    product_message += f"\n... and {len(products) - 10} more products"
+                
+                product_message += "\n\nSelect a product number or type 'main' to return to main menu."
+                
+                logger.info(f"[Category] Sending product message for category '{selected_category}'")
+                send_whatsjet_message(from_number, product_message)
+                context_manager.update_context(
+                    from_number, 
+                    current_state='category_products_menu',
+                    current_menu='category_products_menu',
+                    current_menu_options=[f"Product {i+1}" for i in range(len(products))],
+                    available_products=products, 
+                    current_category=selected_category
+                )
+                logger.info(f"[Category] Updated context to category_products_menu with {len(products)} products")
+            else:
+                logger.warning(f"[Category] No products found in category '{selected_category}'")
+                send_whatsjet_message(from_number, f"❌ No products found in {selected_category}")
+        else:
+            logger.warning(f"[Category] Invalid category selection: '{selection}' (valid range: 1-{len(available_categories)})")
+            send_whatsjet_message(from_number, "❌ Invalid selection. Please choose a valid category number.")
+            
+    except Exception as e:
+        logger.error(f"[Category] Error handling category selection: {e}")
+        # Instead of sending a generic error, return to main menu
+        welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
+        send_whatsjet_message(from_number, welcome_msg)
+        context_manager.update_context(
+            from_number, 
+            current_state='main_menu',
+            current_menu='main_menu',
+            current_menu_options=list(MENU_CONFIG['main_menu']['option_descriptions'].values())
+        )
+
+def get_menu_validation_message(current_state: str, user_context: dict) -> str:
+    """Get specific validation message for the current menu state"""
+    if current_state == 'main_menu':
+        return "❌ Invalid selection for Main Menu. Please choose:\n1️⃣ Search Veterinary Products\n2️⃣ Browse Categories\n3️⃣ Download Catalog\n4️⃣ Chat with Veterinary AI Assistant\n\n💬 Type 'main' to return to main menu."
+    
+    elif current_state == 'all_products_menu':
+        available_products = user_context.get('available_products', [])
+        total_products = len(available_products) if available_products else 23
+        return f"❌ Invalid selection for Product List. Please choose a number between 1 and {total_products}.\n\n💬 Type 'main' to return to main menu."
+    
+    elif current_state == 'category_selection_menu':
+        available_categories = user_context.get('available_categories', [])
+        total_categories = len(available_categories) if available_categories else 0
+        if total_categories > 0:
+            return f"❌ Invalid selection for Category List. Please choose a number between 1 and {total_categories}.\n\n💬 Type 'main' to return to main menu."
+        else:
+            return "❌ No categories available. Please type 'main' to return to main menu."
+    
+    elif current_state == 'category_products_menu':
+        available_products = user_context.get('available_products', [])
+        total_products = len(available_products) if available_products else 0
+        if total_products > 0:
+            return f"❌ Invalid selection for Category Products. Please choose a number between 1 and {total_products}.\n\n💬 Type 'main' to return to main menu."
+        else:
+            return "❌ No products available in this category. Please type 'main' to return to main menu."
+    
+    elif current_state == 'product_inquiry':
+        return "❌ Invalid selection for Product Options. Please choose:\n1️⃣ Talk to Veterinary Consultant\n2️⃣ Inquire about Product Availability\n3️⃣ Back to Main Menu"
+    
+    elif current_state == 'ai_chat_mode':
+        return "💬 *You're in AI Chat mode. Ask me anything about veterinary care!*\n\nType 'main' to return to main menu."
+    
+    else:
+        return "❌ Invalid selection. Please type 'main' to return to main menu."
+
+def is_valid_menu_selection(selection: str, current_state: str, user_context: dict) -> bool:
+    """Check if a selection is valid for the current menu state"""
+    # Use the intelligent voice command processor for consistent handling
+    mapped_selection = process_intelligent_voice_command(selection, current_state, user_context)
+    
+    if not mapped_selection.isdigit():
+        return False
+    
+    selection_num = int(mapped_selection)
+    
+    if current_state == 'main_menu':
+        return 1 <= selection_num <= 4
+    
+    elif current_state == 'all_products_menu':
+        available_products = user_context.get('available_products', [])
+        total_products = len(available_products) if available_products else 23
+        return 1 <= selection_num <= total_products
+    
+    elif current_state == 'category_selection_menu':
+        available_categories = user_context.get('available_categories', [])
+        total_categories = len(available_categories) if available_categories else 0
+        return 1 <= selection_num <= total_categories
+    
+    elif current_state == 'category_products_menu':
+        available_products = user_context.get('available_products', [])
+        total_products = len(available_products) if available_products else 0
+        return 1 <= selection_num <= total_products
+    
+    elif current_state == 'product_inquiry':
+        return 1 <= selection_num <= 3
+    
+    elif current_state == 'ai_chat_mode':
+        return mapped_selection == 'main'
+    
+    return False
+
+# Load products on startup
+load_products_data()
+
 if __name__ == "__main__":
     # Launch FastAPI app
     import uvicorn