Update app.py

app.py

@@ -147,11 +147,13 @@ MENU_CONFIG = {
     },
     'product_inquiry': {
         'name': 'Product Inquiry Menu',
-        'valid_options': ['1', '2', '3'],
+        'valid_options': ['1', '2', '3', '4', '5'],
         'option_descriptions': {
             '1': 'Talk to Veterinary Consultant',
             '2': 'Inquire about Product Availability',
-            '3': 'Back to Main Menu'
+            '3': 'View Product Image',
+            '4': 'Download Detailed PDF',
+            '5': 'Back to Main Menu'
         }
     },
     'ai_chat': {
@@ -2367,7 +2369,8 @@ async def handle_contact_request_response(from_number: str, response: str):
         )
     except Exception as e:
         logger.error(f"[Contact] Error handling contact response: {e}")
-        # Instead of sending a generic error, return to main menu
+        # Get user context before using it
+        user_context = context_manager.get_context(from_number)
         welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
         send_whatsjet_message(from_number, welcome_msg)
         context_manager.update_context(
@@ -2471,7 +2474,8 @@ async def handle_availability_request_response(from_number: str, response: str):
         )
     except Exception as e:
         logger.error(f"[Availability] Error handling availability response: {e}")
-        # Instead of sending a generic error, return to main menu
+        # Get user context before using it
+        user_context = context_manager.get_context(from_number)
         welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
         send_whatsjet_message(from_number, welcome_msg)
         context_manager.update_context(
@@ -2535,6 +2539,8 @@ def send_helpful_guidance(from_number: str, current_state: str):
             )
     except Exception as e:
         logger.error(f"Error sending helpful guidance: {e}")
+        # Get user context before using it
+        user_context = context_manager.get_context(from_number)
         welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
         send_whatsjet_message(from_number, welcome_msg)
         context_manager.update_context(
@@ -2847,7 +2853,6 @@ def generate_veterinary_welcome_message(phone_number=None, user_context=None):
 
 @app.get("/test-whatsjet")
 async def test_whatsjet():
-    import requests
     try:
         resp = requests.get("https://api.whatsjet.com", timeout=5)
         return {"status": resp.status_code, "text": resp.text[:200]}
@@ -2948,10 +2953,13 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
         user_context = context_manager.get_context(from_number)
         current_product = user_context.get('current_product')
         
+        if not current_product:
+            send_whatsjet_message(from_number, "❌ No product selected. Please search for a product first.")
+            return
+        
         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'
+            product_name = current_product.get('Product Name', 'the selected product')
             consultant_msg = (
                 f"📞 Contact Veterinary Consultant\n\n"
                 f"Product: {product_name}\n\n"
@@ -2962,10 +2970,44 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
                 "Type main at any time to go to main menu."
             )
             send_whatsjet_message(from_number, consultant_msg)
+            context_manager.update_context(
+                from_number,
+                current_state='contact_request',
+                current_menu='contact_request',
+                current_menu_options=['Provide contact details']
+            )
+            
         elif selection == '2':
             # Inquire about Product Availability
             await handle_availability_inquiry(from_number, user_context)
+            
         elif selection == '3':
+            # View Product Image
+            await send_product_with_image(from_number, current_product, user_context)
+            # Return to product inquiry menu
+            response_with_media = generate_veterinary_product_response_with_media(current_product, user_context)
+            send_whatsjet_message(from_number, response_with_media['text'])
+            context_manager.update_context(
+                from_number,
+                current_state='product_inquiry',
+                current_menu='product_inquiry',
+                current_menu_options=list(MENU_CONFIG['product_inquiry']['option_descriptions'].values())
+            )
+            
+        elif selection == '4':
+            # Download Detailed PDF
+            await send_enhanced_pdf(from_number, current_product)
+            # Return to product inquiry menu
+            response_with_media = generate_veterinary_product_response_with_media(current_product, user_context)
+            send_whatsjet_message(from_number, response_with_media['text'])
+            context_manager.update_context(
+                from_number,
+                current_state='product_inquiry',
+                current_menu='product_inquiry',
+                current_menu_options=list(MENU_CONFIG['product_inquiry']['option_descriptions'].values())
+            )
+            
+        elif selection == '5':
             # Back to Main Menu
             welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
             send_whatsjet_message(from_number, welcome_msg)
@@ -2977,10 +3019,11 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
             )
             return
         else:
-            send_whatsjet_message(from_number, "❌ Invalid selection. Please choose 1, 2, or 3.")
+            send_whatsjet_message(from_number, "❌ Invalid selection. Please choose 1, 2, 3, 4, or 5.")
             return
     except Exception as e:
         logger.error(f"Error in product follow-up: {e}")
+        user_context = context_manager.get_context(from_number)
         welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
         send_whatsjet_message(from_number, welcome_msg)
         context_manager.update_context(
@@ -2993,7 +3036,7 @@ async def handle_veterinary_product_followup(selection: str, from_number: str) -
 # 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
+    Handle product inquiry with enhanced veterinary domain support and media
     """
     try:
         # Search for the product
@@ -3008,13 +3051,18 @@ async def handle_veterinary_product_inquiry(product_name: str, from_number: str)
                 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)
+            
+            # Use enhanced response with media support
+            response_with_media = generate_veterinary_product_response_with_media(selected_product, context)
+            
+            # Send the response
+            send_whatsjet_message(from_number, response_with_media['text'])
             
             # Add to conversation history
-            context_manager.add_to_history(from_number, product_name, response)
+            context_manager.add_to_history(from_number, product_name, response_with_media['text'])
             
         else:
             # Enhanced "not found" response with veterinary suggestions
@@ -3037,6 +3085,190 @@ async def handle_veterinary_product_inquiry(product_name: str, from_number: str)
             
             send_whatsjet_message(from_number, message)
             
+    except Exception as e:
+        logger.error(f"Error in product inquiry: {e}")
+        # Get user context before using it
+        user_context = context_manager.get_context(from_number)
+        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_intelligent_product_inquiry(from_number: str, query: str, user_context: dict, reply_language: str = 'en'):
+    """Handle product inquiry with OpenAI intelligence and media support"""
+    try:
+        # First try direct product search
+        products = get_veterinary_product_matches(query)
+        
+        if products:
+            # Check if this is a broad/category query (multiple products found)
+            if len(products) > 1:
+                # Use OpenAI to generate a professional summary and list all products
+                if OPENAI_API_KEY:
+                    try:
+                        # Create a comprehensive prompt for multiple products
+                        products_info = []
+                        for i, product in enumerate(products, 1):
+                            product_name = product.get('Product Name', 'N/A')
+                            category = product.get('Category', 'N/A')
+                            target_species = product.get('Target Species', 'N/A')
+                            products_info.append(f"{i}. {product_name} - {category} ({target_species})")
+                        
+                        products_text = "\n".join(products_info)
+                        
+                        prompt = f"""
+You are a professional veterinary product assistant for Apex Biotical. The user asked about "{query}" and we found {len(products)} relevant products.
+
+Available Products:
+{products_text}
+
+Please provide:
+1. A professional, welcoming response acknowledging their query
+2. A brief summary of what these products are for (if it's a category like "poultry products", explain the category)
+3. List all products with their numbers and brief descriptions
+4. Clear instructions on how to proceed
+
+Format your response professionally with emojis and clear structure. Keep it concise but informative.
+"""
+                        
+                        response = openai.ChatCompletion.create(
+                            model="gpt-4o",
+                            messages=[{"role": "user", "content": prompt}],
+                            temperature=0.7,
+                            max_tokens=400
+                        )
+                        
+                        ai_response = response.choices[0].message['content'].strip()
+                        
+                        # Add instructions for selection
+                        selection_instructions = (
+                            f"\n\n💬 *To view detailed information about any product, reply with its number (1-{len(products)})*\n"
+                            "💬 *Type 'main' to return to the main menu*"
+                        )
+                        
+                        full_response = ai_response + selection_instructions
+                        
+                        # Translate response if needed
+                        if reply_language == 'ur':
+                            try:
+                                translated_response = GoogleTranslator(source='auto', target='ur').translate(full_response)
+                                send_whatsjet_message(from_number, translated_response)
+                            except Exception as e:
+                                logger.error(f"[AI] Translation error: {e}")
+                                send_whatsjet_message(from_number, full_response)
+                        else:
+                            send_whatsjet_message(from_number, full_response)
+                        
+                        # Store the product list in context for selection handling
+                        context_manager.update_context(
+                            from_number, 
+                            current_state='intelligent_products_menu',
+                            current_menu='intelligent_products_menu',
+                            current_menu_options=[f"Product {i+1}" for i in range(len(products))],
+                            available_products=products,
+                            last_query=query
+                        )
+                        
+                        # Add to conversation history
+                        context_manager.add_to_history(from_number, query, full_response)
+                        return
+                        
+                    except Exception as e:
+                        logger.error(f"[AI] Error generating product summary: {e}")
+                        # Fall back to simple listing if AI fails
+                        pass
+                
+                # Fallback: Simple listing without AI
+                message = f"🔍 *Found {len(products)} products matching '{query}':*\n\n"
+                
+                for i, product in enumerate(products, 1):
+                    product_name = product.get('Product Name', 'N/A')
+                    category = product.get('Category', 'N/A')
+                    target_species = product.get('Target Species', 'N/A')
+                    message += f"{format_number_with_emoji(i)} {product_name}\n"
+                    message += f"   📦 {category} ({target_species})\n\n"
+                
+                message += (
+                    f"💬 *To view detailed information about any product, reply with its number (1-{len(products)})*\n"
+                    "💬 *Type 'main' to return to the main menu*"
+                )
+                
+                # Translate response if needed
+                if reply_language == 'ur':
+                    try:
+                        translated_message = GoogleTranslator(source='auto', target='ur').translate(message)
+                        send_whatsjet_message(from_number, translated_message)
+                    except Exception as e:
+                        logger.error(f"[AI] Translation error: {e}")
+                        send_whatsjet_message(from_number, message)
+                else:
+                    send_whatsjet_message(from_number, message)
+                
+                # Store the product list in context for selection handling
+                context_manager.update_context(
+                    from_number, 
+                    current_state='intelligent_products_menu',
+                    current_menu='intelligent_products_menu',
+                    current_menu_options=[f"Product {i+1}" for i in range(len(products))],
+                    available_products=products,
+                    last_query=query
+                )
+                
+                # Add to conversation history
+                context_manager.add_to_history(from_number, query, message)
+                
+            else:
+                # Single product found - show detailed information with media support
+                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)
+                
+                # Use enhanced response with media support
+                response_with_media = generate_veterinary_product_response_with_media(selected_product, context)
+                
+                # Send the response
+                send_whatsjet_message(from_number, response_with_media['text'])
+                
+                # Add to conversation history
+                context_manager.add_to_history(from_number, query, response_with_media['text'])
+            
+        else:
+            # Enhanced "not found" response with veterinary suggestions
+            message = (
+                "❌ *Product Not Found*\n\n"
+                f"🔍 *We couldn't find '{query}' 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.*"
+            )
+            
+            # Translate response if needed
+            if reply_language == 'ur':
+                try:
+                    translated_message = GoogleTranslator(source='auto', target='ur').translate(message)
+                    send_whatsjet_message(from_number, translated_message)
+                except Exception as e:
+                    logger.error(f"[AI] Translation error: {e}")
+                    send_whatsjet_message(from_number, message)
+            else:
+                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
@@ -3095,7 +3327,8 @@ async def handle_category_selection(selection: str, from_number: str):
             
     except Exception as e:
         logger.error(f"[Category] Error handling category selection: {e}")
-        # Instead of sending a generic error, return to main menu
+        # Get user context before using it
+        user_context = context_manager.get_context(from_number)
         welcome_msg = generate_veterinary_welcome_message(from_number, user_context)
         send_whatsjet_message(from_number, welcome_msg)
         context_manager.update_context(
@@ -3132,7 +3365,7 @@ def get_menu_validation_message(current_state: str, user_context: dict) -> str:
             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"
+        return "❌ Invalid selection for Product Options. Please choose:\n1️⃣ Talk to Veterinary Consultant\n2️⃣ Inquire about Product Availability\n3️⃣ View Product Image\n4️⃣ Download Detailed PDF\n5️⃣ 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."
@@ -3169,7 +3402,7 @@ def is_valid_menu_selection(selection: str, current_state: str, user_context: di
         return 1 <= selection_num <= total_products
     
     elif current_state == 'product_inquiry':
-        return 1 <= selection_num <= 3
+        return 1 <= selection_num <= 5
     
     elif current_state == 'ai_chat_mode':
         return mapped_selection == 'main'
@@ -3179,6 +3412,225 @@ def is_valid_menu_selection(selection: str, current_state: str, user_context: di
 # Load products on startup
 load_products_data()
 
+# Add these functions after the existing imports and before the main functions
+
+def get_product_image_path(product_name: str) -> str:
+    """
+    Get the image path for a product
+    Returns the path to the product image if it exists, otherwise None
+    """
+    try:
+        # Create images directory if it doesn't exist
+        images_dir = "static/images"
+        os.makedirs(images_dir, exist_ok=True)
+        
+        # Clean product name for filename
+        safe_name = re.sub(r'[^\w\s-]', '', product_name).replace(' ', '_').lower()
+        
+        # Check for common image extensions
+        image_extensions = ['.jpg', '.jpeg', '.png', '.webp', '.gif']
+        
+        for ext in image_extensions:
+            image_path = os.path.join(images_dir, f"{safe_name}{ext}")
+            if os.path.exists(image_path):
+                logger.info(f"[Image] Found product image: {image_path}")
+                return image_path
+        
+        # If no specific product image found, check for a default image
+        default_image_path = os.path.join(images_dir, "default_product.jpg")
+        if os.path.exists(default_image_path):
+            logger.info(f"[Image] Using default product image: {default_image_path}")
+            return default_image_path
+        
+        logger.warning(f"[Image] No image found for product: {product_name}")
+        return None
+        
+    except Exception as e:
+        logger.error(f"[Image] Error getting product image path: {e}")
+        return None
+
+def get_product_image_media_type(image_path: str) -> str:
+    """
+    Determine the media type based on file extension
+    """
+    if not image_path:
+        return None
+    
+    ext = os.path.splitext(image_path)[1].lower()
+    
+    media_type_map = {
+        '.jpg': 'image/jpeg',
+        '.jpeg': 'image/jpeg',
+        '.png': 'image/png',
+        '.webp': 'image/webp',
+        '.gif': 'image/gif'
+    }
+    
+    return media_type_map.get(ext, 'image/jpeg')
+
+async def send_product_with_image(from_number: str, product: Dict[str, Any], user_context: Dict[str, Any]):
+    """
+    Send product information with image if available
+    """
+    try:
+        product_name = product.get('Product Name', 'Unknown Product')
+        
+        # Generate product response
+        response = generate_veterinary_product_response(product, user_context)
+        
+        # Try to get product image
+        image_path = get_product_image_path(product_name)
+        
+        if image_path and os.path.exists(image_path):
+            # Send product info with image
+            media_type = get_product_image_media_type(image_path)
+            filename = f"{product_name.replace(' ', '_')}.jpg"
+            
+            success = send_whatsjet_message(
+                from_number,
+                response,
+                media_type=media_type,
+                media_path=image_path,
+                filename=filename
+            )
+            
+            if success:
+                logger.info(f"[Product] Successfully sent product with image: {product_name}")
+            else:
+                # Fallback to text-only if image send fails
+                logger.warning(f"[Product] Failed to send image, sending text only: {product_name}")
+                send_whatsjet_message(from_number, response)
+        else:
+            # Send text-only response
+            send_whatsjet_message(from_number, response)
+            logger.info(f"[Product] Sent product info without image: {product_name}")
+            
+    except Exception as e:
+        logger.error(f"[Product] Error sending product with image: {e}")
+        # Fallback to text-only
+        response = generate_veterinary_product_response(product, user_context)
+        send_whatsjet_message(from_number, response)
+
+async def send_enhanced_pdf(from_number: str, product: Dict[str, Any], pdf_content: bytes = None):
+    """
+    Send PDF with enhanced formatting and proper WhatsApp document sharing
+    """
+    try:
+        product_name = product.get('Product Name', 'Unknown_Product')
+        safe_name = re.sub(r'[^\w\s-]', '', product_name).replace(' ', '_')
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        filename = f"{safe_name}_Product_Info_{timestamp}.pdf"
+        
+        # Generate PDF if not provided
+        if pdf_content is None:
+            pdf_content = generate_veterinary_pdf(product)
+        
+        # Save PDF to uploads directory
+        uploads_dir = "uploads"
+        os.makedirs(uploads_dir, exist_ok=True)
+        pdf_path = os.path.join(uploads_dir, filename)
+        
+        with open(pdf_path, 'wb') as f:
+            f.write(pdf_content)
+        
+        # Send PDF as document via WhatsApp
+        success = send_whatsjet_message(
+            from_number,
+            f"📄 *{product_name} - Detailed Product Information*\n\n"
+            f"📎 Here's the complete product information in PDF format.\n"
+            f"📋 Includes: Composition, Dosage, Precautions, Storage\n\n"
+            f"💬 Type 'main' to return to main menu.",
+            media_type="application/pdf",
+            media_path=pdf_path,
+            filename=filename
+        )
+        
+        if success:
+            logger.info(f"[PDF] Successfully sent PDF for product: {product_name}")
+        else:
+            # Fallback: Send download link
+            server_url = os.getenv("SERVER_URL", "https://your-huggingface-space-url.hf.space")
+            download_url = f"{server_url}/uploads/{filename}"
+            
+            message = (
+                f"📄 *{product_name} - Product Information*\n\n"
+                f"📎 [Download Product PDF]({download_url})\n\n"
+                f"💬 *Click the link above to download the detailed product information*\n"
+                f"Type 'main' to return to main menu."
+            )
+            send_whatsjet_message(from_number, message)
+            logger.info(f"[PDF] Sent PDF download link for product: {product_name}")
+            
+    except Exception as e:
+        logger.error(f"[PDF] Error sending enhanced PDF: {e}")
+        # Fallback to basic text response
+        response = generate_veterinary_product_response(product, {})
+        send_whatsjet_message(from_number, response)
+
+# Enhanced product response function with image support
+def generate_veterinary_product_response_with_media(product_info: Dict[str, Any], user_context: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Generate comprehensive veterinary product response with media information
+    Returns a dictionary with text response and media info
+    """
+    
+    def clean_text(text):
+        if pd.isna(text) or text is None:
+            return "Not specified"
+        return str(text).strip()
+    
+    # Extract product details
+    product_name = clean_text(product_info.get('Product Name', ''))
+    product_type = clean_text(product_info.get('Type', ''))
+    category = clean_text(product_info.get('Category', ''))
+    indications = clean_text(product_info.get('Indications', ''))
+    
+    # Check for PDF link in the CSV data
+    pdf_link = ""
+    try:
+        # Load CSV data to check for PDF link
+        csv_data = pd.read_csv('Veterinary.csv')
+        product_row = csv_data[csv_data['Product Name'] == product_name]
+        if not product_row.empty:
+            brochure_link = product_row.iloc[0].get('Brochure (PDF)', '')
+            if pd.notna(brochure_link) and brochure_link.strip():
+                pdf_link = brochure_link.strip()
+    except Exception as e:
+        logger.warning(f"Error checking PDF link for {product_name}: {e}")
+    
+    # Build the response
+    response_text = f"""🧪 *Name:* {product_name}
+📦 *Type:* {product_type}
+🏥 *Category:* {category}
+💊 *Used For:* {indications}"""
+    
+    # Add PDF link if available
+    if pdf_link:
+        response_text += f"\n\n📄 Product Brochure Available\n🔗 {product_name} PDF:\n{pdf_link}"
+    
+    # Add menu options
+    response_text += f"""
+
+💬 *Available Actions:*
+1️⃣ Talk to Veterinary Consultant
+2️⃣ Inquire About Availability
+3️⃣ View Product Image
+4️⃣ Download Detailed PDF
+5️⃣ Back to Main Menu
+
+💬 Select an option or ask about related products"""
+    
+    # Check for product image
+    image_path = get_product_image_path(product_name)
+    has_image = image_path is not None and os.path.exists(image_path)
+    
+    return {
+        'text': response_text,
+        'has_image': has_image,
+        'image_path': image_path,
+        'product_name': product_name
+    }
+
 if __name__ == "__main__":
     # Launch FastAPI app
     import uvicorn